moth/ctf/register.py

63 lines
1.9 KiB
Python
Executable File

#! /usr/bin/env python3
import cgitb; cgitb.enable()
import cgi
import fcntl
import string
from . import teams
from . import config
def main():
f = cgi.FieldStorage()
team = f.getfirst('team', '')
pw = f.getfirst('pw')
confirm_pw = f.getfirst('confirm_pw')
html = string.Template(config.start_html('Team Registration') +
('''
<p>
Pick a short team name: you'll be typing it a lot.
</p>
<form method="post" action="%s">
<fieldset>
<legend>Registration information:</legend>
<label>Team Name:</label>
<input type="text" name="team" />
<span class="error">$team_error</span><br />
<label>Password:</label>
<input type="password" name="pw" />
<br />
<label>Confirm Password:</label>
<input type="password" name="confirm_pw" />
<span class="error">$pw_match_error</span><br />
<input type="submit" value="Register" />
</fieldset>
</form>''' % config.url('register.cgi')) +
config.end_html())
if not (team and pw and confirm_pw): # If we're starting from the beginning?
html = html.substitute(team_error='',
pw_match_error='')
elif teams.exists(team):
html = html.substitute(team_error='Team team already taken',
pw_match_error='')
elif pw != confirm_pw:
html = html.substitute(team_error='',
pw_match_error='Passwords do not match')
else:
teams.add(team, pw)
html = (config.start_html('Team registered') +
('<p>Congratulations, <samp>%s</samp> is now registered. Go <a href="%s">back to the front page</a> and start playing!</p>' % (team, config.url(''))) +
config.end_html())
print(html)
if __name__ == '__main__':
main()