diff --git a/ExpenseRatio.py b/ExpenseRatio.py deleted file mode 100644 index a64312e..0000000 --- a/ExpenseRatio.py +++ /dev/null @@ -1,30 +0,0 @@ -# ExpenseRatio.py -# Andrew Dinh -# Python 3.6.1 -# Description: -''' -Asks user for expense ratio of stock (I don't think there's an API for expense ratios) -Runs corrrelation study (I'm not sure if I want another class for this or not) -''' - -import numpy -#import urllib2, re -from urllib.request import urlopen -import re - - -class ExpenseRatio: - def __init__(self): - - -def main(): # For testing purposes - ''' - a = [1,2,3] - b = [2,4,6] - c = numpy.corrcoef(a, b)[0, 1] - print(c) - ''' - - -if __name__ == "__main__": - main() diff --git a/StockData.py b/StockData.py deleted file mode 100644 index 41f81f5..0000000 --- a/StockData.py +++ /dev/null @@ -1,586 +0,0 @@ -# StockData.py -# Andrew Dinh -# Python 3.6.1 -# Description: Returns all available dates and prices for each stock requested. - -import json -import requests -from datetime import datetime -''' -Available API's: Can it do mutual funds? -IEX: No -Alpha Vantage (AV): Yes -Tiingo: Yes -Barchart: No -''' - -# Alpha Vantage API Key: O42ICUV58EIZZQMU -# Barchart API Key: a17fab99a1c21cd6f847e2f82b592838 # Possible other one? f40b136c6dc4451f9136bb53b9e70ffa -# Tiingo API Key: 2e72b53f2ab4f5f4724c5c1e4d5d4ac0af3f7ca8 -# Tradier API Key: n26IFFpkOFRVsB5SNTVNXicE5MPD -# If you're going to take these API keys and abuse it, you should really reconsider your life priorities - -apiAV = 'O42ICUV58EIZZQMU' -# apiBarchart = 'a17fab99a1c21cd6f847e2f82b592838' # 150 getHistory queries per day -apiBarchart = 'f40b136c6dc4451f9136bb53b9e70ffa' -apiTiingo = '2e72b53f2ab4f5f4724c5c1e4d5d4ac0af3f7ca8' -apiTradier = 'n26IFFpkOFRVsB5SNTVNXicE5MPD' -''' -Monthly Bandwidth = 5 GB -Hourly Requests = 500 -Daily Requests = 20,000 -Symbol Requests = 500 -''' - - -class StockData: - - def __init__(self, newName='', newAbsFirstLastDates=[], newFinalDatesAndClose=[], newFinalDatesAndClose2=[], newAllLists=[]): - self.name = newName # Name of stock - # Absolute first and last dates from all sources - self.absFirstLastDates = newAbsFirstLastDates - # All available dates with corresponding close values - self.finalDatesAndClose = newFinalDatesAndClose - # After some consideration, I decided to keep what I had already done here and make a new list that's the same except dates are in datetime format - self.finalDatesAndClose2 = newFinalDatesAndClose2 - self.allLists = newAllLists - ''' - Format: - # List from each source containing: [firstDate, lastDate, allDates, values, timeFrame] - # firstDate & lastDate = '2018-12-18' (year-month-date) - allDates = ['2018-12-17', '2018-12-14'] (year-month-date) - values (close) = ['164.6307', 164.6307] - timeFrame = [days, weeks, years] - ''' - - def set(self, newName, newFirstLastDates, newAbsFirstLastDates, newFinalDatesAndClose, newAllLists): - self.name = newName # Name of stock - # Dates that at least 2 sources have (or should it be all?) - Maybe let user decide - self.firstLastDates = newFirstLastDates - # Absolute first and last dates from all sources - self.absFirstLastDates = newAbsFirstLastDates - self.finalDatesAndClose = newFinalDatesAndClose - self.allLists = newAllLists - - def setName(self, newName): - self.name = newName - - def returnName(self): - return self.name - - def returnAllLists(self): - return self.allLists - - def returnAbsFirstLastDates(self): - return self.absFirstLastDates - - def returnAllLists(self): - return self.allLists - - def returnFinalDatesAndClose(self): - return self.finalDatesAndClose - - def returnFinalDatesAndClose2(self): - return self.finalDatesAndClose2 - - def getIEX(self): - 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("\nSending request to:", url) - f = requests.get(url) - 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) - listIEX = [] - - print("\nFinding first and last date") - # Adding (firstDate, lastDate) to listIEX - # Find firstDate (comes first) - firstLine = loaded_json[0] - #print("firstLine:", firstLine) - firstDate = firstLine['date'] - # print("firstDate:",firstDate) - # Find lastDate (comes last) - # Returns last value of the list (Equivalent to len(loaded_json)-1) - lastLine = loaded_json[-1] - #print("lastLine:", lastLine) - lastDate = lastLine['date'] - #print("last date:", lastDate) - listIEX.append(firstDate) - listIEX.append(lastDate) - print(listIEX[0], ',', listIEX[1]) - - print("\nFinding all dates given") - allDates = [] -# for i in range(0, len(loaded_json), 1): # If you want to do oldest first - for i in range(len(loaded_json)-1, -1, -1): - line = loaded_json[i] - date = line['date'] - allDates.append(date) - listIEX.append(allDates) - - # print(listIEX[2]) - print(len(listIEX[2]), "dates") - - print("\nFinding close values for each date") - values = [] -# for i in range(0, len(loaded_json), 1): # If you want to do oldest first - for i in range(len(loaded_json)-1, -1, -1): - line = loaded_json[i] - value = line['close'] - values.append(value) - listIEX.append(values) - # print(listIEX[3]) - print(len(listIEX[3]), "close values") - - print("\nFinding time frame given [days, weeks, years]") - timeFrame = [] - d1 = datetime.strptime(firstDate, "%Y-%m-%d") - d2 = datetime.strptime(lastDate, "%Y-%m-%d") - timeFrameDays = abs((d2 - d1).days) - # print(timeFrameDays) - timeFrameYears = float(timeFrameDays / 365) - timeFrameWeeks = float(timeFrameDays / 7) - timeFrame.append(timeFrameDays) - timeFrame.append(timeFrameWeeks) - timeFrame.append(timeFrameYears) - listIEX.append(timeFrame) - print(listIEX[4]) - - return listIEX - - def getAV(self): - listAV = [] - #url = ''.join(('https://www.alphavantage.co/query?function=TIME_SERIES_MONTHLY&symbol=', self.name, '&apikey=', apiAV)) - # https://www.alphavantage.co/query?function=TIME_SERIES_MONTHLY&symbol=MSFT&apikey=demo - - #url = ''.join(('https://www.alphavantage.co/query?function=TIME_SERIES_DAILY&symbol=', self.name, '&outputsize=full&apikey=', apiAV)) - # https://www.alphavantage.co/query?function=TIME_SERIES_DAILY&symbol=MSFT&outputsize=full&apikey=demo - - url = ''.join(('https://www.alphavantage.co/query?function=TIME_SERIES_DAILY_ADJUSTED&symbol=', - self.name, '&outputsize=full&apikey=', apiAV)) - # https://www.alphavantage.co/query?function=TIME_SERIES_DAILY_ADJUSTED&symbol=MSFT&outputsize=full&apikey=demo - - print("\nSending request to:", url) - print("(This will take a while)") - f = requests.get(url) - json_data = f.text - 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']) - dailyTimeSeries = loaded_json['Time Series (Daily)'] - # print(monthlyTimeSeries) - listOfDates = list(dailyTimeSeries) - # print(listOfDates) - - firstDate = listOfDates[-1] - lastDate = listOfDates[0] - #print("firstDate:", firstDate) - #print("lastDate:", lastDate) - listAV.append(firstDate) - listAV.append(lastDate) - listAV.append(listOfDates) - - print("\nFinding first and last date") - print(listAV[0], ',', listAV[1]) - print("\nFinding all dates given") - # print(listAV[2]) - print(len(listAV[2]), "dates") - - print("\nFinding close values for each date") - values = [] - for i in range(0, len(listOfDates), 1): - temp = listOfDates[i] - loaded_json2 = dailyTimeSeries[temp] - #value = loaded_json2['4. close'] - value = loaded_json2['5. adjusted close'] - values.append(value) - listAV.append(values) - # print(listOfDates[0]) - #i = listOfDates[0] - # print(monthlyTimeSeries[i]) - # print(listAV[3]) - print(len(listAV[3]), "close values") - - print("\nFinding time frame given [days, weeks, years]") - timeFrame = [] - d1 = datetime.strptime(firstDate, "%Y-%m-%d") - d2 = datetime.strptime(lastDate, "%Y-%m-%d") - timeFrameDays = abs((d2 - d1).days) - # print(timeFrameDays) - timeFrameYears = float(timeFrameDays / 365) - timeFrameWeeks = float(timeFrameDays / 7) - timeFrame.append(timeFrameDays) - timeFrame.append(timeFrameWeeks) - timeFrame.append(timeFrameYears) - listAV.append(timeFrame) - print(listAV[4]) - - return listAV - - def getTiingo(self): - ''' - #OR we can use the token directly in the url - headers = { - 'Content-Type': 'application/json' - } - requestResponse = requests.get("https://api.tiingo.com/api/test?token=", - headers=headers) - print(requestResponse.json()) - ''' - token = ''.join(('Token ', apiTiingo)) - headers = { - 'Content-Type': 'application/json', - 'Authorization': token - } - url = ''.join(('https://api.tiingo.com/tiingo/daily/', self.name)) - print("\nSending request to:", url) - requestResponse = requests.get(url, headers=headers) - # print(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) - ''' - list1 = list(loaded_json) - for i in range (0, len(list1), 1): - if list1[i] == 'startDate': - startNum = i - elif list1[i] == 'endDate': - endNum = i - print(list1[startNum]) - print(list1[endNum]) - ''' - listTiingo = [] - - print("\nFinding first and last date") - firstDate = loaded_json['startDate'] - lastDate = loaded_json['endDate'] - # print(firstDate) - # print(lastDate) - listTiingo.append(firstDate) - listTiingo.append(lastDate) - print(listTiingo[0], ',', listTiingo[1]) - - print("\nFinding all dates given") - dates = [] - values = [] # Used loop for finding values - url2 = ''.join((url, '/prices?startDate=', - firstDate, '&endDate=', lastDate)) - # https://api.tiingo.com/tiingo/daily//prices?startDate=2012-1-1&endDate=2016-1-1 - print("\nSending request to:", url2) - requestResponse2 = requests.get(url2, headers=headers) - loaded_json2 = requestResponse2.json() - # print(loaded_json2) - # print(len(loaded_json2)) - for i in range(len(loaded_json2)-1, -1, -1): - line = loaded_json2[i] - dateWithTime = line['date'] - temp = dateWithTime.split('T00:00:00.000Z') - date = temp[0] - dates.append(date) - - value = line['close'] - values.append(value) - listTiingo.append(dates) - # print(listTiingo[2]) - print(len(listTiingo[2]), "dates") - - print("Finding close values for each date") - # Used loop from finding dates - listTiingo.append(values) - # print(listTiingo[3]) - print(len(listTiingo[3]), "close values") - - print("Finding time frame given [days, weeks, years]") - timeFrame = [] - d1 = datetime.strptime(firstDate, "%Y-%m-%d") - d2 = datetime.strptime(lastDate, "%Y-%m-%d") - timeFrameDays = abs((d2 - d1).days) - # print(timeFrameDays) - timeFrameYears = float(timeFrameDays / 365) - timeFrameWeeks = float(timeFrameDays / 7) - timeFrame.append(timeFrameDays) - timeFrame.append(timeFrameWeeks) - timeFrame.append(timeFrameYears) - listTiingo.append(timeFrame) - print(listTiingo[4]) - - return listTiingo - - def getFirstLastDate(self, listOfFirstLastDates): - listOfFirstDates = [] - listOfLastDates = [] - # print(len(listOfFirstLastDates)) - for i in range(0, len(listOfFirstLastDates), 1): - firstLastDates = listOfFirstLastDates[i] - firstDate = firstLastDates[0] - lastDate = firstLastDates[1] - listOfFirstDates.append(firstDate) - listOfLastDates.append(lastDate) - # print(listOfFirstDates) - # print(listOfLastDates) - for i in range(0, len(listOfFirstDates), 1): - date = listOfFirstDates[i] - if i == 0: - firstDate = date - yearMonthDay = firstDate.split('-') - firstYear = yearMonthDay[0] - firstMonth = yearMonthDay[1] - firstDay = yearMonthDay[2] - else: - yearMonthDay = date.split('-') - year = yearMonthDay[0] - month = yearMonthDay[1] - day = yearMonthDay[2] - if year < firstYear or (year == firstYear and month < firstMonth) or (year == firstYear and month == firstMonth and day < firstDay): - firstDate = date - firstYear = year - firstMonth = month - firstDay = day - # print(firstDate) - if len(listOfFirstDates) > 1: - for i in range(0, len(listOfLastDates), 1): - date = listOfLastDates[i] - if i == 0: - lastDate = date - yearMonthDay = lastDate.split('-') - lastYear = yearMonthDay[0] - lastMonth = yearMonthDay[1] - lastDay = yearMonthDay[2] - else: - yearMonthDay = date.split('-') - year = yearMonthDay[0] - month = yearMonthDay[1] - day = yearMonthDay[2] - if year > lastYear or (year == lastYear and month > lastMonth) or (year == lastYear and month == lastMonth and day > lastDay): - lastDate = date - lastYear = year - lastMonth = month - lastDay = day - # print(lastDate) - absFirstLastDates = [] - absFirstLastDates.append(firstDate) - absFirstLastDates.append(lastDate) - return absFirstLastDates - - def getFinalDatesAndClose(self): - # finalDates and finalClose will coincide (aka i = 1 will correspond to one another) - finalDatesAndClose = [] # Will combine finalDates then finalClose - finalDates = [] - finalClose = [] - # print(self.absFirstLastDates) - absFirstDate = self.absFirstLastDates[0] - absLastDate = self.absFirstLastDates[1] - date = absFirstDate - - allLists = self.allLists - while date != absLastDate: # DOESN'T DO LAST DATE - tempListOfClose = [] - found = False - for j in range(0, len(allLists), 1): # Look for date in all lists - list1 = allLists[j] - listOfDates = list1[2] - listOfClose = list1[3] - for k in range(0, len(listOfDates), 1): - if listOfDates[k] == date: - if found == False: - finalDates.append(date) - found = True - # print(listOfDates[k]) - # print(listOfClose[k]) - # print(listOfClose) - tempListOfClose.append(float(listOfClose[k])) - k = len(listOfDates) # Dates don't repeat - - if found == True: - sum = 0 - for r in range(0, len(tempListOfClose), 1): - sum = sum + tempListOfClose[r] - close = sum/len(tempListOfClose) - - finalClose.append(close) - # print(close) - - # Go to the next day - yearMonthDay = date.split('-') - year = int(yearMonthDay[0]) - month = int(yearMonthDay[1]) - day = int(yearMonthDay[2]) - - day = day + 1 - if day == 32 and month == 12: # Next year - day = 1 - month = 1 - year = year + 1 - elif day == 32: # Next month - month = month + 1 - day = 1 - if day < 10: - day = ''.join(('0', str(day))) - if month < 10: - month = ''.join(('0', str(month))) - date = ''.join((str(year), '-', str(month), '-', str(day))) - # print(date) - - # For last date - finalDates.append(date) - tempListOfClose = [] - for j in range(0, len(allLists), 1): # Look for date in all lists - list1 = allLists[j] - listOfDates = list1[2] - listOfClose = list1[3] - for k in range(0, len(listOfDates), 1): - if listOfDates[k] == date: - tempListOfClose.append(float(listOfClose[k])) - k = len(listOfDates) # Dates don't repeat - sum = 0 - for r in range(0, len(tempListOfClose), 1): - sum = sum + tempListOfClose[r] - close = sum/len(tempListOfClose) - finalClose.append(close) - # print(finalDates) - # print(finalClose) - - # Want lists from most recent to oldest, comment this out if you don't want that - finalDates = list(reversed(finalDates)) - finalClose = list(reversed(finalClose)) - - finalDatesAndClose.append(finalDates) - finalDatesAndClose.append(finalClose) - return finalDatesAndClose - - def datetimeDates(self): - finalDatesAndClose2 = [] - finalDatesAndClose = self.finalDatesAndClose - finalDatesStrings = finalDatesAndClose[0] - finalClose = finalDatesAndClose[1] - finalDates = [] - - from Functions import Functions - for i in range(0, len(finalDatesStrings), 1): - temp = Functions.stringToDate(finalDatesStrings[i]) - finalDates.append(temp) - # print(finalDates) - - finalDatesAndClose2.append(finalDates) - finalDatesAndClose2.append(finalClose) - return(finalDatesAndClose2) - - def is_connected(): - import socket # To check internet connection - 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 - print("\nNo internet connection!") - return False - - def main(self): - print('Beginning StockData.py') - - import importlib.util - import sys # To check whether a package is installed - - packages = ['requests'] - for i in range(0, len(packages), 1): - package_name = packages[i] - spec = importlib.util.find_spec(package_name) - if spec is None: - print(package_name + " is not installed\nPlease type in 'pip install -r requirements.txt' to install all required packages") - - # Test internet connection - internetConnection = StockData.is_connected() - if internetConnection == False: - return - - listOfFirstLastDates = [] - self.allLists = [] - - print('\nNOTE: Only IEX and Alpha Vantage support adjusted returns') - print('NOTE: Only Alpha Vantage and Tiingo support mutual fund data') - - # IEX - print("\nIEX") - listIEX = StockData.getIEX(self) - # print(listIEX) - if listIEX != 'Not available': - listOfFirstLastDates.append((listIEX[0], listIEX[1])) - self.allLists.append(listIEX) - - # Alpha Vantage - print("\nAlpha Vantage (AV)") - listAV = StockData.getAV(self) - # print(listAV) - if listAV != 'Not available': - listOfFirstLastDates.append((listAV[0], listAV[1])) - self.allLists.append(listAV) - - # COMMENTED OUT FOR NOW B/C LIMITED - ''' - print("\nTiingo") - print("NOTE: Tiingo does not return adjusted returns!!") - listTiingo = StockData.getTiingo(self) - #print(listTiingo) - if listTiingo != 'Not available': - listOfFirstLastDates.append((listTiingo[0], listTiingo[1])) - self.allLists.append(listTiingo) - ''' - - # print(self.allLists) - # print(listOfFirstLastDates) - if (len(self.allLists) > 0): - print("\n", end='') - print(len(self.allLists), "available source(s) for", self.name) - self.absFirstLastDates = StockData.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") - # Returns [List of Dates, List of Corresponding Close Values] - self.finalDatesAndClose = StockData.getFinalDatesAndClose(self) - #print("All dates available:", self.finalDatesAndClose[0]) - #print("All close values:\n", self.finalDatesAndClose[1]) - finalDates = self.finalDatesAndClose[0] - finalClose = self.finalDatesAndClose[1] - print(len(finalDates), "unique dates:", - finalDates[len(finalDates)-1], "...", finalDates[0]) - print(len(finalClose), "close values:", - finalClose[len(finalClose)-1], "...", finalClose[0]) - - print("\nConverting list of final dates to datetime\n") - self.finalDatesAndClose2 = StockData.datetimeDates(self) - # print(self.finalDatesAndClose2[0][0]) - - else: - print("No sources have data for", self.name) - - -def main(): # For testing purposes - stockName = 'spy' - stock1 = StockData(stockName) - print("Finding available dates and close values for", stock1.name) - StockData.main(stock1) - - -if __name__ == "__main__": - main() diff --git a/StockReturn.py b/StockReturn.py deleted file mode 100644 index 5e8f78a..0000000 --- a/StockReturn.py +++ /dev/null @@ -1,199 +0,0 @@ -# ExpenseRatio.py -# Andrew Dinh -# Python 3.6.7 -# Description: -''' -Calculates return for each stock from the lists from ExpenseRatio.py -listOfReturn = [Unadjusted Return, Sharpe Ratio, Sortino Ratio, Treynor Ratio, Jensen's Alpha] -''' - -from StockData import StockData -import datetime -from Functions import Functions - - -class Return: - def __init__(self, newListOfReturn=[], newTimeFrame=[], newBeta=0, newStandardDeviation=0, newNegativeStandardDeviation=0, newMarketReturn=0, newSize=0, newSizeOfNeg=0, newFirstLastDates=[], newAllLists=[], newAbsFirstLastDates=''): - self.listOfReturn = newListOfReturn - self.timeFrame = newTimeFrame # [years, months (30 days)] - self.beta = newBeta - self.standardDeviation = newStandardDeviation - self.negativeStandardDeviation = newNegativeStandardDeviation - self.marketReturn = newMarketReturn - self.size = newSize - self.sizeOfNeg = newSizeOfNeg - self.firstLastDates = newFirstLastDates - - def returnTimeFrame(self): - return self.timeFrame - - def setTimeFrame(self, newTimeFrame): - self.timeFrame = newTimeFrame - - def getFirstLastDates(self, stock): - firstLastDates = [] - timeFrame = self.timeFrame - firstDate = datetime.datetime.now( - ) - datetime.timedelta(days=timeFrame[0]*365) - firstDate = firstDate - datetime.timedelta(days=timeFrame[1]*30) - firstDate = ''.join( - (str(firstDate.year), '-', str(firstDate.month), '-', str(firstDate.day))) - - lastDate = StockData.returnAbsFirstLastDates(stock)[1] - # print(lastDate) - firstLastDates.append(firstDate) - firstLastDates.append(lastDate) - return firstLastDates - - def getFirstLastDates2(self, stock): - finalDatesAndClose = StockData.returnFinalDatesAndClose(stock) - finalDatesAndClose2 = StockData.returnFinalDatesAndClose2(stock) - firstDate = self.firstLastDates[0] - lastDate = self.firstLastDates[1] - finalDates = finalDatesAndClose[0] - - firstDateExists = False - lastDateExists = False - for i in range(0, len(finalDates), 1): - if finalDates[i] == str(firstDate): - firstDateExists = True - elif finalDates[i] == lastDate: - lastDateExists = True - i = len(finalDates) - - if firstDateExists == False: - print("Could not find first date. Changing first date to closest date") - tempDate = Functions.stringToDate(firstDate) # Change to datetime - print('Original first date:', tempDate) - #tempDate = datetime.date(2014,1,17) - newFirstDate = Functions.getNearest( - finalDatesAndClose2[0], tempDate) - print('New first date:', newFirstDate) - firstDate = str(newFirstDate) - - if lastDateExists == False: - print("Could not find final date. Changing final date to closest date") - tempDate2 = Functions.stringToDate(lastDate) # Change to datetime - print('Original final date:', tempDate2) - #tempDate2 = datetime.date(2014,1,17) - newLastDate = Functions.getNearest( - finalDatesAndClose2[0], tempDate2) - print('New final date:', newLastDate) - lastDate = str(newLastDate) - - firstLastDates = [] - firstLastDates.append(firstDate) - firstLastDates.append(lastDate) - return firstLastDates - - def getUnadjustedReturn(self, stock): - finalDatesAndClose = StockData.returnFinalDatesAndClose(stock) - firstDate = self.firstLastDates[0] - lastDate = self.firstLastDates[1] - finalDates = finalDatesAndClose[0] - finalClose = finalDatesAndClose[1] - - for i in range(0, len(finalDates), 1): - if finalDates[i] == str(firstDate): - firstClose = finalClose[i] - elif finalDates[i] == lastDate: - lastClose = finalClose[i] - i = len(finalDates) - - print('Close values:', firstClose, '...', lastClose) - fullUnadjustedReturn = float(lastClose/firstClose) - unadjustedReturn = fullUnadjustedReturn**( - 1/(self.timeFrame[0]+(self.timeFrame[1])*.1)) - return unadjustedReturn - - def getBeta(self): - # Can be calculated with correlation - import numpy as np - - finalDatesAndClose = StockData.returnFinalDatesAndClose(stock) - firstDate = self.firstLastDates[0] - lastDate = self.firstLastDates[1] - finalDates = finalDatesAndClose[0] - finalClose = finalDatesAndClose[1] - - for i in range(0, len(finalDates), 1): - if finalDates[i] == str(firstDate): - firstClose = finalClose[i] - - # list1 = - list2 = [1, 2, 4, 1] - - print(numpy.corrcoef(list1, list2)[0, 1]) - -# def getStandardDeviation(self, timeFrame): - - def mainBenchmark(self, stock): - print('Beginning StockReturn.py') - - # Find date to start from and last date - self.timeFrame = [] - self.listOfReturn = [] - - print("\nPlease enter a time frame in years: ", end='') - #timeFrameYear = int(input()) - timeFrameYear = 5 - print(timeFrameYear) - self.timeFrame.append(timeFrameYear) - print("Please enter a time frame in months (30 days): ", end='') - #timeFrameMonth = int(input()) - timeFrameMonth = 0 - print(timeFrameMonth) - self.timeFrame.append(timeFrameMonth) - # print(self.timeFrame) - self.firstLastDates = Return.getFirstLastDates(self, stock) - print('Dates: ', self.firstLastDates) - - print('\nMaking sure dates are within list...') - self.firstLastDates = Return.getFirstLastDates2(self, stock) - print('New dates: ', self.firstLastDates) - - print('\nGetting unadjusted return') - unadjustedReturn = Return.getUnadjustedReturn(self, stock) - self.listOfReturn.append(unadjustedReturn) - print('Average annual return for the past', - self.timeFrame[0], 'years and', self.timeFrame[1], 'months: ', end='') - print((self.listOfReturn[0]-1)*100, '%', sep='') - - def main(self, stock): - print('Beginning StockReturn.py') - - # Find date to start from and last date - self.listOfReturn = [] - - self.firstLastDates = Return.getFirstLastDates(self, stock) - print('Dates: ', self.firstLastDates) - - print('\nMaking sure dates are within list...') - self.firstLastDates = Return.getFirstLastDates2(self, stock) - print('New dates: ', self.firstLastDates) - - print('\nGetting unadjusted return') - unadjustedReturn = Return.getUnadjustedReturn(self, stock) - self.listOfReturn.append(unadjustedReturn) - print('Average annual return for the past', - self.timeFrame[0], 'years and', self.timeFrame[1], 'months: ', end='') - print((self.listOfReturn[0]-1)*100, '%', sep='') - - #print('\nGetting beta') - #beta = Return.getBeta(self, stock) - - -def main(): - stockName = 'spy' - stock1 = StockData(stockName) - print("Finding available dates and close values for", stock1.name) - StockData.main(stock1) - - stock1Return = Return() - Return.setTimeFrame(stock1Return, [5, 0]) - - Return.main(stock1Return, stock1) - - -if __name__ == "__main__": - main()