def save_new_secret_key(app):
if not os.path.exists(LOCAL_CONFIG_FILE):
copy_default_config_to_local()
config = ConfigParser()
config.read(LOCAL_CONFIG_FILE)
rng = SystemRandom()
secret_key = "".join([rng.choice(string.ascii_letters + string.digits) for _ in range(25)])
app.config['SECRET_KEY'] = secret_key
try:
config.set('faraday_server', 'secret_key', secret_key)
except NoSectionError:
config.add_section('faraday_server')
config.set('faraday_server', 'secret_key', secret_key)
with open(LOCAL_CONFIG_FILE, 'w') as configfile:
config.write(configfile)
def setup_storage_path():
default_path = join(CONST_FARADAY_HOME_PATH, 'storage')
if not os.path.exists(default_path):
logger.info('Creating directory {0}'.format(default_path))
os.mkdir(default_path)
config = ConfigParser()
config.read(faraday.server.config.LOCAL_CONFIG_FILE)
try:
config.add_section('storage')
config.set('storage', 'path', default_path)
except DuplicateSectionError:
logger.info('Duplicate section storage. skipping.')
with open(faraday.server.config.LOCAL_CONFIG_FILE, 'w') as configfile:
config.write(configfile)
return default_path
def setup_storage_path():
default_path = CONST_FARADAY_HOME_PATH / 'storage'
if not default_path.exists():
logger.info(f'Creating directory {default_path}')
default_path.mkdir()
config = ConfigParser()
config.read(faraday.server.config.LOCAL_CONFIG_FILE)
try:
config.add_section('storage')
config.set('storage', 'path', str(default_path))
except DuplicateSectionError:
logger.info('Duplicate section storage. skipping.')
with faraday.server.config.LOCAL_CONFIG_FILE.open('w') as configfile:
config.write(configfile)
return default_path
def _check_current_config(self, config):
try:
config.get('database', 'connection_string')
reconfigure = None
while not reconfigure:
reconfigure = input(
f'Database section {Fore.YELLOW} already found{Fore.WHITE}. Do you want to reconfigure database? (yes/no) ')
if reconfigure.lower() == 'no':
return False
elif reconfigure.lower() == 'yes':
continue
else:
reconfigure = None
except NoSectionError:
config.add_section('database')
return True
