Add createTime and updateTime to DateTimeTransformation for enhanced …

…timestamp handling
teamcalo · Jan 20, 2025 · aa8e910 · aa8e910
1 parent 5d6f4f8
commit aa8e910
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/connectors/aws-amplify-source/source_aws_amplify/components.py b/connectors/aws-amplify-source/source_aws_amplify/components.py
@@ -80,7 +80,13 @@ def __post_init__(self, parameters: Mapping[str, Any]):
 
 
 class DateTimeTransformation(RecordTransformation):
-    DATETIME_FIELDS = ["commitTime", "startTime", "endTime"]
+    DATETIME_FIELDS = [
+        "commitTime",
+        "startTime",
+        "endTime",
+        "createTime",
+        "updateTime",
+    ]
 
     def transform(
         self,

diff --git a/connectors/aws-amplify-source/source_aws_amplify/manifest.yaml b/connectors/aws-amplify-source/source_aws_amplify/manifest.yaml
@@ -61,6 +61,9 @@ definitions:
 
   apps_stream:
     $ref: "#/definitions/base_stream"
+    transformations:
+      - type: CustomTransformation
+        class_name: source_aws_amplify.components.DateTimeTransformation
     $parameters:
       name: "apps"
       primary_key: "appId"
@@ -79,6 +82,9 @@ definitions:
 
   branches_stream:
     $ref: "#/definitions/base_stream"
+    transformations:
+      - type: CustomTransformation
+        class_name: source_aws_amplify.components.DateTimeTransformation
     retriever:
       $ref: "#/definitions/retriever"
       requester: