def before_request():
db.connect()
g.redis = redis.StrictRedis(host=config.redis.host,
port=config.redis.port,
db=config.redis.db,
password=config.redis.password)
g.connected = True
def run():
r = redis.StrictRedis(host=config.redis.host, port=config.redis.port, db=config.redis.db)
db.connect()
def set_complex(key, val):
r.set(key, json.dumps(val))
data = scoreboard.calculate_scores()
graphdata = scoreboard.calculate_graph(data)
set_complex("scoreboard", data)
set_complex("graph", graphdata)
db.close()
def run():
db.connect()
try:
shell = spur.SshShell(hostname=config.shell_host,
username=config.secret.shell_username,
private_key_file=config.secret.shell_privkey,
port=config.shell_port,
missing_host_key=spur.ssh.MissingHostKey.accept)
result = shell.run(["sudo", "useradd", "--help"])
if result.return_code != 0:
raise ValueError("Unable to sudo useradd.")
account_count = ssh.count_accounts()
if account_count >= config.shell_max_accounts:
raise ValueError("Max SSH accounts already created!")
free_account_count = ssh.count_unassigned()
new_accounts = max(config.shell_free_acounts - free_account_count, 0)
print("Checking that all teams have been assigned accounts...")
print("{}/{} shell accounts allocated adding {}...".format(
free_account_count, account_count, new_accounts))
teams = ssh.get_teams_without_ssh()
if len(teams) > 0:
print("{} teams without accounts present! Adding these as well.".
format(len(teams)))
new_accounts += len(teams)
accounts = []
while new_accounts > 0:
username = random.choice(config.shell_user_prefixes) + \
str(random.randint(0, config.shell_max_accounts))
plaintext_password = utils.misc.generate_random_string(
config.shell_password_length,
chars=
"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ123456789"
)
hashed_password = shell.run([
"bash", "-c", "echo '{}' | openssl passwd -1 -stdin".format(
plaintext_password)
])
hashed_password = hashed_password.output.decode("utf-8").strip()
shell_cmd = config.shell_user_creation.format(
username=username, password=hashed_password)
result = shell.run(shell_cmd.split(), allow_error=True)
if result.return_code == 9:
print("Collision! Retrying.")
continue
elif result.return_code != 0:
raise RuntimeError(result.stderr)
print("\t{}:{}".format(username, plaintext_password))
account = {
"username": username,
"password": plaintext_password,
"hostname": config.shell_host,
"port": config.shell_port
}
accounts.append(account)
new_accounts -= 1
if len(accounts) > 0:
ssh.create_accounts(accounts)
print("Successfully imported accounts.")
for team in teams:
ssh.assign_shell_account(team)
except spur.ssh.ConnectionError:
raise RuntimeError("Could not connect to shell server.")
db.close()
def before_request():
db.connect(reuse_if_open=True)
g.redis = redis.StrictRedis(host=config.redis.host,
port=config.redis.port,
db=config.redis.db)
g.connected = True