def run_2(master_articles, master_tickers, master_stock_data,
master_ticker_reference, selections):
# Need to get all input necessary to make function calls
prompt1 = '\nWhat tickers would you like to get financial data for? '
desired_tickers = raw_input(prompt1).split(' ')
prompt2 = '\nHow many years of financial data would you like to get? '
data_years = int(float(raw_input(prompt2)))
# Fill selections list to be passed to required functions
selections['Quickrun'] = 2
selections['Start'] = selections['End'] - datetime.timedelta(
days=(365 * data_years))
for ticker in desired_tickers:
selections['Tickers'].append(ticker)
print selections['Start'], selections['End']
# Add tickers to ticker list
menu_options.edit_ticker_list(master_tickers, 'quick', selections)
# Get market data for identified tickers
menu_options.get_financial_data(master_tickers, master_stock_data,
master_ticker_reference, 'quick',
selections)
# Analyze market data (do calculations) for identified tickers
menu_options.compute_stock_metrics(master_stock_data,
master_ticker_reference, 'quick',
selections)
def run_1(master_articles, master_tickers, master_stock_data,
master_ticker_reference, selections):
# Need to get all input necessary to make function calls
prompt1 = raw_input("\nHow many days back do you want to search? ")
days_to_search = int(float(prompt1))
prompt2 = '\nHow many top tickers would you like to identify? '
num_top_tickers = int(float(raw_input(prompt2)))
prompt3 = '\nHow many years of financial data would you like to get? '
data_years = int(float(raw_input(prompt3)))
# Fill selections list to be passed to required functions
selections['Quickrun'] = 1
selections['Days to Search'] = days_to_search
selections['Num Top Tickers'] = num_top_tickers
selections['Start'] = selections['End'] - datetime.timedelta(
days=(365 * data_years))
# Load articles from streetinsider
menu_options.load_articles(master_articles, master_tickers, 'quick',
selections)
# Need to clear out non-user added top tickers
for key, value in master_tickers.items():
if value != 'user added':
del master_tickers[key]
# Determine top tickers
menu_options.determine_top_tickers(master_articles, master_tickers,
'quick', selections)
for key, value in master_tickers.items():
if value != 'user added':
selections['Tickers'].append(key)
# Get market data for identified tickers
menu_options.get_financial_data(master_tickers, master_stock_data,
master_ticker_reference, 'quick',
selections)
# Analyze market data (do calculations) for identified tickers
menu_options.compute_stock_metrics(master_stock_data,
master_ticker_reference, 'quick',
selections)
menu_options.view_ticker_list(master_articles, master_tickers)
#########################################################################
# Option [6] Manually edit ticker list
elif use == 6:
menu_options.edit_ticker_list(master_tickers, 'manual', [])
#########################################################################
# Option [7] Get market data for ticker(s)
elif use == 7:
menu_options.get_financial_data(master_tickers, master_stock_data, master_ticker_reference, 'manual', [])
#########################################################################
# Option [8] Analyze market data
elif use == 8:
menu_options.compute_stock_metrics(master_stock_data, master_ticker_reference, 'manual', [])
#########################################################################
# Option [9] Exit program
elif use == 9:
menu_options.exit_program()
continue_code = False
#########################################################################
# Error catching
else:
print "\nInvalid selection. Please try again."
''' Kyle's stuff commented out for now
f.write('\n----------------------------\n----------------------------\n\n')
def run_3(master_articles, master_tickers, master_stock_data,
master_ticker_reference, selections):
# Get list of available Excel files from previous computations
available_metrics = glob2.glob('stock-data/*.xlsx')
# Ask user which file they want to use to load articles
prompt = '\nWhich stock metric Excel file do you want to load from?\n'
n = 0
for file_name in available_metrics:
n += 1
file_details = file_name.split('_')
file_details.reverse()
end_date = file_details[0][:-5]
start_date = file_details[2]
file_details[len(file_details) - 1] = file_details[len(file_details) -
1][11:]
ticker_list = ', '.join(file_details[3:len(file_details)])
prompt += '[{0}] Tickers: {1}\n'.format(n, ticker_list)
if n < 10:
space = ' '
else:
space = ' '
prompt += '{0}From: {1}\tTo: {2}\n\n'.format(space, start_date,
end_date)
prompt += 'Enter number: '
chosen_num = int(float(raw_input(prompt)))
chosen_file = available_metrics[chosen_num - 1]
# Get the tickers in the file by parsing the chosen file name
chosen_file_details = chosen_file.split('_')
chosen_file_details.reverse()
chosen_file_details[len(chosen_file_details) -
1] = chosen_file_details[len(chosen_file_details) -
1][11:]
file_tickers = chosen_file_details[3:len(chosen_file_details)]
# Get the start date of the data by parsing the chosen file name
start_date = chosen_file_details[2]
start_year = int(float(start_date.split('-')[0]))
start_month = int(float(start_date.split('-')[1]))
start_day = int(float(start_date.split('-')[2]))
start = datetime.datetime(start_year, start_month, start_day)
# Delete old Excel file
os.remove(chosen_file)
# Fill selections list to be passed to required functions
selections['Quickrun'] = 3
for ticker in file_tickers:
selections['Tickers'].append(ticker)
selections['Start'] = start
# Add tickers to ticker list
menu_options.edit_ticker_list(master_tickers, 'quick', selections)
# Get market data for identified tickers
menu_options.get_financial_data(master_tickers, master_stock_data,
master_ticker_reference, 'quick',
selections)
# Analyze market data (do calculations) for identified tickers
menu_options.compute_stock_metrics(master_stock_data,
master_ticker_reference, 'quick',
selections)
selections['Days to Search'] = int((end_date - start_date).days)
# Start and end dates
selections['Start'] = start_datetime
selections['End'] = end_datetime
#print selections
######################################################################################
# Call functions
######################################################################################
# Add tickers to ticker list
menu_options.edit_ticker_list(master_tickers, 'quick', selections)
# Get market data for identified tickers
menu_options.get_financial_data(master_tickers, master_stock_data,
master_ticker_reference, 'quick',
selections)
# Analyze market data (do calculations) for identified tickers
output = menu_options.compute_stock_metrics(master_stock_data,
master_ticker_reference,
'quick', selections)
print output
# Construct response
#output.seek(0)
#with ctx:
# send_file(output, attachment_filename="testing.xlsx", as_attachment=True)