Update supabase_utils.py

supabase_utils.py

@@ -98,6 +98,56 @@ def get_harmonic_predictions(station_id, start_time, end_time):
         traceback.print_exc()
         return []
 
+def get_tide_predictions(station_id, start_time, end_time):
+    """사용자가 입력한 KST 시간을 UTC로 변환하여 조석 예측 데이터 조회"""
+    supabase = get_supabase_client()
+    if not supabase:
+        print("Supabase 클라이언트를 생성할 수 없습니다.")
+        return []
+    
+    try:
+        kst = pytz.timezone('Asia/Seoul')
+        
+        # 안전한 datetime 변환
+        if isinstance(start_time, str):
+            start_time = pd.to_datetime(start_time)
+        if isinstance(end_time, str):
+            end_time = pd.to_datetime(end_time)
+            
+        # pandas Timestamp를 datetime으로 변환
+        if hasattr(start_time, 'to_pydatetime'):
+            start_time = start_time.to_pydatetime()
+        if hasattr(end_time, 'to_pydatetime'):
+            end_time = end_time.to_pydatetime()
+        
+        # KST 시간대가 없으면 추가
+        if start_time.tzinfo is None:
+            start_time = kst.localize(start_time)
+        if end_time.tzinfo is None:
+            end_time = kst.localize(end_time)
+            
+        # UTC로 변환
+        start_utc = start_time.astimezone(pytz.UTC)
+        end_utc = end_time.astimezone(pytz.UTC)
+        
+        start_str = start_utc.strftime('%Y-%m-%dT%H:%M:%S+00:00')
+        end_str = end_utc.strftime('%Y-%m-%dT%H:%M:%S+00:00')
+        
+        result = supabase.table('tide_predictions')\
+            .select('*')\
+            .eq('station_id', station_id)\
+            .gte('predicted_at', start_str)\
+            .lte('predicted_at', end_str)\
+            .order('predicted_at')\
+            .limit(1000)\
+            .execute()
+        
+        return result.data if result.data else []
+    except Exception as e:
+        print(f"조석 예측값 조회 오류: {e}")
+        traceback.print_exc()
+        return []
+
 def save_predictions_to_supabase(station_id, prediction_results):
     """예측 결과를 Supabase에 저장"""
     supabase = get_supabase_client()