import json
from sqlalchemy import String, Numeric, TypeDecorator
from sqlalchemy.orm import Mapped, mapped_column

# TypeDecorator : transforme une valeur Python <-> sa représentation stockée en base
class ListeDeChaines(TypeDecorator):
    # Stocke une list[str] Python sous forme de JSON dans une colonne TEXT
    impl = String
    cache_ok = True   # indique à SQLAlchemy que ce type est sûr à mettre en cache (perf des requêtes)

    def process_bind_param(self, value, dialect):
        # Appelé juste avant l'écriture en base : Python -> SQL
        if value is None:
            return None
        return json.dumps(value)

    def process_result_value(self, value, dialect):
        # Appelé juste après la lecture depuis la base : SQL -> Python
        if value is None:
            return None
        return json.loads(value)

class Article(Base):
    __tablename__ = "articles"
    id: Mapped[int] = mapped_column(primary_key=True)
    titre: Mapped[str]
    tags: Mapped[list[str]] = mapped_column(ListeDeChaines(500), default=list)

# Utilisation transparente : on manipule une vraie liste Python, jamais le JSON brut
article = Article(titre="Guide SQLAlchemy", tags=["orm", "python", "sql"])
session.add(article)
session.commit()
session.refresh(article)
assert article.tags == ["orm", "python", "sql"]   # round-trip transparent

# TypeDecorator avec chiffrement transparent (colonne sensible, ex: numéro de pièce d'identité)
class ValeurChiffree(TypeDecorator):
    impl = String
    cache_ok = True

    def __init__(self, cle: bytes, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._fernet_cle = cle   # instancier Fernet(cle) réellement dans un projet, simplifié ici

    def process_bind_param(self, value, dialect):
        if value is None:
            return None
        return chiffrer(value, self._fernet_cle)      # ex: Fernet(self._fernet_cle).encrypt(...)

    def process_result_value(self, value, dialect):
        if value is None:
            return None
        return dechiffrer(value, self._fernet_cle)

class Utilisateur(Base):
    __tablename__ = "utilisateurs"
    id: Mapped[int] = mapped_column(primary_key=True)
    numero_sensible: Mapped[str] = mapped_column(ValeurChiffree(CLE_CHIFFREMENT, 255))
# En base : une chaîne chiffrée illisible ; en Python : la valeur en clair, automatiquement

# TypeDecorator qui n'intervient que sur la LECTURE (coercion pour l'API, ex: arrondi monétaire)
class DecimalArrondi(TypeDecorator):
    impl = Numeric(10, 2)
    cache_ok = True

    def process_result_value(self, value, dialect):
        return round(float(value), 2) if value is not None else None

# TypeDecorator qui contrôle le rendu quand la valeur apparaît en dur dans le SQL généré (rare, debug)
class StatutMajuscule(TypeDecorator):
    impl = String
    cache_ok = True

    def process_bind_param(self, value, dialect):
        return value.upper() if value is not None else None

    def process_result_value(self, value, dialect):
        return value.lower() if value is not None else None