Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def setUpClass(cls):
cls.patcher = patch(
"tabpy.tabpy_server.app.app.TabPyApp._parse_cli_arguments",
return_value=Namespace(config=None),
)
cls.patcher.start()
prefix = "__TestEndpointsHandlerWithAuth_"
# create password file
cls.pwd_file = tempfile.NamedTemporaryFile(
mode="w+t", prefix=prefix, suffix=".txt", delete=False
)
username = "username"
password = "password"
cls.pwd_file.write(f"{username} {hash_password(username, password)}")
cls.pwd_file.close()
# create state.ini dir and file
cls.state_dir = tempfile.mkdtemp(prefix=prefix)
cls.state_file = open(os.path.join(cls.state_dir, "state.ini"), "w+")
cls.state_file.write(
"[Service Info]\n"
"Name = TabPy Serve\n"
"Description = \n"
"Creation Time = 0\n"
"Access-Control-Allow-Origin = \n"
"Access-Control-Allow-Headers = \n"
"Access-Control-Allow-Methods = \n"
"\n"
"[Query Objects Service Versions]\n"
"\n"
def setUpClass(cls):
cls.patcher = patch(
"tabpy.tabpy_server.app.app.TabPyApp._parse_cli_arguments",
return_value=Namespace(config=None),
)
cls.patcher.start()
prefix = "__TestEndpointHandlerWithAuth_"
# create password file
cls.pwd_file = tempfile.NamedTemporaryFile(
mode="w+t", prefix=prefix, suffix=".txt", delete=False
)
username = "username"
password = "password"
cls.pwd_file.write(f"{username} {hash_password(username, password)}")
cls.pwd_file.close()
# create state.ini dir and file
cls.state_dir = tempfile.mkdtemp(prefix=prefix)
cls.state_file = open(os.path.join(cls.state_dir, "state.ini"), "w+")
cls.state_file.write(
"[Service Info]\n"
"Name = TabPy Serve\n"
"Description = \n"
"Creation Time = 0\n"
"Access-Control-Allow-Origin = \n"
"Access-Control-Allow-Headers = \n"
"Access-Control-Allow-Methods = \n"
"\n"
"[Query Objects Service Versions]\n"
"\n"
def update_user(args, credentials):
username = args.username.lower()
logger.info(f'Updating username "{username}"')
if username not in credentials:
logger.error(
f'Username "{username}" not found in passwords file. '
'Do you want to run "add" command instead?'
)
return False
password = args.password
logger.info(f'Updating username "{username}" password to "{password}"')
credentials[username] = hash_password(username, password)
return store_passwords_file(args.pwdfile, credentials)
def add_user(args, credentials):
username = args.username.lower()
logger.info(f'Adding username "{username}"')
if username in credentials:
logger.error(
f"Can't add username {username} as it is already present"
" in passwords file. Do you want to run the "
'"update" command instead?'
)
return False
password = args.password
logger.info(f'Adding username "{username}" with password "{password}"...')
credentials[username] = hash_password(username, password)
if store_passwords_file(args.pwdfile, credentials):
logger.info(f'Added username "{username}" with password "{password}"')
else:
logger.info(
f'Could not add username "{username}" , ' f'password "{password}" to file'
)