Export Lambda module as aws_lambda instead of lambda_

smithy-lang · Nov 15, 2022 · 75bd902 · 75bd902
1 parent d82ea4e
commit 75bd902
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/...amazon/smithy/rust/codegen/server/python/smithy/generators/PythonServerModuleGenerator.kt b/...amazon/smithy/rust/codegen/server/python/smithy/generators/PythonServerModuleGenerator.kt
@@ -167,14 +167,14 @@ class PythonServerModuleGenerator(
     private fun RustWriter.renderPyLambdaTypes() {
         rustTemplate(
             """
-            let lambda_ = #{pyo3}::types::PyModule::new(py, "lambda_")?;
-            lambda_.add_class::<#{SmithyPython}::lambda::PyLambdaContext>()?;
+            let aws_lambda = #{pyo3}::types::PyModule::new(py, "aws_lambda")?;
+            aws_lambda.add_class::<#{SmithyPython}::lambda::PyLambdaContext>()?;
             pyo3::py_run!(
                 py,
-                lambda_,
-                "import sys; sys.modules['$libName.lambda_'] = lambda_"
+                aws_lambda,
+                "import sys; sys.modules['$libName.aws_lambda'] = aws_lambda"
             );
-            m.add_submodule(lambda_)?;
+            m.add_submodule(aws_lambda)?;
             """,
             *codegenScope,
         )

diff --git a/rust-runtime/aws-smithy-http-server-python/examples/pokemon_service.py b/rust-runtime/aws-smithy-http-server-python/examples/pokemon_service.py
@@ -10,7 +10,7 @@
 from typing import List, Optional, Callable, Awaitable
 
 from libpokemon_service_server_sdk import App
-from libpokemon_service_server_sdk.lambda_ import LambdaContext  # type: ignore
+from libpokemon_service_server_sdk.aws_lambda import LambdaContext  # type: ignore
 from libpokemon_service_server_sdk.error import ResourceNotFoundException  # type: ignore
 from libpokemon_service_server_sdk.input import (  # type: ignore
     DoNothingInput,