fix: convert tz

TEAMSchools · Jan 11, 2024 · 4f6053f · 4f6053f
1 parent 3b97bd4
commit 4f6053f
Show file tree

Hide file tree

Showing 6 changed files with 20 additions and 9 deletions.
diff --git a/src/teamster/core/powerschool/schedules.py b/src/teamster/core/powerschool/schedules.py
@@ -74,8 +74,10 @@ def _schedule(
                         "partition_column"
                     ]
 
-                    latest_materialization_fmt = latest_materialization_datetime.format(
-                        "YYYY-MM-DDTHH:mm:ss.SSSSSS"
+                    latest_materialization_fmt = (
+                        latest_materialization_datetime.in_timezone(
+                            tz=execution_timezone
+                        ).format("YYYY-MM-DDTHH:mm:ss.SSSSSS")
                     )
 
                     [(count,)] = db_powerschool.engine.execute_query(

diff --git a/src/teamster/core/powerschool/sensors.py b/src/teamster/core/powerschool/sensors.py
@@ -15,7 +15,10 @@
 
 
 def build_powerschool_sensor(
-    name, asset_defs: list[AssetsDefinition], minimum_interval_seconds=None
+    name,
+    asset_defs: list[AssetsDefinition],
+    execution_timezone,
+    minimum_interval_seconds=None,
 ):
     @sensor(
         name=name,
@@ -61,8 +64,10 @@ def _sensor(
                     )  # type: ignore
                 )
 
-                latest_materialization_fmt = latest_materialization_datetime.format(
-                    "YYYY-MM-DDTHH:mm:ss.SSSSSS"
+                latest_materialization_fmt = (
+                    latest_materialization_datetime.in_timezone(
+                        tz=execution_timezone
+                    ).format("YYYY-MM-DDTHH:mm:ss.SSSSSS")
                 )
 
                 [(count,)] = db_powerschool.engine.execute_query(

diff --git a/src/teamster/kippcamden/powerschool/sensors.py b/src/teamster/kippcamden/powerschool/sensors.py
@@ -1,11 +1,12 @@
 from teamster.core.powerschool.sensors import build_powerschool_sensor
 
-from .. import CODE_LOCATION
+from .. import CODE_LOCATION, LOCAL_TIMEZONE
 from .assets import partition_assets
 
 partition_sensor = build_powerschool_sensor(
     name=f"{CODE_LOCATION}_powerschool_partition_sensor",
     asset_defs=partition_assets,
+    execution_timezone=LOCAL_TIMEZONE,
     minimum_interval_seconds=(60 * 10),
 )
 

diff --git a/src/teamster/kippmiami/powerschool/sensors.py b/src/teamster/kippmiami/powerschool/sensors.py
@@ -1,11 +1,12 @@
 from teamster.core.powerschool.sensors import build_powerschool_sensor
 
-from .. import CODE_LOCATION
+from .. import CODE_LOCATION, LOCAL_TIMEZONE
 from .assets import partition_assets
 
 partition_sensor = build_powerschool_sensor(
     name=f"{CODE_LOCATION}_powerschool_partition_sensor",
     asset_defs=partition_assets,
+    execution_timezone=LOCAL_TIMEZONE,
     minimum_interval_seconds=(60 * 10),
 )
 

diff --git a/src/teamster/kippnewark/powerschool/sensors.py b/src/teamster/kippnewark/powerschool/sensors.py
@@ -1,11 +1,12 @@
 from teamster.core.powerschool.sensors import build_powerschool_sensor
 
-from .. import CODE_LOCATION
+from .. import CODE_LOCATION, LOCAL_TIMEZONE
 from .assets import partition_assets
 
 partition_sensor = build_powerschool_sensor(
     name=f"{CODE_LOCATION}_powerschool_partition_sensor",
     asset_defs=partition_assets,
+    execution_timezone=LOCAL_TIMEZONE,
     minimum_interval_seconds=(60 * 10),
 )
 

diff --git a/tests/sensors/test_powerschool_sensor.py b/tests/sensors/test_powerschool_sensor.py
@@ -7,6 +7,7 @@
 )
 from teamster.core.utils.functions import get_dagster_cloud_instance
 from teamster.kippnewark.powerschool.assets import partition_assets
+from teamster.staging import LOCAL_TIMEZONE
 
 
 def test_powerschool_sensor():
@@ -15,7 +16,7 @@ def test_powerschool_sensor():
     )
 
     dynamic_partition_sensor = build_powerschool_sensor(
-        name="test", asset_defs=partition_assets
+        name="test", asset_defs=partition_assets, execution_timezone=LOCAL_TIMEZONE
     )
 
     sensor_results = dynamic_partition_sensor(