sync fix

dmapp/dmweb/get_period_times.py

@@ -6,9 +6,31 @@ from zoneinfo import ZoneInfo
 
 from pymongo import MongoClient
 
-timezone = ZoneInfo("America/Argentina/Buenos_Aires")
+default_timezone = ZoneInfo("America/Argentina/Buenos_Aires")
+timezone = default_timezone  # Keep for backwards compatibility
 utctz = ZoneInfo("UTC")
 
+SUPPORTED_TIMEZONES = [
+    ("America/Argentina/Buenos_Aires", "Buenos Aires"),
+    ("America/New_York", "New York"),
+    ("America/Los_Angeles", "Los Angeles"),
+    ("Europe/London", "London"),
+    ("Europe/Paris", "Paris"),
+    ("UTC", "UTC"),
+]
+
+
+def get_timezone(tz_name=None):
+    """Get ZoneInfo for timezone name, with validation."""
+    if not tz_name:
+        return default_timezone
+    # Validate against supported list
+    valid_names = [tz[0] for tz in SUPPORTED_TIMEZONES]
+    if tz_name in valid_names:
+        return ZoneInfo(tz_name)
+    return default_timezone
+
+
 client = MongoClient(os.environ.get("MONGODB_HOST", "localhost"))
 db = client.deskmeter
 switches = db.switch
@@ -209,7 +231,7 @@ def get_work_period_totals(start, end):
 
 
 def get_task_blocks_calendar(
-    start, end, task=None, min_block_seconds=300, grid_hours=1
+    start, end, task=None, min_block_seconds=300, grid_hours=1, tz=None
 ):
     """
     Get task blocks for calendar-style visualization, aggregated by time grid.
@@ -231,6 +253,8 @@ def get_task_blocks_calendar(
         'active_ratio': float (always 1.0)
     }, ...]
     """
+    local_tz = tz if tz else default_timezone
+
     task_query = {"$in": task.split(",")} if task else {}
 
     match_query = {
@@ -252,7 +276,7 @@ def get_task_blocks_calendar(
 
     for switch in raw_switches:
         task_id = switch.get("task")
-        switch_start = switch["date"].replace(tzinfo=utctz).astimezone(timezone)
+        switch_start = switch["date"].replace(tzinfo=utctz).astimezone(local_tz)
         switch_duration = switch["delta"]
         switch_end = switch_start + timedelta(seconds=switch_duration)
 
@@ -290,7 +314,7 @@ def get_task_blocks_calendar(
     for (date, grid_hour, task_id), data in grid_task_time.items():
         if data["duration"] >= min_block_seconds:
             grid_start = datetime(
-                date.year, date.month, date.day, grid_hour, 0, 0, tzinfo=timezone
+                date.year, date.month, date.day, grid_hour, 0, 0, tzinfo=local_tz
             )
 
             blocks.append(
@@ -310,7 +334,7 @@ def get_task_blocks_calendar(
     return sorted(blocks, key=lambda x: (x["start"], x["task_path"]))
 
 
-def get_raw_switches(start, end, task=None):
+def get_raw_switches(start, end, task=None, tz=None):
     """
     Get all raw switch documents in the period.
 
@@ -323,6 +347,8 @@ def get_raw_switches(start, end, task=None):
         'delta': int (seconds)
     }, ...]
     """
+    local_tz = tz if tz else default_timezone
+
     task_query = {"$in": task.split(",")} if task else {}
 
     match_query = {"date": {"$gte": start, "$lte": end}}
@@ -342,7 +368,7 @@ def get_raw_switches(start, end, task=None):
                 "workspace": switch["workspace"],
                 "task_id": task_id,
                 "task_path": task_path,
-                "date": switch["date"].replace(tzinfo=utctz).astimezone(timezone),
+                "date": switch["date"].replace(tzinfo=utctz).astimezone(local_tz),
                 "delta": switch["delta"],
             }
         )