andrewkdinh/fund-indicators
1
0
Fork 0
mirror of https://github.com/andrewkdinh/fund-indicators.git synced 2024-11-21 20:34:21 -08:00
Code Issues Releases Wiki Activity

Added internet connection test

Browse Source
This commit is contained in:
Andrew Dinh 2018-12-21 09:59:40 -08:00
parent e43659ddc8
commit 8f7ea905d0
2 changed files with 106 additions and 75 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

163
StockData.py
View File

@ -27,17 +27,18 @@ Daily Requests = 20,000
Symbol Requests = 500 Symbol Requests = 500
''' '''
import requests, json import requests, json, socket
import importlib.util, sys # To check whether a package is installed import importlib.util, sys # To check whether a package is installed
from datetime import datetime from datetime import datetime
class Stock: class Stock:
def __init__(self, newName = '', newfirstLastDates = [], newAbsFirstLastDates = [], newfinalDatesAndClose = [], newAllLists = []): def __init__(self, newName = '', newFirstLastDates = [], newAbsFirstLastDates = [], newFinalDatesAndClose = [], newAllLists = []):
self.name = newName # Name of stock self.name = newName # Name of stock
self.firstLastDates = newfirstLastDates # Dates that at least 2 sources have (or should it be all?) - Maybe let user decide self.firstLastDates = newFirstLastDates # Dates that at least 2 sources have (or should it be all?) - Maybe let user decide
self.absFirstLastDates = newAbsFirstLastDates # Absolute first and last dates from all sources self.absFirstLastDates = newAbsFirstLastDates # Absolute first and last dates from all sources
self.finalDatesAndClose = newfinalDatesAndClose # All available dates self.finalDatesAndClose = newFinalDatesAndClose # All available dates
self.allLists = newAllLists
''' '''
Format: Format:
# List from each source containing: [firstDate, lastDate, allDates, values, timeFrame] # List from each source containing: [firstDate, lastDate, allDates, values, timeFrame]
@ -46,32 +47,16 @@ class Stock:
values (close) = ['164.6307', 164.6307] values (close) = ['164.6307', 164.6307]
timeFrame = [days, weeks, years] timeFrame = [days, weeks, years]
''' '''
self.allLists = newAllLists
''' def set(self, newName, newFirstLastDates, newAbsFirstLastDates, newFinalDatesAndClose, newAllLists):
def set(self, newName, newfirstLastDates, newAbsFirstLastDates, newDates, newListIEX, newListAV, newListTiingo):
self.name = newName # Name of stock self.name = newName # Name of stock
self.firstLastDates = newfirstLastDates # Dates that at least 2 sources have (or should it be all?) - Maybe let user decide self.firstLastDates = newFirstLastDates # Dates that at least 2 sources have (or should it be all?) - Maybe let user decide
self.absFirstLastDates = newAbsFirstLastDates # Absolute first and last dates from all sources self.absFirstLastDates = newAbsFirstLastDates # Absolute first and last dates from all sources
self.dates = newDates # All available dates self.finalDatesAndClose = newFinalDatesAndClose
self.allLists = newAllLists
# List from each source containing: ['firstDate', 'lastDate', allDates, values] def setName(self, newName):
self.listIEX = newListIEX # Dates available from IEX self.name = newName
self.listAV = newListAV # Dates available from AV
self.listTiingo = newListTiingo # Dates available from Tiingo
def setFirstLastDates(newFirstLastDates):
self.firstLastDates = newFirstLastDates
def setAbsFirstLastDates(newAbsFirstLastDates):
self.absFirstLastDates = newAbsFirstLastDates
def setDates(newDates):
self.dates = newDates
def setListIEX(newListIEX):
self.listIEX = newListIEX
def setListAV(newListAV):
self.listAV = newListAV
def setListTiingo(newListTiingo):
self.listTiingo = newListTiingo
'''
def getIEX(self): def getIEX(self):
url = ''.join(('https://api.iextrading.com/1.0/stock/', self.name, '/chart/5y')) url = ''.join(('https://api.iextrading.com/1.0/stock/', self.name, '/chart/5y'))
@ -79,6 +64,10 @@ class Stock:
print("\nSending request to:", url) print("\nSending request to:", url)
f = requests.get(url) f = requests.get(url)
json_data = f.text json_data = f.text
#print(json_data)
if (json_data == 'Unknown symbol'):
print("IEX not available")
return 'Not available'
loaded_json = json.loads(json_data) loaded_json = json.loads(json_data)
listIEX = [] listIEX = []
@ -145,6 +134,14 @@ class Stock:
f = requests.get(url) f = requests.get(url)
json_data = f.text json_data = f.text
loaded_json = json.loads(json_data) loaded_json = json.loads(json_data)
#print(loaded_json)
#print(type(loaded_json)) # Dictionary
#print(len(loaded_json))
if len(loaded_json) == 1:
print("Alpha Vantage not available")
return 'Not available'
#print(loaded_json['Monthly Time Series']) #print(loaded_json['Monthly Time Series'])
monthlyTimeSeries = loaded_json['Monthly Time Series'] monthlyTimeSeries = loaded_json['Monthly Time Series']
#print(monthlyTimeSeries) #print(monthlyTimeSeries)
@ -215,6 +212,10 @@ class Stock:
requestResponse = requests.get(url, headers=headers) requestResponse = requests.get(url, headers=headers)
#print(requestResponse.json()) #print(requestResponse.json())
loaded_json = requestResponse.json() loaded_json = requestResponse.json()
#print(len(loaded_json))
if len(loaded_json) == 1:
print("Tiingo not available")
return 'Not available'
#print(loaded_json) #print(loaded_json)
''' '''
list1 = list(loaded_json) list1 = list(loaded_json)
@ -264,7 +265,7 @@ class Stock:
# Used loop from finding dates # Used loop from finding dates
listTiingo.append(values) listTiingo.append(values)
#print(listTiingo[3]) #print(listTiingo[3])
print(len(listAV[3]), "close values") print(len(listTiingo[3]), "close values")
print("Finding time frame given [days, weeks, years]") print("Finding time frame given [days, weeks, years]")
timeFrame = [] timeFrame = []
@ -313,24 +314,25 @@ class Stock:
firstMonth = month firstMonth = month
firstDay = day firstDay = day
#print(firstDate) #print(firstDate)
for i in range(0, len(listOfLastDates),1): if len(listOfFirstDates) > 1:
date = listOfLastDates[i] for i in range(0, len(listOfLastDates),1):
if i == 0: date = listOfLastDates[i]
lastDate = date if i == 0:
yearMonthDay = lastDate.split('-') lastDate = date
lastYear = yearMonthDay[0] yearMonthDay = lastDate.split('-')
lastMonth = yearMonthDay[1] lastYear = yearMonthDay[0]
lastDay = yearMonthDay[2] lastMonth = yearMonthDay[1]
else: lastDay = yearMonthDay[2]
yearMonthDay = date.split('-') else:
year = yearMonthDay[0] yearMonthDay = date.split('-')
month = yearMonthDay[1] year = yearMonthDay[0]
day = yearMonthDay[2] month = yearMonthDay[1]
if year > lastYear or (year == lastYear and month > lastMonth) or (year == lastYear and month == lastMonth and day > lastDay): day = yearMonthDay[2]
lastDate = date if year > lastYear or (year == lastYear and month > lastMonth) or (year == lastYear and month == lastMonth and day > lastDay):
lastYear = year lastDate = date
lastMonth = month lastYear = year
lastDay = day lastMonth = month
lastDay = day
#print(lastDate) #print(lastDate)
absFirstLastDates = [] absFirstLastDates = []
absFirstLastDates.append(firstDate) absFirstLastDates.append(firstDate)
@ -384,7 +386,7 @@ class Stock:
day = day + 1 day = day + 1
if day == 32 and month == 12: # Next year if day == 32 and month == 12: # Next year
day = 1 day = 1
month = 2 month = 1
year = year + 1 year = year + 1
elif day == 32: # Next month elif day == 32: # Next month
month = month + 1 month = month + 1
@ -423,6 +425,16 @@ class Stock:
finalDatesAndClose.append(finalClose) finalDatesAndClose.append(finalClose)
return finalDatesAndClose return finalDatesAndClose
def is_connected():
try:
# connect to the host -- tells us if the host is actually
# reachable
socket.create_connection(("www.andrewkdinh.com", 80))
return True
except OSError:
pass
return False
def main(self): def main(self):
packages = ['requests'] packages = ['requests']
@ -432,7 +444,11 @@ class Stock:
if spec is None: if spec is None:
print(package_name +" is not installed\nPlease type in 'pip install -r requirements.txt' to install all required packages") print(package_name +" is not installed\nPlease type in 'pip install -r requirements.txt' to install all required packages")
# Makes list with ['firstDate', 'lastDate', [allDates], values] # Test internet connection
internetConnection = Stock.is_connected()
if internetConnection == False:
print("\nNo internet connection!")
return
listOfFirstLastDates = [] listOfFirstLastDates = []
@ -441,40 +457,48 @@ class Stock:
print("\nIEX") print("\nIEX")
listIEX = Stock.getIEX(self) listIEX = Stock.getIEX(self)
#print(listIEX) #print(listIEX)
listOfFirstLastDates.append((listIEX[0], listIEX[1])) if listIEX != 'Not available':
self.allLists.append(listIEX) listOfFirstLastDates.append((listIEX[0], listIEX[1]))
self.allLists.append(listIEX)
# Alpha Vantage # Alpha Vantage
print("\nAlpha Vantage (AV)") print("\nAlpha Vantage (AV)")
listAV = Stock.getAV(self) listAV = Stock.getAV(self)
#print(listAV) #print(listAV)
listOfFirstLastDates.append((listAV[0], listAV[1])) if listAV != 'Not available':
self.allLists.append(listAV) listOfFirstLastDates.append((listAV[0], listAV[1]))
self.allLists.append(listAV)
# COMMENTED OUT FOR NOW B/C LIMITED # COMMENTED OUT FOR NOW B/C LIMITED
''' '''
print("\nTiingo") print("\nTiingo")
listTiingo = Stock.getTiingo(self) listTiingo = Stock.getTiingo(self)
#print(listTiingo) #print(listTiingo)
listOfFirstLastDates.append((listTiingo[0], listTiingo[1])) if listTiingo != 'Not available':
self.allLists.append(listTiingo) listOfFirstLastDates.append((listTiingo[0], listTiingo[1]))
self.allLists.append(listTiingo)
''' '''
#print(self.allLists) #print(self.allLists)
#print(listOfFirstLastDates) #print(listOfFirstLastDates)
self.absFirstLastDates = Stock.getFirstLastDate(self, listOfFirstLastDates) print("\n")
print("\nThe absolute first date with close values is:", self.absFirstLastDates[0]) print(len(self.allLists), "available sources for", self.name)
print("The absolute last date with close values is:", self.absFirstLastDates[1]) if (len(self.allLists) > 0):
self.absFirstLastDates = Stock.getFirstLastDate(self, listOfFirstLastDates)
print("\nThe absolute first date with close values is:", self.absFirstLastDates[0])
print("The absolute last date with close values is:", self.absFirstLastDates[1])
print("\nCombining dates and averaging close values") print("\nCombining dates and averaging close values")
self.finalDatesAndClose = Stock.getFinalDatesAndClose(self) # Returns [List of Dates, List of Corresponding Close Values] self.finalDatesAndClose = Stock.getFinalDatesAndClose(self) # Returns [List of Dates, List of Corresponding Close Values]
#print("All dates available:", self.finalDatesAndClose[0]) #print("All dates available:", self.finalDatesAndClose[0])
#print("All close values:\n", self.finalDatesAndClose[1]) #print("All close values:\n", self.finalDatesAndClose[1])
finalDates = self.finalDatesAndClose[0] finalDates = self.finalDatesAndClose[0]
finalClose = self.finalDatesAndClose[1] finalClose = self.finalDatesAndClose[1]
print(len(finalDates), "dates:", finalDates[len(finalDates)-1], "...", finalDates[0]) print(len(finalDates), "unique dates:", finalDates[len(finalDates)-1], "...", finalDates[0])
print(len(finalClose), "close values:", finalClose[len(finalClose)-1], "...", finalClose[0]) print(len(finalClose), "close values:", finalClose[len(finalClose)-1], "...", finalClose[0])
#print("Uncomment above line in code to see output") #print("Uncomment above line in code to see output")
else:
print("No sources have data for", self.name)
def main(): # For testing purposes def main(): # For testing purposes
stockName = 'IWV' stockName = 'IWV'
@ -482,12 +506,5 @@ def main(): # For testing purposes
print("Finding available dates and close values for", stock1.name) print("Finding available dates and close values for", stock1.name)
Stock.main(stock1) Stock.main(stock1)
def imported(): # When this file has been imported
# Add stuff here
return
if __name__ == "__main__": if __name__ == "__main__":
main() main()
else:
imported()

18
main.py
View File

@ -17,9 +17,23 @@ FIRST TESTING WITH EXPENSE RATIO
from StockData import Stock from StockData import Stock
#listOfStocks = [spy, ] listOfStocks = []
numberOfStocks = int(input("How many stocks or mutual funds would you like to analyze? "))
for i in range(0, numberOfStocks, 1):
print("Stock", i+1, ": ", end='')
stockName = str(input())
listOfStocks.append(i)
listOfStocks[i] = Stock()
listOfStocks[i].setName(stockName)
print(listOfStocks[i].name)
for i in range(0, numberOfStocks, 1):
print("\n")
print(listOfStocks[i].name)
Stock.main(listOfStocks[i])
'''
stockName = 'IWV' stockName = 'IWV'
stock1 = Stock(stockName) stock1 = Stock(stockName)
print("Finding available dates and close values for", stock1.name) print("Finding available dates and close values for", stock1.name)
Stock.main(stock1) Stock.main(stock1)
'''