From 68f40e5c50291979afd82c355a92379f78bb9aa0 Mon Sep 17 00:00:00 2001
From: Andrew Dinh <adinh987@gmail.com>
Date: Mon, 17 Dec 2018 16:05:36 -0800
Subject: [PATCH] Moved main.py to StockData.py

---
 StockData.py                         | 145 +++++++++++++++++++++++++++
 __pycache__/StockData.cpython-37.pyc | Bin 0 -> 2413 bytes
 main.py                              |  82 +--------------
 3 files changed, 146 insertions(+), 81 deletions(-)
 create mode 100644 StockData.py
 create mode 100644 __pycache__/StockData.cpython-37.pyc

diff --git a/StockData.py b/StockData.py
new file mode 100644
index 0000000..e8cf182
--- /dev/null
+++ b/StockData.py
@@ -0,0 +1,145 @@
+# StockData.py
+# Andrew Dinh
+# Python 3.6.1
+# Description: 
+'''
+Returns all available dates and prices for each stock requested.
+'''
+
+# Alpha Vantage API Key: O42ICUV58EIZZQMU
+# Barchart API Key: a17fab99a1c21cd6f847e2f82b592838
+# Tiingo API Key: 2e72b53f2ab4f5f4724c5c1e4d5d4ac0af3f7ca8
+# If you're going to take these API keys and abuse it, you should really reconsider your life priorities
+
+apiAV = 'O42ICUV58EIZZQMU'
+apiBarchart = 'a17fab99a1c21cd6f847e2f82b592838'
+apiTiingo = '2e72b53f2ab4f5f4724c5c1e4d5d4ac0af3f7ca8'
+
+import requests, json
+'''
+def install(package):
+	if hasattr(pip, 'main'):
+        pip.main(['install', package])
+    else:
+        pip._internal.main(['install', package])
+'''
+class Stock:
+   def __init__(self, newStock = '', newfirstLastDates = [], newDates = []):
+      self.name = newStock
+      self.firstLastDates = newfirstLastDates
+      self.dates = newDates
+
+   def getDates(self): # returns beginning and end dates available
+      print("Getting available dates for", self, "...\n")
+      # Gets first and last possible dates from each source
+      # Also gets all dates from each source
+
+      # IEX
+      print("\nDates from IEX...")
+      firstLastDatesIEX = Stock.getDatesIEX(self, 'firstLast')
+      print("\nFirst and last dates:", firstLastDatesIEX)
+      datesIEX = Stock.getDatesIEX(self, 'all')
+      #print("All dates (recent first):", datesIEX, "\n")
+
+      # Alpha Vantage
+      print("\nDates from Alpha Vantage...")
+      firstLastDatesAV = Stock.getDatesAV(self, 'firstLast')
+      print("\nFirst and last dates:", firstLastDatesAV)
+      datesAV = Stock.getDatesAV(self, 'all')
+      #print("All dates (recent first):", datesAV, "\n")
+
+      #
+    
+   def getDatesIEX(self, which):
+      url = ''.join(('https://api.iextrading.com/1.0/stock/', self, '/chart/5y'))
+      #link = "https://api.iextrading.com/1.0/stock/spy/chart/5y"
+      print("URL:", url)
+      f = requests.get(url)
+      json_data = f.text
+      loaded_json = json.loads(json_data)
+      dates = []
+      if which == 'firstLast':
+         # Find firstDate (comes first)
+         firstLine = loaded_json[0]
+         #print("firstLine:", firstLine)
+         firstDate = firstLine['date']
+         #print("firstDate:",firstDate)
+         # Find finalDate (comes last)
+         #print("Length:", len(loaded_json))
+         lastLine = loaded_json[-1] # Returns last value of the list (Equivalent to len(loaded_json)-1)
+         #print("lastLine:", lastLine)
+         finalDate = lastLine['date']
+         #print("Final date:", finalDate)
+         dates.append((firstDate, finalDate))
+      else:
+#         for i in range(0, len(loaded_json), 1): # If you want to do oldest first
+         for i in range(len(loaded_json)-1, 0, -1):
+            line = loaded_json[i]
+            date = line['date']
+            dates.append(date)
+      return dates
+
+   def getDatesAV(self, which):
+      url = ''.join(('https://www.alphavantage.co/query?function=TIME_SERIES_MONTHLY&symbol=', self, '&apikey=', apiAV))
+      # https://www.alphavantage.co/query?function=TIME_SERIES_MONTHLY&symbol=MSFT&apikey=demo
+      print("URL:", url)
+      f = requests.get(url)
+      json_data = f.text
+      loaded_json = json.loads(json_data)
+      #print(loaded_json['Monthly Time Series'])
+      monthlyTimeSeries = loaded_json['Monthly Time Series']
+      #print(monthlyTimeSeries)
+      #print(len(monthlyTimeSeries))
+      #length = len(monthlyTimeSeries)
+      #print(monthlyTimeSeries['2018-12-17'])
+      #print(type(monthlyTimeSeries))
+      listOfDates = list(monthlyTimeSeries)
+      #print(listOfDates)
+      dates = []
+      if which == 'firstLast':
+         firstDate = listOfDates[-1]
+         lastDate = listOfDates[0]
+         #print("firstDate:", firstDate)
+         #print("lastDate:", lastDate)
+         dates.append((firstDate, lastDate))
+      else:
+         dates = listOfDates
+      return dates
+'''
+      # Find finalDate (comes first)
+      print("Length:", len(loaded_json))
+      lastLine = loaded_json[-1] # Returns last value of the list (Equivalent to len(loaded_json)-1)
+      print("lastLine:", lastLine)
+      finalDate = lastLine['date']
+      print("Final date:", finalDate)
+      dates = []
+      dates.append((firstDate, finalDate))
+'''
+'''
+  def printDates(self):
+    print("Getting data from IEX...")
+    url = ''.join(('https://api.iextrading.com/1.0/stock/', self.name, '/chart/5y'))
+    #link = "https://api.iextrading.com/1.0/stock/spy/chart/5y"
+    print("URL:", url)
+    f = requests.get(url)
+    #print(f.text)
+    json_data = f.text
+    loaded_json = json.loads(json_data)
+
+    print("Printing from IEX...")
+    for i in range (0,len(loaded_json),1):
+      a = loaded_json[i]
+      print(a['date'])
+'''
+
+def main():
+   #if __name__ == '__main__':
+        #  install('requests')
+
+   stock = 'spy'
+   spy = Stock(stock)
+   #print(spy.name)
+   Stock.getDates(stock)
+   #Stock.printDates(spy)
+
+main()
diff --git a/__pycache__/StockData.cpython-37.pyc b/__pycache__/StockData.cpython-37.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..92af2c90f8e842581027eb801c728fd4f7edf8c1
GIT binary patch
literal 2413
zcmZWqTW{Mo6ee}IoMcYBcD-)iuod-y*^TQoY0z#<o3ss(q*!j-gbV^BO14^Aaw)lK
z4CN`s_BY(e0sB7#^tz}1g}rR&(3iEH1Rj!y=kRjAkB1LtXKjMt%}a-Wo+sol*%=EJ
z>hGbXC^&Hlrw-+cqj1$xxyJQKb2M4&j{cZ%gPX4jH_sHu;1(~vCXVTwXM_^2?~<B5
zcwq1NSzp9yl|@mNy<}m;jv~LxS>}T-=GC4EJ!l4rsQS$7RMRZ+o>ztcqVK1f&+E2R
zdD?1jJlsE6TwdRBoFBIKohn;c3fR%zyKKR0E_nP-u-sbmo56DPXz^}y`S!ANv*|BE
z=XTI!N3CEnXe~8c-lDhQxA-D&G4B=&ZU;*qTYd}YB-<X!$v2Hb3p74~^Ak8fZId<$
zR1T4NO-HeS(`us#IQsbX2Djnd<wD=ZV-&>G#<OCrKzRryy$B+wIdMs@fTXz!QsD$#
z>2^&mw3u~$XD$dunr*T)Tf=A-x(r%PE40*)0>_H|W4Vem53ToY<)HNXod^?F`}^L*
zmBani7wIA6Vca=fOT2#9kF)gfahN^n9~}n$n12(-JoH$WfSH^creO8nNnyEe7>Akb
zUd8PjAP6<!AIwR!`efLnI0T;#1&=I`%b}40lXAk9GnHanJ)=<5AsnA;Z<JFCIz6Xn
z%2aEBRspR!wuBH#OI2E`V@uScr6z-CZ4a(K_OmRE+yCPl5Nx2;>vel@&X$p>1|sQJ
zH`adwWl@^OZ*b9mgmIVTj3D?UuC5d+FnDn7UEkHH*J0HI7H6#O!z=)e4s?z)fk+qS
zwm(7+)+p>rXZ4__3yh#MKV4Jq2do^6BhuKayg#^l;Fxl3^+4h|5*99FO=hk_FFgl>
z&>7vLGt{Ot+Nhr3G%@vXgZRxsL98TBe?WuNA)o|NRRi@61q3M9az#d48LGJgIyG0H
zX{TDI4~<-dzTPIMMozi@ToJuo=f-7nYUU<4p|+rJL0!sAvTx_c(9Ese4qzrq;^kL&
zGAk?P`WfY=OPEirGj)yp^66<g*Yh%X&g3)Fb5?rJqSuRGe<Znq`DI(+WgUFg$+BL$
z(rB<=SP%VInLv`a>t51rEYxo`QoM|fL8;+&n8+H7Cj)JNXLChh9u!E~Lc`#`{Rh6c
z*ss}z_ACiwF@uBFI2)yf3IQy%3|4^gG>Hp6N*GTIgY|lT%nMzxxa}8e<j1u+iE^PI
zcS5gIsC^L?G$=~w=E7>MD5I~>UDOB!!5OnKjQueo*I-wUg)H0zC631|LS;e2LW@8r
zD&Y75CB!V~7NMkHfFQJ@*uVt?ibgF3ehp<sy+SSc=jdR5bc-kK0RBx0QichTt&$)E
zX#7Wzj0On5k||S)s6!yjm?An*1i_XR0UGE4rjc2>aYj!~po|5SF@Q2+OqCMUNEJJ`
zhCr7oMYP-mTMOVWjp2SYf&2LQxX$nrz8v2{FKNKLAx^%7Z02QQ65rq3*jjgY*LOD7
zcipY0+j~zof4-5PbdQqg{=m8ce0lDl+#k_laDFR^vrcqU-3z;Zb=Mc6pN=>(4+ob-
zso4UaOj79J30D#6AwEWN0R`rD@d=7cAP}H56PKY`D@!tnYdCft#iuAfLs13c%y&mi
z<I1Cjoichq4G`^A(nEZXmXZyq{19}!$~#a}T!heb6VMgQuoM6uf3*UwLFIk8R}Z#p
zCZ>3NbzHZwUALR?euV8A*L~4v(P%`JU}EtR3S^kT72nwfzB2+-l1zVBppjzFq7~h`
zU|pA54;1pl1cY~2@(OJ^J%xuMC%_JP42JMn;8mAtUD+<ww0BZd<P>rPHO0~8trj)#
zeP<IgC2Cz3#=pV9=&90>_Tp>k*Q~;HU7mQZE53xjQwAI#Fd_4TQ-aQ32#F&ZVJeZt
hjY{8l7MY&3u<ngQ^)0GW42WiGaLqKOV$F3e`(M1gOvV5J

literal 0
HcmV?d00001

diff --git a/main.py b/main.py
index 77e1baa..415775e 100644
--- a/main.py
+++ b/main.py
@@ -15,84 +15,4 @@ Gives correlation value using equation at the end (from 0 to 1)
 FIRST TESTING WITH EXPENSE RATIO
 '''
 
-# Alpha Vantage API Key: O42ICUV58EIZZQMU
-# Barchart API Key: a17fab99a1c21cd6f847e2f82b592838
-# Tiingo API Key: 2e72b53f2ab4f5f4724c5c1e4d5d4ac0af3f7ca8
-# If you're going to take these API keys and abuse it, you should really reconsider your life priorities
-
-import requests, json
-
-def install(package):
-    if hasattr(pip, 'main'):
-        pip.main(['install', package])
-    else:
-        pip._internal.main(['install', package])
-
-class Stock:
-  def __init__(self, newStock = ''):
-    self.name = newStock
-
-  def getDates(self): # returns beginning and end dates available
-    print("Getting dates from sources...")
-    # Gets first and last possible dates from each source
-    # IEX
-    print("Getting dates from IEX")
-    datesIEX = Stock.getDatesIEX(self)
-    print(datesIEX)
-  '''
-    print("Getting dates from Alpha Vantage")
-    datesAV = Stock.getDatesAV(self)
-    print(datesAV)
-  '''
-    
-  def getDatesIEX(self):
-    dates = []
-    url = ''.join(('https://api.iextrading.com/1.0/stock/', self, '/chart/5y'))
-    #link = "https://api.iextrading.com/1.0/stock/spy/chart/5y"
-    print("URL:", url)
-    f = requests.get(url)
-    json_data = f.text
-    loaded_json = json.loads(json_data)
-    # Find firstDate (comes first)
-    firstLine = loaded_json[0]
-    print("firstLine:", firstLine)
-    firstDate = firstLine['date']
-    print("firstDate:",firstDate)
-    # Find finalDate (comes last)
-    print("Length:", len(loaded_json))
-    lastLine = loaded_json[-1] # Returns last value of the list (Equivalent to len(loaded_json)-1)
-    print("lastLine:", lastLine)
-    finalDate = lastLine['date']
-    print("Final date:", finalDate)
-    dates = []
-    dates.append((firstDate, finalDate))
-    return dates
-
-'''
-  def printDates(self):
-    print("Getting data from IEX...")
-    url = ''.join(('https://api.iextrading.com/1.0/stock/', self.name, '/chart/5y'))
-    #link = "https://api.iextrading.com/1.0/stock/spy/chart/5y"
-    print("URL:", url)
-    f = requests.get(url)
-    #print(f.text)
-    json_data = f.text
-    loaded_json = json.loads(json_data)
-
-    print("Printing from IEX...")
-    for i in range (0,len(loaded_json),1):
-      a = loaded_json[i]
-      print(a['date'])
-'''
-
-def main():
-  #if __name__ == '__main__':
-  #  install('requests')
-
-  stock = 'spy'
-  spy = Stock(stock)
-  #print(spy.name)
-  Stock.getDates(stock)
-  #Stock.printDates(spy)
-
-main()
+import StockData
\ No newline at end of file