mirror of
https://github.com/andrewkdinh/fund-indicators.git
synced 2024-11-24 10:14:19 -08:00
77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
# main.py
|
|
# Andrew Dinh
|
|
# Python 3.6.1
|
|
# Description:
|
|
'''
|
|
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)
|
|
|
|
FIRST TESTING WITH EXPENSE RATIO
|
|
'''
|
|
|
|
from StockData import StockData
|
|
|
|
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] = StockData()
|
|
listOfStocks[i].setName(stockName)
|
|
#print(listOfStocks[i].name)
|
|
|
|
sumOfListLengths = 0
|
|
for i in range(0, numberOfStocks, 1):
|
|
print(listOfStocks[i].name)
|
|
StockData.main(listOfStocks[i])
|
|
# Count how many stocks are available
|
|
temp = StockData.returnAllLists(listOfStocks[i])
|
|
sumOfListLengths = sumOfListLengths + len(temp)
|
|
|
|
if sumOfListLengths == 0:
|
|
print("No sources have data for given stocks")
|
|
exit()
|
|
|
|
# Find return over time using either Jensen's Alpha, Sharpe Ratio, Sortino Ratio, or Treynor Ratio
|
|
#from StockReturn import Return
|
|
|
|
|
|
# Runs correlation or regression study
|
|
#print(listOfStocks[0].name, listOfStocks[0].absFirstLastDates, listOfStocks[0].finalDatesAndClose)
|
|
indicatorFound = False
|
|
while indicatorFound == False:
|
|
print("\n1. Expense Ratio\n2. Asset Size\n3. Turnover\n4. Persistence\nWhich indicator would you like to look at? ", end='')
|
|
indicator = str(input())
|
|
indicatorFound = True
|
|
|
|
if indicator == 'Expense Ratio' or indicator == '1' or indicator == 'expense ratio':
|
|
#from ExpenseRatio import ExpenseRatio
|
|
print('\nExpense Ratio')
|
|
|
|
elif indicator == 'Asset Size' or indicator == '2' or indicator == 'asset size':
|
|
print('\nAsset Size')
|
|
|
|
elif indicator == 'Turnover' or indicator == '3' or indicator == 'turnover':
|
|
print('\nTurnover')
|
|
|
|
elif indicator == 'Persistence' or indicator == '4' or indicator == 'persistence':
|
|
print('\nPersistence')
|
|
|
|
else:
|
|
indicatorFound = False
|
|
print('\nInvalid input, please enter indicator again')
|
|
|
|
'''
|
|
stockName = 'IWV'
|
|
stock1 = Stock(stockName)
|
|
print("Finding available dates and close values for", stock1.name)
|
|
StockData.main(stock1)
|
|
'''
|