Update main.py

This commit is contained in:
Andrew Dinh 2018-12-16 17:58:25 -08:00
parent eba040ede9
commit 1682bb96ce

92
main.py
View File

@ -1,34 +1,98 @@
# main.py # main.py
# Andrew Dinh # Andrew Dinh
# Python 3.6.1 # 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...") 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" #link = "https://api.iextrading.com/1.0/stock/spy/chart/5y"
print("URL:", url) print("URL:", url)
f = requests.get(url) f = requests.get(url)
#print(f.text) #print(f.text)
json_data = f.text json_data = f.text
loaded_json = json.loads(json_data) 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): for i in range (0,len(loaded_json),1):
a = loaded_json[i] a = loaded_json[i]
print(a['date']) print(a['date'])
'''
def main(): def main():
stock = 'spy' #if __name__ == '__main__':
IEX(stock) # install('oauth2client')
main() stock = 'spy'
spy = Stock(stock)
#print(spy.name)
Stock.getDates(stock)
#Stock.printDates(spy)
main()