Add comprehensive comments to all source files and expand documentation

This commit is contained in:
Troll (Hermes Agent) 2026-07-31 22:23:40 +00:00
parent 9e532a1920
commit b4f0222bd5
16 changed files with 1230 additions and 552 deletions

View file

@ -1,8 +1,23 @@
"""
models.py
=========
SQLite database layer for the Theme Song Booth.
This module defines the schema and all database operations. Flask's
application context (`g`) is used to manage one connection per request.
Schema overview (see SCHEMA constant):
- requests table stores customer data, generated prompts, file paths,
approval state, email timestamps, payment reference, and player token.
- Indexes on status and player_token for fast queue/lookup.
"""
import sqlite3
import secrets
from datetime import datetime, timezone
from flask import current_app, g
# SQL executed by init_db() to create the requests table and indexes.
SCHEMA = """
CREATE TABLE IF NOT EXISTS requests (
id INTEGER PRIMARY KEY AUTOINCREMENT,
@ -33,30 +48,45 @@ CREATE INDEX IF NOT EXISTS idx_requests_status ON requests(status);
CREATE INDEX IF NOT EXISTS idx_requests_token ON requests(player_token);
"""
def get_db():
"""Get or create a SQLite connection tied to the current Flask request context."""
if 'db' not in g:
g.db = sqlite3.connect(current_app.config['DATABASE'])
g.db.row_factory = sqlite3.Row
return g.db
def close_db(e=None):
"""Close the request-scoped SQLite connection. Registered as teardown handler."""
db = g.pop('db', None)
if db is not None:
db.close()
def init_db():
"""Create the database file and tables. Safe to run multiple times."""
db = sqlite3.connect(current_app.config['DATABASE'])
db.executescript(SCHEMA)
db.commit()
db.close()
def new_token():
"""Generate a URL-safe random token used for private player links."""
return secrets.token_urlsafe(32)
def now_utc():
"""Return current UTC time as ISO-8601 string for timestamp columns."""
return datetime.now(timezone.utc).isoformat()
def create_request(name, email, hobbies, notable_facts, style_genre, extra_requests):
"""
Insert a new customer request.
Returns the auto-generated request id.
"""
db = get_db()
cur = db.execute(
"""INSERT INTO requests
@ -67,17 +97,23 @@ def create_request(name, email, hobbies, notable_facts, style_genre, extra_reque
db.commit()
return cur.lastrowid
def get_request_by_id(request_id):
"""Fetch one request by numeric id. Returns dict or None."""
db = get_db()
row = db.execute('SELECT * FROM requests WHERE id = ?', (request_id,)).fetchone()
return dict(row) if row else None
def get_request_by_token(token):
"""Fetch one request by its private player token. Returns dict or None."""
db = get_db()
row = db.execute('SELECT * FROM requests WHERE player_token = ?', (token,)).fetchone()
return dict(row) if row else None
def list_requests(status=None):
"""List all requests, optionally filtered by status, newest first."""
db = get_db()
if status:
rows = db.execute('SELECT * FROM requests WHERE status = ? ORDER BY created_at DESC', (status,)).fetchall()
@ -85,7 +121,12 @@ def list_requests(status=None):
rows = db.execute('SELECT * FROM requests ORDER BY created_at DESC').fetchall()
return [dict(r) for r in rows]
def update_request(request_id, **fields):
"""
Update arbitrary columns for a request.
Example: update_request(1, status='prompt_ready', suno_style='...')
"""
if not fields:
return
db = get_db()
@ -94,12 +135,16 @@ def update_request(request_id, **fields):
db.execute(f'UPDATE requests SET {cols} WHERE id = ?', vals)
db.commit()
def delete_request(request_id):
"""Delete a single request by id. Does NOT delete associated files."""
db = get_db()
db.execute('DELETE FROM requests WHERE id = ?', (request_id,))
db.commit()
def reset_all_requests():
"""Delete every row in the requests table. Does NOT delete files."""
db = get_db()
db.execute('DELETE FROM requests')
db.commit()