organization

This commit is contained in:
2025-11-27 22:52:14 -08:00
parent 4584b1f58f
commit 61f94e53ca

View File

@@ -1,101 +1,111 @@
import bcrypt import bcrypt
import datetime import datetime
def create_user(db, login_info, user_info): class Users:
""" @staticmethod
Creates a new user in the database def create_user(db, login_info, user_info):
Args: """
db (psycopg2.connection): connection to postgres database Creates a new user in the 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: Args:
int: 0 if the username is non-unique, user id otherwise 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
username, password = login_info["username"], login_info["password"] Returns:
password_hash = bcrypt.hashpw(password.encode(), bcrypt.gensalt()).decode() int: 0 if the username is non-unique, user id otherwise
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): 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
""" @staticmethod
Get a user id based on their username def get_user_from_username(db, username):
Args: """
db (psycopg2.connection): connection to postgres database Get a user id based on their username
username (str): the username used to query
Returns: Args:
int: 0 if the username is not in the system, user id otherwise db (psycopg2.connection): connection to postgres database
""" username (str): the username used to query
with db: Returns:
with db.cursor() as curs: int: 0 if the username is not in the system, user id otherwise
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): 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]
""" @staticmethod
Gets the details of a user based on their id def get_user_details(db, user_id):
Args: """
db (psycopg2.connection): connection to postgres database Gets the details of a user based on their id
id (int): user id
Returns: Args:
bool, dict of str to str, dict of str to value: user existance, login info (username, password), user info (name (str), dob (datetime.date)) db (psycopg2.connection): connection to postgres database
""" id (int): user id
with db: Returns:
with db.cursor() as curs: bool, dict of str to str, dict of str to value: user existance, login info (username, password), user info (name (str), dob (datetime.date))
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): 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]}
""" class Tournaments:
Creates a new event in the database
Args: @staticmethod
db (psycopg2.connection): connection to postgres database def create_event(db, name, time, max_age):
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 Creates a new event in the database
"""
with db: Args:
with db.cursor() as curs: db (psycopg2.connection): connection to postgres database
curs.execute("INSERT INTO Events (name, datetime, max_age, completed, registration, results) VALUES (%s, %s, %s, False, '', '') RETURNING id", (name, time, max_age)) name (str): event name
event_id = curs.fetchone()[0] time (date.datetime): event time
db.commit() max_age (int): how old a fencer is allowed to be at the time of the event starting
return event_id
def create_tournament(db, events): Returns:
pass int: id of the event created
"""
def finish_event(db, event): with db:
pass 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
@staticmethod
def create_tournament(db, events):
pass
@staticmethod
def finish_event(db, event):
pass