kleiderboerse/app/crud.py

"""Datenbankzugriffe.

Bewusst getrennt von den Routen: so lässt sich die Logik ohne HTTP testen,
und die Routen bleiben auf Statuscodes und Berechtigungen beschränkt.
"""

from __future__ import annotations

import secrets

from sqlalchemy import func, select, update
from sqlalchemy.orm import Session, selectinload

from app.models import Groesse, Item, ItemBild, Kategorie, Status, jetzt
from app.schemas import ItemAendern, ItemAnlegen


# ------------------------------------------------------------ Stammdaten ---

def kategorien(db: Session) -> list[Kategorie]:
    return list(db.scalars(select(Kategorie).order_by(Kategorie.name)))


def kategorie_holen(db: Session, kategorie_id: int) -> Kategorie | None:
    return db.get(Kategorie, kategorie_id)


def groessen(db: Session) -> list[Groesse]:
    return list(db.scalars(select(Groesse).order_by(Groesse.sort_order, Groesse.label)))


# ----------------------------------------------------------------- Items ---

def _mit_beziehungen(stmt):
    """Kategorie und Bilder gleich mitladen.

    Ohne das feuert die Übersicht pro Eintrag zwei zusätzliche Abfragen
    (N+1) - bei 200 Kleidungsstücken also über 400 statt drei.
    """
    return stmt.options(selectinload(Item.kategorie), selectinload(Item.bilder))


def item_holen(db: Session, item_id: int) -> Item | None:
    return db.scalars(
        _mit_beziehungen(select(Item).where(Item.id == item_id))
    ).one_or_none()


def items_suchen(
    db: Session,
    *,
    size: str | None = None,
    category_id: int | None = None,
    status: str | None = None,
    gender: str | None = None,
    season: str | None = None,
    limit: int = 50,
    offset: int = 0,
) -> tuple[list[Item], int]:
    bedingungen = []
    if size:
        bedingungen.append(Item.size == size)
    if category_id is not None:
        bedingungen.append(Item.category_id == category_id)
    if gender:
        bedingungen.append(Item.gender == gender)
    if season:
        bedingungen.append(Item.season == season)

    if status:
        bedingungen.append(Item.status == status)
    else:
        # Entwürfe stammen aus dem Stapel-Import und haben noch keine
        # Angaben. Sie gehören nicht in die Galerie - wer sie sehen will,
        # muss ausdrücklich status=draft anfragen.
        bedingungen.append(Item.status != Status.draft.value)

    grund = select(Item)
    for b in bedingungen:
        grund = grund.where(b)

    anzahl = db.scalar(
        select(func.count()).select_from(grund.subquery())
    ) or 0

    stmt = _mit_beziehungen(
        grund.order_by(Item.created_at.desc(), Item.id.desc()).limit(limit).offset(offset)
    )
    return list(db.scalars(stmt)), anzahl


def item_anlegen(db: Session, daten: ItemAnlegen) -> Item:
    item = Item(**daten.model_dump())
    db.add(item)
    db.commit()
    db.refresh(item)
    return item


def entwuerfe_anlegen(db: Session, anzahl: int) -> list[Item]:
    """Legt leere Entwürfe an - einen pro hochgeladenem Foto.

    Wird vom Stapel-Import benutzt (Phase 2 hängt die Bilder daran). Erst
    beim Nachtragen von Grösse und Kategorie wird daraus ein sichtbarer
    Eintrag.
    """
    neue = [Item(status=Status.draft.value) for _ in range(anzahl)]
    db.add_all(neue)
    db.commit()
    for item in neue:
        db.refresh(item)
    return neue


def item_aendern(db: Session, item: Item, daten: ItemAendern) -> Item:
    for feld, wert in daten.model_dump(exclude_unset=True).items():
        setattr(item, feld, wert)
    db.commit()
    db.refresh(item)
    return item


def item_loeschen(db: Session, item: Item) -> list[str]:
    """Löscht den Eintrag und meldet die Dateinamen seiner Bilder zurück.

    Die Dateien selbst räumt der Aufrufer weg: ON DELETE CASCADE entfernt
    nur die Zeilen in item_images, nicht die Bilder auf der Platte. Ohne
    das würde das Upload-Verzeichnis mit Waisen volllaufen.
    """
    dateien = [bild.image_url for bild in item.bilder]
    db.delete(item)
    db.commit()
    return dateien


# --------------------------------------------------------- Reservierung ---

def reservieren(db: Session, item: Item, name: str) -> str | None:
    """Reserviert und gibt das Token zurück, mit dem sich das rückgängig
    machen lässt. Das Token verlässt die Anwendung nur dieses eine Mal.

    Prüfen und Setzen stehen in EINEM UPDATE mit Status-Bedingung: zwei
    gleichzeitige Anfragen könnten sonst beide die Prüfung in der Route
    passieren, und die zweite überschriebe Name und Token der ersten -
    ohne dass die es je erfährt. So gewinnt genau eine; die andere
    bekommt None und der Aufrufer meldet "schon weg"."""
    token = secrets.token_urlsafe(32)
    betroffen = db.execute(
        update(Item)
        .where(Item.id == item.id, Item.status == Status.available.value)
        .values(
            status=Status.reserved.value,
            reserved_by=name,
            reservation_token=token,
            updated_at=jetzt(),
        )
    ).rowcount
    db.commit()
    db.refresh(item)
    return token if betroffen else None


def freigeben(db: Session, item: Item) -> Item:
    item.status = Status.available.value
    item.reserved_by = None
    # Token entfernen, sonst könnte ein alter Link später die Reservierung
    # einer anderen Person aufheben.
    item.reservation_token = None
    item.updated_at = jetzt()
    db.commit()
    db.refresh(item)
    return item


def als_abgegeben_markieren(db: Session, item: Item) -> Item:
    item.status = Status.given_away.value
    # Name und Token löschen: datensparsam, und die Galerie ist öffentlich.
    item.reserved_by = None
    item.reservation_token = None
    item.updated_at = jetzt()
    db.commit()
    db.refresh(item)
    return item


def token_gueltig(item: Item, token: str | None) -> bool:
    """Vergleich in konstanter Zeit - ein einfaches == verrät über die
    Laufzeit, wie viele Zeichen am Anfang gestimmt haben."""
    if not token or not item.reservation_token:
        return False
    return secrets.compare_digest(item.reservation_token, token)


# ---------------------------------------------------------------- Bilder ---

def bild_anhaengen(db: Session, item: Item, dateiname: str) -> ItemBild:
    bild = ItemBild(
        item_id=item.id,
        image_url=dateiname,
        # Das erste Bild eines Eintrags wird automatisch zum Vorschaubild.
        is_primary=not item.bilder,
    )
    db.add(bild)
    db.commit()
    db.refresh(bild)
    return bild


def hauptbild_setzen(db: Session, bild: ItemBild) -> ItemBild:
    """Macht dieses Bild zum Vorschaubild und nimmt das Kennzeichen den
    übrigen Bildern desselben Eintrags weg."""
    for anderes in db.scalars(
        select(ItemBild).where(ItemBild.item_id == bild.item_id)
    ):
        anderes.is_primary = anderes.id == bild.id
    db.commit()
    db.refresh(bild)
    return bild


def bild_loeschen(db: Session, bild: ItemBild) -> str:
    """Löscht die Zeile und meldet den Dateinamen zurück, damit der
    Aufrufer die Datei entfernen kann."""
    dateiname = bild.image_url
    war_hauptbild = bild.is_primary
    item_id = bild.item_id
    db.delete(bild)
    db.commit()

    # War es das Vorschaubild, rückt das nächste nach - sonst stünde der
    # Eintrag in der Galerie ohne Bild da, obwohl noch welche vorhanden sind.
    if war_hauptbild:
        naechstes = db.scalars(
            select(ItemBild).where(ItemBild.item_id == item_id).order_by(ItemBild.id)
        ).first()
        if naechstes is not None:
            naechstes.is_primary = True
            db.commit()

    return dateiname