def test_get_correct_answer():
correct_answer = u'Yea Boy'
result = get_correct_answer(u'Fav. phrase',
answer=correct_answer,
is_answer_correct=u'yes')
assert result == correct_answer
def add_url(self):
self.display_menu_header('Add Url')
url = get_correct_answer('Enter url to add.')
self.config.UPDATE_URLS.append(url)
self.config.UPDATE_URLS = list(set(self.config.UPDATE_URLS))
self.save(self.config)
input(u'\nUrl added. Press enter to continue')
self()
def add_url(self):
self.display_menu_header("Add Url")
url = get_correct_answer("Enter url to add.")
self.config.UPDATE_URLS.append(url)
self.config.UPDATE_URLS = list(set(self.config.UPDATE_URLS))
self.save(self.config)
input(u"\nUrl added. Press enter to continue")
self()
def update_settings(self):
log.debug(u'Update Settings Menu')
self.display_menu_header(u'Update Settings')
msg = (u'Enter the number of each item you\'d like to update.')
self.display_msg(msg)
self.display_msg(u'Examples:')
self.display_msg(u'-->13 ')
self.display_msg(u'-->235 ')
self.display_msg(u'-->513')
self.display_msg(u'Then press Enter')
print((u'1. App Name\n'
'2. Company Name\n'
'3. Username\n'
'4. Password\n'
'5. ssh key path\n'
'6. Remote dir or bucket name\n'
'7. Host settings\n'))
answers = input(u'-->')
self.display_menu_header(u'Updating Settings')
if '1' in answers:
app_name = get_correct_answer(u'Enter APP NAME',
default=self.config.APP_NAME)
self.config.APP_NAME = app_name
if '2' in answers:
if not hasattr(self.config, u'COMPANY_NAME'):
self.config.COMPANY_NAME = None
company_name = get_correct_answer(u'Enter Company Name',
default=self.config.COMPANY_NAME)
self.config.COMPANY_NAME = company_name
if '3' in answers:
username = get_correct_answer(u'Enter new username',
default=self.config.USERNAME)
self.config.USERNAME = username
if '4' in answers:
password = get_correct_answer(u'Enter new password',
default=self.config.PASSWORD)
self.config.PASSWORD = password
if '5' in answers:
password = get_correct_answer(u'Enter new ssh key path',
default=self.config.PASSWORD)
self.config.PASSWORD = password
if '6' in answers:
r_dir = get_correct_answer(u'Enter new remote directory or bucket',
default=self.config.REMOTE_DIR)
self.config.REMOTE_DIR = r_dir
if '7' in answers:
host = get_correct_answer(u'Enter host name or ip',
default=self.config.HOST)
self.config.HOST = host
self.save(self.config)
print(u'Saving new config....')
time.sleep(3)
self()
def update_settings(self):
log.debug(u"Update Settings Menu")
self.display_menu_header(u"Update Settings")
msg = u"Enter the number of each item you'd like to update."
self.display_msg(msg)
self.display_msg(u"Examples:")
self.display_msg(u"-->13 ")
self.display_msg(u"-->235 ")
self.display_msg(u"-->513")
self.display_msg(u"Then press Enter")
print(
(
u"1. App Name\n"
"2. Company Name\n"
"3. Username\n"
"4. Password\n"
"5. ssh key path\n"
"6. Remote dir or bucket name\n"
"7. Host settings\n"
)
)
answers = input(u"-->")
self.display_menu_header(u"Updating Settings")
if "1" in answers:
app_name = get_correct_answer(u"Enter APP NAME", default=self.config.APP_NAME)
self.config.APP_NAME = app_name
if "2" in answers:
if not hasattr(self.config, u"COMPANY_NAME"):
self.config.COMPANY_NAME = None
company_name = get_correct_answer(u"Enter Company Name", default=self.config.COMPANY_NAME)
self.config.COMPANY_NAME = company_name
if "3" in answers:
username = get_correct_answer(u"Enter new username", default=self.config.USERNAME)
self.config.USERNAME = username
if "4" in answers:
password = get_correct_answer(u"Enter new password", default=self.config.PASSWORD)
self.config.PASSWORD = password
if "5" in answers:
password = get_correct_answer(u"Enter new ssh key path", default=self.config.PASSWORD)
self.config.PASSWORD = password
if "6" in answers:
r_dir = get_correct_answer(u"Enter new remote directory or bucket", default=self.config.REMOTE_DIR)
self.config.REMOTE_DIR = r_dir
if "7" in answers:
host = get_correct_answer(u"Enter host name or ip", default=self.config.HOST)
self.config.HOST = host
self.save(self.config)
print(u"Saving new config....")
time.sleep(3)
self()
def test_get_correct_answer():
correct_answer = u'Yea Boy'
result = get_correct_answer(u'Fav. phrase', answer=correct_answer,
is_answer_correct=u'yes')
assert result == correct_answer
def initial_setup(self):
log.debug(u'Starting initial setup')
self.display_menu_header(u'Setup Assistant')
self.display_msg(u'Let\'s begin...')
self.config.APP_NAME = get_correct_answer(u'Please enter app name',
required=True)
self.config.COMPANY_NAME = get_correct_answer(u'Please enter your '
'company or name',
required=True)
self.config.DEV_DATA_DIR = cwd
url = get_correct_answer(u'Enter a url to ping for updates.',
required=True)
self.config.UPDATE_URLS = [url]
while 1:
answer = ask_yes_no(u'Would you like to add another '
'url for backup?', default='no')
if answer is True:
url = get_correct_answer(u'Enter another url.',
required=True)
self.config.UPDATE_URLS.append(url)
else:
break
self.config.UPDATE_PATCHES = ask_yes_no(u'Would you like to enable '
'patch updates?',
default='yes')
answer1 = ask_yes_no(u'Would you like to add scp settings?',
default='no')
answer2 = ask_yes_no(u'Would you like to add S3 settings?',
default='no')
if answer1:
self.config.REMOTE_DIR = get_correct_answer(u'Enter remote dir',
required=True)
self.config.HOST = get_correct_answer(u'Enter host', required=True)
self.config.USERNAME = get_correct_answer(u'Enter usernmae',
required=True)
key_path = get_correct_answer(u'Enter path to ssh key',
required=True)
# Path to private key
self.config.PASSWORD = _directory_fixer(key_path)
if answer2:
self.config.USERNAME = get_correct_answer(u'Enter access key ID',
required=True)
self.config.PASSWORD = get_correct_answer(u'Enter secret '
'Access Key',
required=True)
self.config.REMOTE_DIR = get_correct_answer(u'Enter bucket name',
required=True)
password = verify_password(u'Enter password')
self.file_crypt._update_timer()
self.save_config(self.config, password)
self.package_handler.setup()
print(u'Making keys...')
self.keys_menu.make_keys()
self.display_menu_header(u'Setup Complete')
self.display_msg(u'Now let\'s update some apps')
time.sleep(3)
def initial_setup(self):
log.debug(u'Starting initial setup')
self.display_menu_header(u'Setup Assistant')
self.display_msg(u'Let\'s begin...')
self.config.APP_NAME = get_correct_answer(u'Please enter app name',
required=True)
self.config.COMPANY_NAME = get_correct_answer(
u'Please enter your '
'company or name', required=True)
self.config.DEV_DATA_DIR = cwd
url = get_correct_answer(u'Enter a url to ping for updates.',
required=True)
self.config.UPDATE_URLS = [url]
while 1:
answer = ask_yes_no(
u'Would you like to add another '
'url for backup?',
default='no')
if answer is True:
url = get_correct_answer(u'Enter another url.', required=True)
self.config.UPDATE_URLS.append(url)
else:
break
self.config.UPDATE_PATCHES = ask_yes_no(
u'Would you like to enable '
'patch updates?', default='yes')
answer1 = ask_yes_no(u'Would you like to add scp settings?',
default='no')
answer2 = ask_yes_no(u'Would you like to add S3 settings?',
default='no')
if answer1:
self.config.REMOTE_DIR = get_correct_answer(u'Enter remote dir',
required=True)
self.config.HOST = get_correct_answer(u'Enter host', required=True)
self.config.USERNAME = get_correct_answer(u'Enter usernmae',
required=True)
key_path = get_correct_answer(u'Enter path to ssh key',
required=True)
# Path to private key
self.config.PASSWORD = _directory_fixer(key_path)
if answer2:
self.config.USERNAME = get_correct_answer(u'Enter access key ID',
required=True)
self.config.PASSWORD = get_correct_answer(
u'Enter secret '
'Access Key', required=True)
self.config.REMOTE_DIR = get_correct_answer(u'Enter bucket name',
required=True)
password = verify_password(u'Enter password')
self.file_crypt._update_timer()
self.save_config(self.config, password)
self.package_handler.setup()
print(u'Making keys...')
self.keys_menu.make_keys()
self.display_menu_header(u'Setup Complete')
self.display_msg(u'Now let\'s update some apps')
time.sleep(3)
