From 16965ded917b4cba36fad70d22111c55b9659599 Mon Sep 17 00:00:00 2001
From: Dmitry Maylarov <maylarov@yandex.ru>
Date: Thu, 3 Feb 2022 21:08:53 +0300
Subject: [PATCH] fix for new pandas-datareader version

---
 lib.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

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):