Fallback to yfinance if encountering API error with pandas-datareader.

Author: stefmolin

requirements.txt

@@ -5,3 +5,4 @@ pandas-datareader>=0.7.0
 seaborn>=0.11.0
 statsmodels>=0.11.1
 mplfinance>=0.12.7a4
+yfinance>=0.2.4

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='stock_analysis',
-    version='0.2',
+    version='0.2.1',
     description='Classes for technical analysis of stocks.',
     author='Stefanie Molin',
     author_email='24376333+stefmolin@users.noreply.github.com',
@@ -11,11 +11,12 @@
     packages=['stock_analysis'],
     install_requires=[
         'matplotlib>=3.0.2',
+        'mplfinance>=0.12.7a4',
         'numpy>=1.15.2',
         'pandas>=0.23.4',
         'pandas-datareader>=0.7.0',
         'seaborn>=0.11.0',
         'statsmodels>=0.11.1',
-        'mplfinance>=0.12.7a4'
+        'yfinance>=0.2.4'
     ],
 )

stock_analysis/stock_reader.py

@@ -5,6 +5,7 @@
 
 import pandas as pd
 import pandas_datareader.data as web
+import yfinance as yf
 
 from .utils import label_sanitizer
 
@@ -94,8 +95,18 @@ def get_ticker_data(self, ticker):
         Returns:
             A `pandas.DataFrame` object with the stock data.
         """
-        return web.get_data_yahoo(ticker, self.start, self.end)
-
+        try:
+            return web.get_data_yahoo(ticker, self.start, self.end)
+        except TypeError: # API issue upstream – switch to yfinance
+            # https://github.com/pydata/pandas-datareader/issues/952
+            start, end = (
+                dt.datetime.strptime(str_date, '%Y%m%d')
+                for str_date in (self.start, self.end)
+            )
+            return yf.download(
+                ticker, start, end + dt.timedelta(days=1),
+                progress=False, ignore_tz=True
+            )
 
     def get_index_data(self, index):
         """