DH-4749 Set timezone for last_schema_sync field (#191)

Dataherald · Oct 2, 2023 · 52dd426 · 52dd426
1 parent 79784de
commit 52dd426
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/dataherald/db_scanner/models/types.py b/dataherald/db_scanner/models/types.py
@@ -1,8 +1,8 @@
-from datetime import datetime
+from datetime import datetime, timezone
 from enum import Enum
 from typing import Any
 
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
 
 
 class ForeignKeyDetail(BaseModel):
@@ -39,3 +39,9 @@ class TableSchemaDetail(BaseModel):
     last_schema_sync: datetime | None
     status: str = TableDescriptionStatus.SYNCHRONIZED.value
     error_message: str | None
+
+    @validator("last_schema_sync", pre=True)
+    def parse_datetime_with_timezone(cls, value):
+        if not value:
+            return None
+        return value.replace(tzinfo=timezone.utc)  # Set the timezone to UTC