30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
from datetime import date, datetime, timedelta
|
|
|
|
from database.ohlc import insert
|
|
from iqfeed import get_daily_data, get_historical_data, minutes
|
|
|
|
from symbols_and_timeframes import symbols_and_timeframes
|
|
|
|
start_date = datetime(2024, 11, 20)
|
|
end_date = date.today() - timedelta(days = 1)
|
|
|
|
for symbol, timeframe in symbols_and_timeframes:
|
|
if timeframe == '1d':
|
|
data = get_daily_data(symbol,
|
|
start_date = datetime.combine(start_date, datetime.min.time()),
|
|
end_date = datetime.combine(end_date, datetime.max.time()))
|
|
else: # Assuming minutes for now.
|
|
data = get_historical_data(symbol, minutes(int(timeframe[:-1])),
|
|
start_date = datetime.combine(start_date, datetime.min.time()),
|
|
end_date = datetime.combine(end_date, datetime.max.time()))
|
|
|
|
data['Symbol'] = symbol
|
|
data['Timeframe'] = timeframe
|
|
|
|
if 'Time' in data.columns:
|
|
data['Timestamp'] = data.apply(lambda row: datetime.combine(row['Date'], row['Time']), axis = 1)
|
|
else:
|
|
data['Timestamp'] = data['Date'].apply(lambda date: datetime.combine(date, datetime.min.time()))
|
|
|
|
data = data[['Symbol', 'Date', 'Timeframe', 'Timestamp', 'Open', 'High', 'Low', 'Close', 'Volume']]
|
|
insert(data) |