fix: Correct date formatting in stock data query for ClickHouse

This commit is contained in:
Bobby Abellana (aider) 2025-02-06 22:29:11 -08:00
parent 319f18d882
commit 62b63388a8
No known key found for this signature in database
GPG Key ID: 647714CC45F3647B

View File

@ -48,7 +48,7 @@ def get_stock_data(ticker: str, start_date: datetime, end_date: datetime, interv
volume volume
FROM stock_db.{table} FROM stock_db.{table}
WHERE ticker = '{ticker}' WHERE ticker = '{ticker}'
AND {date_col} BETWEEN '{start_date}' AND '{end_date}' AND {date_col} BETWEEN '{start_date.date()}' AND '{end_date.date()}'
ORDER BY date ASC ORDER BY date ASC
""" """
else: else:
@ -73,7 +73,9 @@ def get_stock_data(ticker: str, start_date: datetime, end_date: datetime, interv
sum(volume) as volume sum(volume) as volume
FROM stock_db.{table} FROM stock_db.{table}
WHERE ticker = '{ticker}' WHERE ticker = '{ticker}'
AND window_start/1000000000 BETWEEN toUnixTimestamp('{start_date}') AND toUnixTimestamp('{end_date}') AND window_start/1000000000 BETWEEN
toUnixTimestamp('{start_date.date()}') AND
toUnixTimestamp('{end_date.date()}')
GROUP BY interval_start GROUP BY interval_start
ORDER BY interval_start ASC ORDER BY interval_start ASC
""" """