diff --git a/lib.py b/lib.py index 29359ee..297551b 100644 --- a/lib.py +++ b/lib.py @@ -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):