Set default namespace for json codec

Author: JordonPhillips

codegen/core/src/main/java/software/amazon/smithy/python/codegen/generators/ConfigGenerator.java

@@ -130,8 +130,8 @@ private static List<ConfigProperty> getProtocolProperties(GenerationContext cont
                         .build())
                 .documentation("The protocol to serialize and deserialize requests with.")
                 .initialize(w -> {
-                    w.write("self.protocol = protocol or $T()",
-                            context.protocolGenerator().getProtocolSymbol(context));
+                    w.write("self.protocol = protocol or ${C|}",
+                            w.consumer(writer -> context.protocolGenerator().initializeProtocol(context, writer)));
                 });
 
         var transportBuilder = ConfigProperty.builder()

codegen/core/src/main/java/software/amazon/smithy/python/codegen/generators/ProtocolGenerator.java

@@ -4,10 +4,10 @@
  */
 package software.amazon.smithy.python.codegen.generators;
 
-import software.amazon.smithy.codegen.core.Symbol;
 import software.amazon.smithy.model.shapes.ShapeId;
 import software.amazon.smithy.python.codegen.ApplicationProtocol;
 import software.amazon.smithy.python.codegen.GenerationContext;
+import software.amazon.smithy.python.codegen.writer.PythonWriter;
 import software.amazon.smithy.utils.SmithyUnstableApi;
 
 /**
@@ -41,7 +41,7 @@ default String getName() {
      */
     ApplicationProtocol getApplicationProtocol(GenerationContext context);
 
-    Symbol getProtocolSymbol(GenerationContext context);
+    void initializeProtocol(GenerationContext context, PythonWriter writer);
 
     /**
      * Generates the code for validating the generated protocol's serializers and deserializers.

codegen/core/src/main/java/software/amazon/smithy/python/codegen/integrations/RestJsonProtocolGenerator.java

@@ -6,7 +6,6 @@
 
 import java.util.Set;
 import software.amazon.smithy.aws.traits.protocols.RestJson1Trait;
-import software.amazon.smithy.codegen.core.Symbol;
 import software.amazon.smithy.model.node.ArrayNode;
 import software.amazon.smithy.model.node.ObjectNode;
 import software.amazon.smithy.model.shapes.ShapeId;
@@ -15,7 +14,9 @@
 import software.amazon.smithy.python.codegen.GenerationContext;
 import software.amazon.smithy.python.codegen.HttpProtocolTestGenerator;
 import software.amazon.smithy.python.codegen.SmithyPythonDependency;
+import software.amazon.smithy.python.codegen.SymbolProperties;
 import software.amazon.smithy.python.codegen.generators.ProtocolGenerator;
+import software.amazon.smithy.python.codegen.writer.PythonWriter;
 import software.amazon.smithy.utils.SmithyUnstableApi;
 
 /**
@@ -84,12 +85,12 @@ public ApplicationProtocol getApplicationProtocol(GenerationContext context) {
     }
 
     @Override
-    public Symbol getProtocolSymbol(GenerationContext context) {
-        return Symbol.builder()
-                .name("RestJsonClientProtocol")
-                .namespace("smithy_aws_core.aio.protocols", ".")
-                .dependencies(SmithyPythonDependency.SMITHY_AWS_CORE)
-                .build();
+    public void initializeProtocol(GenerationContext context, PythonWriter writer) {
+        writer.addDependency(SmithyPythonDependency.SMITHY_AWS_CORE);
+        writer.addImport("smithy_aws_core.aio.protocols", "RestJsonClientProtocol");
+        var serviceSymbol = context.symbolProvider().toSymbol(context.settings().service(context.model()));
+        var serviceSchema = serviceSymbol.expectProperty(SymbolProperties.SCHEMA);
+        writer.write("RestJsonClientProtocol($T)", serviceSchema);
     }
 
     // This is here rather than in HttpBindingProtocolGenerator because eventually

packages/smithy-aws-core/src/smithy_aws_core/aio/protocols.py

@@ -4,8 +4,9 @@
 
 from smithy_core.codecs import Codec
 from smithy_core.exceptions import DiscriminatorError
-from smithy_core.schemas import APIOperation
+from smithy_core.schemas import APIOperation, Schema
 from smithy_core.shapes import ShapeID, ShapeType
+from smithy_core.types import TimestampFormat
 from smithy_http.aio.interfaces import HTTPErrorIdentifier, HTTPResponse
 from smithy_http.aio.protocols import HttpBindingClientProtocol
 from smithy_json import JSONCodec, JSONDocument
@@ -50,10 +51,20 @@ class RestJsonClientProtocol(HttpBindingClientProtocol):
     """An implementation of the aws.protocols#restJson1 protocol."""
 
     _id: Final = RestJson1Trait.id
-    _codec: Final = JSONCodec(document_class=AWSJSONDocument)
     _contentType: Final = "application/json"
     _error_identifier: Final = AWSErrorIdentifier()
 
+    def __init__(self, service_schema: Schema) -> None:
+        """Initialize a RestJsonClientProtocol.
+
+        :param service: The schema for the service to interact with.
+        """
+        self._codec: Final = JSONCodec(
+            document_class=AWSJSONDocument,
+            default_namespace=service_schema.id.namespace,
+            default_timestamp_format=TimestampFormat.EPOCH_SECONDS,
+        )
+
     @property
     def id(self) -> ShapeID:
         return self._id