101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
import bcrypt
|
|
import datetime
|
|
|
|
def create_user(db, login_info, user_info):
|
|
|
|
"""
|
|
Creates a new user in the database
|
|
|
|
Args:
|
|
db (psycopg2.connection): connection to postgres database
|
|
login_info (dict of str:str): username and password info
|
|
user_info (dict of str:value): name (str) and dob (datetime.date) info
|
|
|
|
Returns:
|
|
int: 0 if the username is non-unique, user id otherwise
|
|
"""
|
|
|
|
username, password = login_info["username"], login_info["password"]
|
|
password_hash = bcrypt.hashpw(password.encode(), bcrypt.gensalt()).decode()
|
|
name, dob = user_info["name"], user_info["dob"]
|
|
with db:
|
|
with db.cursor() as curs:
|
|
curs.execute("SELECT * FROM Users WHERE username = %s", (username,))
|
|
conflicts = curs.fetchall()
|
|
if conflicts:
|
|
return 0
|
|
curs.execute("INSERT INTO Users (username, password_hash, name, dob) VALUES (%s, %s, %s, %s) RETURNING id", (username, password_hash, name, dob))
|
|
user_id = curs.fetchone()[0]
|
|
db.commit()
|
|
return user_id
|
|
|
|
def get_user_from_username(db, username):
|
|
|
|
"""
|
|
Get a user id based on their username
|
|
|
|
Args:
|
|
db (psycopg2.connection): connection to postgres database
|
|
username (str): the username used to query
|
|
|
|
Returns:
|
|
int: 0 if the username is not in the system, user id otherwise
|
|
"""
|
|
|
|
with db:
|
|
with db.cursor() as curs:
|
|
curs.execute("SELECT * FROM Users WHERE username = %s", (username,))
|
|
users = curs.fetchall()
|
|
if not users:
|
|
return 0
|
|
return users[0][0]
|
|
|
|
def get_user_details(db, user_id):
|
|
|
|
"""
|
|
Gets the details of a user based on their id
|
|
|
|
Args:
|
|
db (psycopg2.connection): connection to postgres database
|
|
id (int): user id
|
|
|
|
Returns:
|
|
bool, dict of str to str, dict of str to value: user existance, login info (username, password), user info (name (str), dob (datetime.date))
|
|
"""
|
|
|
|
with db:
|
|
with db.cursor() as curs:
|
|
curs.execute("SELECT * FROM Users WHERE id = %s", (user_id,))
|
|
user = curs.fetchall()
|
|
if not user:
|
|
return False, {}, {}
|
|
user = user[0]
|
|
return True, {"username": user[1], "password_hash": user[2]}, {"name": user[3], "dob": user[4]}
|
|
|
|
def create_event(db, name, time, max_age):
|
|
|
|
"""
|
|
Creates a new event in the database
|
|
|
|
Args:
|
|
db (psycopg2.connection): connection to postgres database
|
|
name (str): event name
|
|
time (date.datetime): event time
|
|
max_age (int): how old a fencer is allowed to be at the time of the event starting
|
|
|
|
Returns:
|
|
int: id of the event created
|
|
"""
|
|
|
|
with db:
|
|
with db.cursor() as curs:
|
|
curs.execute("INSERT INTO Events (name, datetime, max_age, completed, registration, results) VALUES (%s, %s, %s, False, '', '') RETURNING id", (name, time, max_age))
|
|
event_id = curs.fetchone()[0]
|
|
db.commit()
|
|
return event_id
|
|
|
|
def create_tournament(db, events):
|
|
pass
|
|
|
|
def finish_event(db, event):
|
|
pass |