fix for new pandas-datareader version

master
Dmitry Maylarov 4 years ago
parent c99c22de3c
commit 16965ded91

@ -24,22 +24,22 @@ def get_data(
start=datetime.datetime.fromisoformat("1928-01-01"),
end=datetime.datetime.now(),
):
from pandas_datareader.yahoo.headers import DEFAULT_HEADERS
from pandas_datareader import data as pdr
from pandas_datareader.yahoo.daily import YahooDailyReader
import requests_cache
expire_after = datetime.timedelta(days=1)
session = requests_cache.CachedSession(
cache_name="cache", backend="sqlite", expire_after=expire_after
)
session.headers = DEFAULT_HEADERS
stockData = pdr.get_data_yahoo(
reader = YahooDailyReader(
stocks,
datetime.datetime.fromisoformat("1900-01-01"),
datetime.datetime.now(),
session=session,
)
return stockData.loc[start:end]
session.headers = reader.headers
return reader.read().loc[start:end]
def get_data_borders(stockData):

Loading…
Cancel
Save