Stocker now supports BSE and NSE exchanges

learning_skills/images/navier_stokes.gif

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2673b62edc905bd725ffb71d73ac6e09f27df1e1fe77cb657330746b184bca7b
+size 16936

stocker/stocker.py

@@ -17,7 +17,7 @@ import matplotlib
 class Stocker():
     
     # Initialization requires a ticker symbol
-    def __init__(self, ticker):
+    def __init__(self, ticker, exchange='WIKI'):
         
         # Enforce capitalization
         ticker = ticker.upper()
@@ -30,7 +30,7 @@ class Stocker():
 
         # Retrieval the financial data
         try:
-            stock = quandl.get('WIKI/%s' % ticker)
+            stock = quandl.get('%s/%s' % (exchange, ticker))
         
         except Exception as e:
             print('Error Retrieving Data.')
@@ -42,6 +42,11 @@ class Stocker():
         
         # Columns required for prophet
         stock['ds'] = stock['Date']
+
+        if ('Adj. Close' not in stock.columns):
+            stock['Adj. Close'] = stock['Close']
+            stock['Adj. Open'] = stock['Open']
+        
         stock['y'] = stock['Adj. Close']
         stock['Daily Change'] = stock['Adj. Close'] - stock['Adj. Open']