mudar timezone

prefeitura-rio · Oct 18, 2023 · a8bb7f1 · a8bb7f1
1 parent ee0c440
commit a8bb7f1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pipelines/rj_smtr/utils.py b/pipelines/rj_smtr/utils.py
@@ -595,7 +595,11 @@ def custom_serialization(obj: Any) -> Any:
     Returns:
         Any: Serialized object
     """
-    if isinstance(obj, (datetime, date, pd.Timestamp)):
+    if isinstance(obj, pd.Timestamp):
+        if obj.tzinfo is None:
+            obj = obj.tz_localize("UTC").tz_convert(
+                emd_constants.DEFAULT_TIMEZONE.value
+            )
         # if obj.tzinfo is None:
         #     obj = obj.tz_localize(emd_constants.DEFAULT_TIMEZONE.value)
         # else: