from sqlalchemy import String, ForeignKey, event
from sqlalchemy.orm import Mapped, mapped_column, declared_attr, relationship, Session

from .base import Base

__all__ = ["Status"]


class Status(Base):
    """Status of a record. Can be used in any table by using MixIn :class:`StatusForeignKey`."""

    id: Mapped[str] = mapped_column(String(3), primary_key=True)
    name: Mapped[str] = mapped_column(String(30), unique=True)


class StatusForeignKey:
    """Foreign Key Mixin for :py:class:`.Status`

    By adding this mixin every record will get a status assigned.
    """

    status_id: Mapped[str] = mapped_column(ForeignKey("status.id"), default="N", sort_order=1000000)

    # noinspection PyMethodParameters
    @declared_attr
    def status(cls) -> Mapped[Status]:
        return relationship()


# noinspection PyUnusedLocal
@event.listens_for(Status.__table__, "after_create")
def initialize_status(target, connection, **kwargs):
    with Session(connection) as session:
        for kwargs in (
                dict(id="A", name="Active"),
                dict(id="I", name="Inactive"),
                dict(id="N", name="New"),
                dict(id="PRE", name="Prepared"),
                dict(id="X", name="eXcluded"),
        ):
            session.add(Status(**kwargs))
        session.commit()