Update main.py

main.py

@@ -1,34 +1,98 @@
 # main.py
 # Andrew Dinh
 # Python 3.6.1
-# Description: Get, parse, and interpret JSON files from IEX
+# Description: 
-import urllib, requests, json
+'''
+Asks users for mutual funds/stocks to compare
+Asks to be compared (expense ratio, turnover, market capitalization, or persistence)
+Asks for time period (Possibly: 1 year, 5 years, 10 years)
+Makes the mutual funds as class Stock
+Gets data from each API
+Compare and contrast dates and end changeOverTime for set time period
+  NOTES: Later can worry about getting close values to make a graph or something
+Gives correlation value using equation at the end (from 0 to 1)
 
-class iex:
+FIRST TESTING WITH EXPENSE RATIO
+'''
 
-  def __init__(self, newStock = 'spy')
+# Alpha Vantage API Key: O42ICUV58EIZZQMU
-  self.iex
+# 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
 
-  def printDates(self)
+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/', stock, '/chart/5y'))
+    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(json_data)
-    #print(loaded_json)
-    #print(loaded_json[0])
 
-    print("Printing dates given from IEX...")
+    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('oauth2client')
+
   stock = 'spy'
-  IEX(stock)
+  spy = Stock(stock)
+  #print(spy.name)
+  Stock.getDates(stock)
+  #Stock.printDates(spy)
 
 main()