Update api_utils.py

api_utils.py

@@ -3,40 +3,70 @@ import pandas as pd
 import pytz
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
+import logging
 from supabase_utils import get_supabase_client
 from config import STATION_NAMES
 
+# Basic logging configuration
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
 def fetch_tide_data(station_id, start_utc, end_utc, table='historical_tide'):
-    """공통 데이터 조회 함수"""
+    """Fetches data from a specified Supabase table within a date range."""
     supabase = get_supabase_client()
     if not supabase:
+        logging.error("Failed to create a Supabase client.")
         raise ValueError("Supabase 클라이언트를 생성할 수 없습니다.")
 
     try:
+        query_column = 'observed_at' if table == 'historical_tide' else 'predicted_at'
+        select_columns = 'observed_at, tide_level' if table == 'historical_tide' else 'predicted_at, final_tide_level'
+
         result = supabase.table(table) \
-            .select('observed_at, tide_level' if table == 'historical_tide' else 'predicted_at, final_tide_level') \
+            .select(select_columns) \
             .eq('station_id', station_id) \
-            .gte('observed_at' if table == 'historical_tide' else 'predicted_at', start_utc) \
-            .lte('observed_at' if table == 'historical_tide' else 'predicted_at', end_utc) \
-            .order('observed_at' if table == 'historical_tide' else 'predicted_at') \
+            .gte(query_column, start_utc) \
+            .lte(query_column, end_utc) \
+            .order(query_column) \
             .execute()
+
         if not result.data:
-            raise ValueError(f"데이터가 없습니다: {station_id}, {start_utc} ~ {end_utc}")
+            logging.warning(f"No data found for station {station_id} from {start_utc} to {end_utc} in table '{table}'.")
+            return pd.DataFrame()  # Return empty DataFrame for robustness
+
         return pd.DataFrame(result.data)
     except Exception as e:
+        logging.error(f"Error fetching data for station {station_id}: {e}", exc_info=True)
         raise ValueError(f"데이터 조회 오류: {e}")
 
 def get_tide_data(station_id, start_date=None, end_date=None, include_extremes=False, return_plot=False):
-    """조위 데이터 조회 및 시각화"""
+    """
+    Retrieves and processes tide data, optionally including tide extremes and a plot.
+
+    :param station_id: The station identifier.
+    :param start_date: Start date in 'YYYY-MM-DD' format. Defaults to today.
+    :param end_date: End date in 'YYYY-MM-DD' format. Defaults to the start date.
+    :param include_extremes: Whether to calculate and include tidal extremes (high/low tides).
+    :param return_plot: Whether to generate and return a Plotly figure.
+    :return: A dictionary containing the data, and optionally extremes and a plot.
+    """
+    # Default to today (in Seoul timezone) if start_date is not provided.
     start_date = start_date or datetime.now(pytz.timezone('Asia/Seoul')).strftime('%Y-%m-%d')
     start_time = pytz.timezone('Asia/Seoul').localize(datetime.strptime(start_date, '%Y-%m-%d'))
+
+    # The query period ends 24 hours after the start of the end_date.
+    # If no end_date is given, the period is 24 hours from the start_time.
     end_time = start_time + timedelta(hours=24) if not end_date else \
                pytz.timezone('Asia/Seoul').localize(datetime.strptime(end_date, '%Y-%m-%d')) + timedelta(hours=24)
 
+    # Convert local time to UTC for the database query.
     start_utc = start_time.astimezone(pytz.UTC).isoformat()
     end_utc = end_time.astimezone(pytz.UTC).isoformat()
 
     df = fetch_tide_data(station_id, start_utc, end_utc)
+    if df.empty:
+        logging.warning(f"No tide data available for station {station_id} for the selected period.")
+        return {"data": pd.DataFrame(), "extremes": pd.DataFrame(), "plot": go.Figure()}
+
     df['observed_at'] = pd.to_datetime(df['observed_at']).dt.tz_convert('Asia/Seoul')
     df['tide_level'] = pd.to_numeric(df['tide_level'])