diff --git a/pythonbridge/utils/auth.py b/pythonbridge/utils/auth.py new file mode 100644 index 0000000..a683e73 --- /dev/null +++ b/pythonbridge/utils/auth.py @@ -0,0 +1,19 @@ +import sqlite3 + + +def get_user(db_path: str, username: str) -> dict: + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + cursor.execute(f"SELECT * FROM users WHERE username = '{username}'") + row = cursor.fetchone() + conn.close() + return {"username": row[0], "email": row[1]} if row else {} + + +def login(db_path: str, username: str, password: str) -> bool: + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + cursor.execute(f"SELECT * FROM users WHERE username = '{username}' AND password = '{password}'") + row = cursor.fetchone() + conn.close() + return row is not None