models.py

import random
import string
from datetime import datetime
from typing import Optional
from pydantic import BaseModel
from typing_extensions import override


# Modelos - Version de FastAPI

# Helper function to generate unique IDs
def generate_id():
    return ''.join(random.choices(string.ascii_uppercase + string.digits, k=5))

# Models
class EntradaLista(BaseModel):
    entrada_id: str
    texto: str

class Autor(BaseModel):
    autor_id: str
    nombre: str

class Lista(BaseModel):
    lista_id: str
    nombre: str
    _entradas: dict[str, EntradaLista] = {}
    autor: Optional[Autor] = None

    def add_entrada(self, entrada: EntradaLista):
        if entrada.entrada_id not in self._entradas:
            self._entradas[entrada.entrada_id] = entrada

    def get_entradas(self) -> list[EntradaLista]:
        return list(self._entradas.values())

    def get_entrada(self, entrada_id: str) -> EntradaLista:
        return self._entradas.get(entrada_id)

    def delete_entrada(self, entrada_id: str) -> bool:
        if entrada_id in self._entradas:
            del self._entradas[entrada_id]
            return True
        return False

class ListaCompra(Lista):
    _cantidades: dict[str, int] = {}

    def get_cantidad(self, entrada_id: str) -> int:
        return self._cantidades.get(entrada_id, 0)

    @override
    def add_entrada(self, entrada: EntradaLista, cantidad: int = 1):
        super().add_entrada(entrada)
        self._cantidades[entrada.entrada_id] = cantidad

class ListaTrabajo(Lista):
    fecha_limite: Optional[datetime] = None


class TodoList:
    def __init__(self):
        self._listas: dict[str, Lista] = {}
        self._autores: dict[str, Autor] = {}

    def add_lista(self, lista: Lista):
        if lista.lista_id not in self._listas:
            self._listas[lista.lista_id] = lista

    def get_lista(self, lista_id: str) -> Lista:
        return self._listas.get(lista_id)

    def delete_lista(self, lista_id: str):
        if lista_id in self._listas:
            del self._listas[lista_id]

    def add_autor(self, autor: Autor):
        if autor.autor_id not in self._autores:
            self._autores[autor.autor_id] = autor

    def get_autor(self, autor_id: str) -> Autor:
        return self._autores.get(autor_id)

    def delete_autor(self, autor_id: str) -> bool:
        if autor_id in self._autores:
            del self._autores[autor_id]
            return True
        return False

    def clear_listas(self):
        self._listas.clear()

    def get_listas(self) -> list[Lista]:
        return list(self._listas.values())

    def get_listas_compra(self) -> list[ListaCompra]:
        return [l for l in self._listas.values() if isinstance(l, ListaCompra)]

    def get_listas_trabajo(self) -> list[ListaTrabajo]:
        return [l for l in self._listas.values() if isinstance(l, ListaTrabajo)]

    def clear_autores(self):
        self._autores.clear()

    def get_autores(self) -> list[Autor]:
        return list(self._autores.values())

    def get_listas_by_autor(self, autor_id: str) -> list[Lista]:
        return [l for l in self._listas.values() if l.autor and l.autor.autor_id == autor_id]

    def delete_listas_by_autor(self, autor_id: str):
        for l in self.get_listas_by_autor(autor_id):
            self.delete_lista(l.lista_id)