src/auth/db.py : CRUD tokens de vérification et reset (#73)

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Colin Maudry
2026-04-20 17:23:02 +02:00
parent ae4a206de7
commit 1a07e5ac0f
2 changed files with 137 additions and 0 deletions
+61
View File
@@ -110,3 +110,64 @@ def update_password_hash(user_id: int, password_hash: str) -> None:
def delete_user(user_id: int) -> None:
get_conn().execute("DELETE FROM users WHERE id = ?", (user_id,))
def create_email_verification_token(
token_hash: str, user_id: int, expires_at: str
) -> None:
get_conn().execute(
"INSERT INTO email_verification_tokens (token_hash, user_id, expires_at, created_at) "
"VALUES (?, ?, ?, ?)",
(token_hash, user_id, expires_at, _now()),
)
def find_email_verification_token(token_hash: str) -> sqlite3.Row | None:
return (
get_conn()
.execute(
"SELECT * FROM email_verification_tokens "
"WHERE token_hash = ? AND expires_at > ?",
(token_hash, _now()),
)
.fetchone()
)
def delete_email_verification_tokens_for_user(user_id: int) -> None:
get_conn().execute(
"DELETE FROM email_verification_tokens WHERE user_id = ?", (user_id,)
)
def create_password_reset_token(token_hash: str, user_id: int, expires_at: str) -> None:
get_conn().execute(
"INSERT INTO password_reset_tokens (token_hash, user_id, expires_at, created_at) "
"VALUES (?, ?, ?, ?)",
(token_hash, user_id, expires_at, _now()),
)
def find_password_reset_token(token_hash: str) -> sqlite3.Row | None:
return (
get_conn()
.execute(
"SELECT * FROM password_reset_tokens "
"WHERE token_hash = ? AND expires_at > ?",
(token_hash, _now()),
)
.fetchone()
)
def delete_password_reset_tokens_for_user(user_id: int) -> None:
get_conn().execute(
"DELETE FROM password_reset_tokens WHERE user_id = ?", (user_id,)
)
def purge_expired_tokens() -> None:
now = _now()
conn = get_conn()
conn.execute("DELETE FROM email_verification_tokens WHERE expires_at <= ?", (now,))
conn.execute("DELETE FROM password_reset_tokens WHERE expires_at <= ?", (now,))