full str

smithy-lang · Aug 9, 2024 · 75f1ddb · 75f1ddb
1 parent 7e84e50
commit 75f1ddb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/codegen/smithy-dafny-codegen/src/main/java/software/amazon/polymorph/CodegenConstants.java b/codegen/smithy-dafny-codegen/src/main/java/software/amazon/polymorph/CodegenConstants.java
@@ -1,5 +1,5 @@
 package software.amazon.polymorph;
 
 public class CodegenConstants {
-    public static String DAFNY_DATETIME_STRING_FORMAT = "%Y-%m-%dT%H:%M:%S";
+    public static String DAFNY_DATETIME_STRING_FORMAT = "%Y-%m-%dT%H:%M:%S.%fZ";
 }
diff --git a/...software/amazon/polymorph/smithypython/awssdk/shapevisitor/AwsSdkToDafnyShapeVisitor.java b/...software/amazon/polymorph/smithypython/awssdk/shapevisitor/AwsSdkToDafnyShapeVisitor.java
@@ -246,7 +246,8 @@ public String enumShape(EnumShape shape) {
   @Override
   public String timestampShape(TimestampShape shape) {
     writer.addStdlibImport("_dafny");
-    return "_dafny.Seq(%1$s.strftime(\"%2$s\"))".formatted(
+    writer.addStdlibImport("datetime", "timezone");
+    return "_dafny.Seq(%1$s.astimezone(timezone.utc).strftime(\"%2$s\"))".formatted(
             dataSource,
             DAFNY_DATETIME_STRING_FORMAT
     );