def test_prompt_int_prefill(self):
# User enters empty string, select prefill
util.get_input = lambda msg: ''
self.assertEqual(util.prompt_int('msg', 15), 15)
def get_investment_settings(self):
"""
Display a series of prompts to get how the user wants to invest their available cash.
This fills out the investing dictionary.
"""
# Minimum cash
print '---------'
print 'The auto investor will automatically try to invest ALL available cash into a diversified portfolio'
while(True):
self.investing['minCash'] = util.prompt_int('What\'s the MINIMUM amount of cash you want to invest each round?', self.investing['minCash'])
if self.investing['minCash'] < 25:
print '\nYou cannot invest less than $25. Please try again.'
else:
break
# Min/max percent
print '---------'
while(True):
print 'When auto investing, the LendingClub API will provide us a list of possible investment portfolios available at that moment.'
print 'To pick the appropriate one for you, it needs to know what the minimum and maximum AVERAGE interest rate value you will accept.'
print 'The investment option closest to the maximum value will be chosen and all your available cash will be invested in it.\n'
self.investing['minPercent'] = util.prompt_float('What\'s MININUM average interest rate portfolio that you will accept?', self.investing['minPercent'])
# Max percent should default to being larger than the min percent
if self.investing['maxPercent'] is False or self.investing['maxPercent'] < self.investing['minPercent']:
self.investing['maxPercent'] = self.investing['minPercent'] + 1
self.investing['maxPercent'] = util.prompt_float('What\'s MAXIMUM average interest rate portfolio that you will accept?', self.investing['maxPercent'])
# Validation
if self.investing['maxPercent'] < self.investing['minPercent']:
print 'The maximum value must be larger than, or equal to, the minimum value. Please try again.'
elif self.investing['maxPercent'] == self.investing['minPercent']:
print 'It\'s very uncommon to find an available portfolio that will match an exact percent.'
if not util.prompt_yn('Would you like to specify a broader range?'):
break
else:
break
# Portfolio
print '---------'
folioOption = False
if self.investing['portfolio']: # if saved settings has a portfolio set, default the prompt to 'Y' to choose
folioOption = 'y'
if util.prompt_yn('Do you want to put your new investments into a named portfolio?', folioOption):
self.investing['portfolio'] = self.portfolio_picker(self.investing['portfolio'])
else:
self.investing['portfolio'] = False
# Advanced filter settings
print '\n---------'
if util.prompt_yn('Would you like to set advanced filter settings?', self.investing['filters'] is not False):
self.get_filter_settings()
# Review summary
self.show_summary()
if util.prompt_yn('Would you like to continue with these settings?', 'y'):
self.save()
else:
self.get_investment_settings()
return True
