Bobby/stock_system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: Update stock data retrieval to use consistent 60-day lookback

Browse Source
This commit is contained in:
Bobby Abellana (aider) 2025-02-06 22:17:48 -08:00
parent 1d04986a07
commit bf4c5a1b94
No known key found for this signature in database
GPG Key ID: 647714CC45F3647B
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/screener/t_sunnyband.py
View File

@ -31,6 +31,10 @@ def get_stock_data(ticker: str, start_date: datetime, end_date: datetime, interv
"""Fetch stock data from the database""" """Fetch stock data from the database"""
client = create_client() client = create_client()
# Always look back 60 days for proper DMA calculation
today = datetime.now().date()
start_date = today - timedelta(days=60)
if interval == "daily": if interval == "daily":
table = "stock_prices_daily" table = "stock_prices_daily"
date_col = "date" date_col = "date"
@ -44,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.date()}' AND '{end_date.date()}' AND {date_col} BETWEEN '{start_date}' AND '{today}'
ORDER BY date ASC ORDER BY date ASC
""" """
else: else:
@ -69,7 +73,7 @@ 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 {date_col} BETWEEN toUnixTimestamp('{start_date.date()}') AND toUnixTimestamp('{end_date.date()}') AND {date_col} BETWEEN toUnixTimestamp('{start_date}') AND toUnixTimestamp('{today}')
GROUP BY interval_start GROUP BY interval_start
ORDER BY interval_start ASC ORDER BY interval_start ASC
""" """
@ -166,12 +170,11 @@ def run_sunny_scanner(min_price: float, max_price: float, min_volume: int) -> No
# Get user's preferred interval # Get user's preferred interval
interval = get_interval_choice() interval = get_interval_choice()
# Adjust date range based on interval # Set date range to 60 days for proper DMA calculation
end_date = datetime.now() end_date = datetime.now()
if interval == "daily": start_date = end_date - timedelta(days=60)
start_date = end_date - timedelta(days=60)
else: print(f"\nAnalyzing data from {start_date.date()} to {end_date.date()}")
start_date = end_date - timedelta(days=5) # Less history needed for intraday
# Get valid tickers # Get valid tickers
print("\nFetching qualified stocks...") print("\nFetching qualified stocks...")