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

fix: Correct indentation error in get_stock_data function

Browse Source
This commit is contained in:
Bobby Abellana (aider) 2025-02-07 08:54:16 -08:00 committed by Bobby Abellana
parent b67586116c
commit 76075cd104
No known key found for this signature in database
GPG Key ID: 647714CC45F3647B
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/screener/t_sunnyband.py
View File

@ -96,24 +96,24 @@ def get_stock_data(ticker: str, start_date: datetime, end_date: datetime, interv
result.result_rows,
columns=['date', 'open', 'high', 'low', 'close', 'volume']
)
if interval != "daily" and interval != "5min":
# Resample to desired interval
df.set_index('date', inplace=True)
minutes = minutes_map[interval]
rule = f'{minutes}T'
if interval != "daily" and interval != "5min":
# Resample to desired interval
df.set_index('date', inplace=True)
minutes = minutes_map[interval]
rule = f'{minutes}T'
df = df.resample(rule).agg({
'open': 'first',
'high': 'max',
'low': 'min',
'close': 'last',
'volume': 'sum'
}).dropna()
df.reset_index(inplace=True)
return df
df = df.resample(rule).agg({
'open': 'first',
'high': 'max',
'low': 'min',
'close': 'last',
'volume': 'sum'
}).dropna()
df.reset_index(inplace=True)
return df
except Exception as e:
print(f"Error fetching data for {ticker}: {str(e)}")
return pd.DataFrame()