mirror of
https://github.com/andrewkdinh/fund-indicators.git
synced 2024-11-24 16:04:20 -08:00
Update main.py
This commit is contained in:
parent
eba040ede9
commit
1682bb96ce
92
main.py
92
main.py
@ -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()
|
||||||
|
Loading…
Reference in New Issue
Block a user