Python: Support document type

[unexge]

Motivation and Context

Support for document type in Python Server SDKs. Closes #2163.

Testing

I've added some Rust tests, but we need some end-to-end tests that also touches codegen, makes requests and asserts responses. I've realized that we don't run protocol tests for Python SDKs. Should we add protocol tests for Python? Or is there any other way to achieve this?

Update:
I've added some end-to-end tests with 94044a9. The reason I want end-to-end tests is that not every part of Python SSDK generation is tested thoroughly, for example if you add Timestamp to a model it fails to compile (I'll send a follow-up PR for to fix it). Examining as much of the SSDK generation in tests will be very helpful until we have a good test coverage.

You can also test it by adding a document type to Pokémon service.

Add `Echo` operation to Pokémon service

diff --git a/codegen-server-test/python/model/pokemon.smithy b/codegen-server-test/python/model/pokemon.smithy
index 84ce9fff8..fa3aac1d5 100644
--- a/codegen-server-test/python/model/pokemon.smithy
+++ b/codegen-server-test/python/model/pokemon.smithy
@@ -22,7 +22,8 @@ service PokemonService {
       GetServerStatistics,
       DoNothing,
       CheckHealth,
-        StreamPokemonRadio
+        StreamPokemonRadio,
+        Echo
   ],
}

@@ -41,3 +42,20 @@ structure StreamPokemonRadioOutput {

@streaming
blob StreamingBlob
+
+
+@http(uri: "/echo", method: "POST")
+operation Echo {
+    input: EchoInput,
+    output: EchoOutput,
+}
+
+@input
+structure EchoInput {
+    value: Document,
+}
+
+@output
+structure EchoOutput {
+    value: Document,
+}
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
index 1f45c5fe1..1cc5b7ecd 100644
--- a/rust-runtime/aws-smithy-http-server-python/examples/pokemon_service.py
+++ b/rust-runtime/aws-smithy-http-server-python/examples/pokemon_service.py
@@ -20,6 +20,7 @@ from pokemon_service_server_sdk.input import (  # type: ignore
   GetServerStatisticsInput,
   CheckHealthInput,
   StreamPokemonRadioInput,
+   EchoInput,
)
from pokemon_service_server_sdk.logging import TracingHandler  # type: ignore
from pokemon_service_server_sdk.middleware import (  # type: ignore
@@ -34,6 +35,7 @@ from pokemon_service_server_sdk.output import (  # type: ignore
   GetServerStatisticsOutput,
   CheckHealthOutput,
   StreamPokemonRadioOutput,
+   EchoOutput,
)
from pokemon_service_server_sdk.types import ByteStream  # type: ignore

@@ -244,6 +246,12 @@ def check_health(_: CheckHealthInput) -> CheckHealthOutput:
   return CheckHealthOutput()


+@app.echo
+def echo(input: EchoInput) -> EchoOutput:
+    logging.debug(f"Echoing back {input.value}")
+    return EchoOutput(input.value)
+
+
# Stream a random Pokémon song.
@app.stream_pokemon_radio
async def stream_pokemon_radio(

Run the service:

$ cd rust-runtime/aws-smithy-http-server-python/examples
$ make run

Test it:

$ curl -d '{"value": [1, 2, "foobar"]}' http:///localhost:13734/echo
{"value":[1,2,"foobar"]}

---

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

[github-actions]

A new generated diff is ready to view.

• AWS SDK (ignoring whitespace)
• No codegen difference in the Server Test
• No codegen difference in the Server Test Python

A new doc preview is ready to view.

[github-actions]

A new generated diff is ready to view.

• AWS SDK (ignoring whitespace)
• No codegen difference in the Server Test
• Server Test Python (ignoring whitespace)

A new doc preview is ready to view.

[github-actions]

A new generated diff is ready to view.

• AWS SDK (ignoring whitespace)
• No codegen difference in the Server Test
• Server Test Python (ignoring whitespace)

A new doc preview is ready to view.

[github-actions]

A new generated diff is ready to view.

• AWS SDK (ignoring whitespace)
• No codegen difference in the Server Test
• Server Test Python (ignoring whitespace)

A new doc preview is ready to view.

[crisidev]

LGTM. This will be properly tested once we add protocol tests and generate more complex models, but for now it is impossible without the Streaming trait..

[crisidev]

This is the same for also the rest of the types in this file. I think handling this in the JsonSerializer is hard as I was not able to find the right customization entrypoints at the time.

[github-actions]

A new generated diff is ready to view.

• AWS SDK (ignoring whitespace)
• No codegen difference in the Server Test
• Server Test Python (ignoring whitespace)

A new doc preview is ready to view.

[github-actions]

A new generated diff is ready to view.

• AWS SDK (ignoring whitespace)
• No codegen difference in the Server Test
• Server Test Python (ignoring whitespace)

A new doc preview is ready to view.