后端统一时区为 UTC

app/infra/db/timescaledb/repositories/scheme.py

@@ -1,11 +1,9 @@
 from typing import List, Any, Dict
-from datetime import datetime, timedelta, timezone
+from datetime import datetime, timedelta
 from collections import defaultdict
 from psycopg import AsyncConnection, Connection, sql
 import app.services.globals as globals
-
-# 定义UTC+8时区
-UTC_8 = timezone(timedelta(hours=8))
+from app.services.time_api import parse_utc_time
 
 
 class SchemeRepository:
@@ -466,24 +464,9 @@ class SchemeRepository:
             link_result_list: List of link simulation results
             result_start_time: Start time for the results (ISO format string)
         """
-        # Convert result_start_time string to datetime if needed
-        if isinstance(result_start_time, str):
-            # 如果是ISO格式字符串，解析并转换为UTC+8
-            if result_start_time.endswith("Z"):
-                # UTC时间，转换为UTC+8
-                utc_time = datetime.fromisoformat(
-                    result_start_time.replace("Z", "+00:00")
-                )
-                simulation_time = utc_time.astimezone(UTC_8)
-            else:
-                # 假设已经是UTC+8时间
-                simulation_time = datetime.fromisoformat(result_start_time)
-                if simulation_time.tzinfo is None:
-                    simulation_time = simulation_time.replace(tzinfo=UTC_8)
-        else:
-            simulation_time = result_start_time
-            if simulation_time.tzinfo is None:
-                simulation_time = simulation_time.replace(tzinfo=UTC_8)
+        simulation_time = parse_utc_time(
+            result_start_time, field_name="result_start_time"
+        )
 
         timestep_parts = globals.hydraulic_timestep.split(":")
         timestep = timedelta(
@@ -564,24 +547,9 @@ class SchemeRepository:
             link_result_list: List of link simulation results
             result_start_time: Start time for the results (ISO format string)
         """
-        # Convert result_start_time string to datetime if needed
-        if isinstance(result_start_time, str):
-            # 如果是ISO格式字符串，解析并转换为UTC+8
-            if result_start_time.endswith("Z"):
-                # UTC时间，转换为UTC+8
-                utc_time = datetime.fromisoformat(
-                    result_start_time.replace("Z", "+00:00")
-                )
-                simulation_time = utc_time.astimezone(UTC_8)
-            else:
-                # 假设已经是UTC+8时间
-                simulation_time = datetime.fromisoformat(result_start_time)
-                if simulation_time.tzinfo is None:
-                    simulation_time = simulation_time.replace(tzinfo=UTC_8)
-        else:
-            simulation_time = result_start_time
-            if simulation_time.tzinfo is None:
-                simulation_time = simulation_time.replace(tzinfo=UTC_8)
+        simulation_time = parse_utc_time(
+            result_start_time, field_name="result_start_time"
+        )
 
         timestep_parts = globals.hydraulic_timestep.split(":")
         timestep = timedelta(
@@ -664,21 +632,7 @@ class SchemeRepository:
         Returns:
             List of records matching the criteria
         """
-        # Convert query_time string to datetime
-        if isinstance(query_time, str):
-            if query_time.endswith("Z"):
-                # UTC时间，转换为UTC+8
-                utc_time = datetime.fromisoformat(query_time.replace("Z", "+00:00"))
-                target_time = utc_time.astimezone(UTC_8)
-            else:
-                # 假设已经是UTC+8时间
-                target_time = datetime.fromisoformat(query_time)
-                if target_time.tzinfo is None:
-                    target_time = target_time.replace(tzinfo=UTC_8)
-        else:
-            target_time = query_time
-            if target_time.tzinfo is None:
-                target_time = target_time.replace(tzinfo=UTC_8)
+        target_time = parse_utc_time(query_time, field_name="query_time")
 
         # Create time range: query_time ± 1 second
         start_time = target_time - timedelta(seconds=1)
@@ -727,21 +681,7 @@ class SchemeRepository:
         Returns:
             List of records matching the criteria
         """
-        # Convert query_time string to datetime
-        if isinstance(query_time, str):
-            if query_time.endswith("Z"):
-                # UTC时间，转换为UTC+8
-                utc_time = datetime.fromisoformat(query_time.replace("Z", "+00:00"))
-                target_time = utc_time.astimezone(UTC_8)
-            else:
-                # 假设已经是UTC+8时间
-                target_time = datetime.fromisoformat(query_time)
-                if target_time.tzinfo is None:
-                    target_time = target_time.replace(tzinfo=UTC_8)
-        else:
-            target_time = query_time
-            if target_time.tzinfo is None:
-                target_time = target_time.replace(tzinfo=UTC_8)
+        target_time = parse_utc_time(query_time, field_name="query_time")
 
         # Create time range: query_time ± 1 second
         start_time = target_time - timedelta(seconds=1)