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

fix: Update intraday ticker query to use today's market hours

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

9
src/screener/t_sunnyband.py
View File

@ -93,6 +93,7 @@ def get_valid_tickers(min_price: float, max_price: float, min_volume: int, inter
"""Get tickers that meet the price and volume criteria""" """Get tickers that meet the price and volume criteria"""
client = create_client() client = create_client()
yesterday = (datetime.now() - timedelta(days=1)).date() yesterday = (datetime.now() - timedelta(days=1)).date()
today = datetime.now().date()
if interval == "daily": if interval == "daily":
table = "stock_prices_daily" table = "stock_prices_daily"
@ -101,10 +102,10 @@ def get_valid_tickers(min_price: float, max_price: float, min_volume: int, inter
else: else:
table = "stock_prices" table = "stock_prices"
date_col = "window_start" date_col = "window_start"
# Get yesterday's timestamp range # Get today's trading hours timestamp range (9:30 AM to 4:00 PM EST)
start_ts = int(datetime.combine(yesterday, datetime.min.time()).timestamp()) market_open = int(datetime.combine(today, datetime.strptime("09:30", "%H:%M").time()).timestamp())
end_ts = int(datetime.combine(yesterday, datetime.max.time()).timestamp()) market_close = int(datetime.combine(today, datetime.strptime("16:00", "%H:%M").time()).timestamp())
date_condition = f"{date_col} BETWEEN {start_ts} AND {end_ts}" date_condition = f"{date_col} BETWEEN {market_open} AND {market_close}"
query = f""" query = f"""
SELECT DISTINCT ticker SELECT DISTINCT ticker