horror/backend/app/watched.py

"""Bereits geschaute Filme und ihre Bewertungen.

Der Host trägt ein, was am Abend lief. Bewerten darf jeder, eine Stimme
je Sitzung — dieselbe Cookie-Sitzung wie bei den Vorschlägen. Der Name ist
freiwillig, ohne ihn zählt die Stimme trotzdem.

Titel und Poster werden mitgespeichert, damit ein Eintrag lesbar bleibt,
auch wenn der Film später aus dem Katalog fällt.
"""

from datetime import datetime, timezone

from . import users
from .db import connect
from .users import WEM, besitzer, gehoert_mir

SCHEMA = """
CREATE TABLE IF NOT EXISTS watched (
    id          INTEGER PRIMARY KEY AUTOINCREMENT,
    movie_id    INTEGER NOT NULL UNIQUE,
    title       TEXT NOT NULL,
    year        INTEGER,
    poster_path TEXT,
    watched_at  TEXT NOT NULL,
    hidden      INTEGER NOT NULL DEFAULT 0
);
"""

RATINGS = """
CREATE TABLE IF NOT EXISTS ratings (
    id         INTEGER PRIMARY KEY AUTOINCREMENT,
    watched_id INTEGER NOT NULL,
    session_id TEXT NOT NULL,
    user_id    INTEGER,
    stars      INTEGER NOT NULL,
    name       TEXT,
    created_at TEXT NOT NULL
);
"""

MAX_STERNE = 5


def init():
    with connect() as conn:
        conn.executescript(SCHEMA + RATINGS)
        # Eine Stimme je Person und Film
        users.je_person(conn, "ratings", RATINGS, "watched_id")
        # Die Tabelle stand vor der Spalte schon in bestehenden Datenbanken
        spalten = {r["name"] for r in conn.execute("PRAGMA table_info(watched)")}
        if "hidden" not in spalten:
            conn.execute(
                "ALTER TABLE watched ADD COLUMN hidden INTEGER NOT NULL DEFAULT 0"
            )


def _jetzt():
    return datetime.now(timezone.utc).isoformat(timespec="seconds")


def alle(session_id):
    """Zuletzt geschaut zuerst, mit Schnitt und der eigenen Stimme."""
    with connect() as conn:
        filme = conn.execute(
            "SELECT * FROM watched ORDER BY watched_at DESC, id DESC"
        ).fetchall()
        stimmen = conn.execute(
            "SELECT id, watched_id, session_id, user_id, stars, name FROM ratings "
            "ORDER BY created_at ASC"
        ).fetchall()

    ich = besitzer(session_id)
    je_film = {}
    for s in stimmen:
        je_film.setdefault(s["watched_id"], []).append(s)

    out = []
    for film in filme:
        eigene = je_film.get(film["id"], [])
        werte = [s["stars"] for s in eigene]
        out.append(
            {
                **dict(film),
                "ratings": [
                    {"id": s["id"], "stars": s["stars"], "name": s["name"]}
                    for s in eigene
                ],
                "average": round(sum(werte) / len(werte), 1) if werte else None,
                "count": len(werte),
                "mine": next(
                    (
                        s["stars"]
                        for s in eigene
                        if gehoert_mir(s, session_id, ich)
                    ),
                    None,
                ),
            }
        )
    return out


def markieren(movie):
    """Film als geschaut eintragen. Ein zweiter Klick ändert nichts."""
    with connect() as conn:
        conn.execute(
            "INSERT OR IGNORE INTO watched "
            "(movie_id, title, year, poster_path, watched_at) "
            "VALUES (?, ?, ?, ?, ?)",
            (
                movie["id"],
                movie["title"],
                movie.get("year"),
                movie.get("poster_path"),
                _jetzt(),
            ),
        )
        row = conn.execute(
            "SELECT * FROM watched WHERE movie_id = ?", (movie["id"],)
        ).fetchone()
    return dict(row)


def entfernen(watched_id):
    with connect() as conn:
        conn.execute("DELETE FROM ratings WHERE watched_id = ?", (watched_id,))
        return conn.execute(
            "DELETE FROM watched WHERE id = ?", (watched_id,)
        ).rowcount


def datum_setzen(watched_id, tag):
    """tag im Format JJJJ-MM-TT. Mittags in UTC, damit die Anzeige in
    keiner Zeitzone auf den Nachbartag rutscht."""
    with connect() as conn:
        return conn.execute(
            "UPDATE watched SET watched_at = ? WHERE id = ?",
            (f"{tag}T12:00:00+00:00", watched_id),
        ).rowcount


def sichtbarkeit(watched_id, versteckt):
    """Versteckte Filme sieht nur der Host."""
    with connect() as conn:
        return conn.execute(
            "UPDATE watched SET hidden = ? WHERE id = ?",
            (1 if versteckt else 0, watched_id),
        ).rowcount


def bewerten(watched_id, session_id, stars, name=None):
    """Gibt None zurück, wenn der Film gar nicht in der Liste steht."""
    with connect() as conn:
        da = conn.execute(
            "SELECT 1 FROM watched WHERE id = ?", (watched_id,)
        ).fetchone()
        if not da:
            return None
        # Eine Stimme je Person — die eigene alte fliegt vorher raus
        ich = besitzer(session_id)
        conn.execute(
            f"DELETE FROM ratings WHERE watched_id = ? AND {WEM}",
            (watched_id, ich, ich, session_id),
        )
        conn.execute(
            "INSERT INTO ratings "
            "(watched_id, session_id, user_id, stars, name, created_at) "
            "VALUES (?, ?, ?, ?, ?, ?)",
            (
                watched_id,
                session_id,
                ich,
                stars,
                (name or "").strip()[:40] or None,
                _jetzt(),
            ),
        )
    return {"watched_id": watched_id, "stars": stars}


def stimme_loeschen(rating_id):
    """Der Host räumt eine einzelne Wertung weg."""
    with connect() as conn:
        return conn.execute(
            "DELETE FROM ratings WHERE id = ?", (rating_id,)
        ).rowcount


def ids():
    """movie_ids der geschauten Filme, für die Detailansicht."""
    with connect() as conn:
        return [r["movie_id"] for r in conn.execute("SELECT movie_id FROM watched")]