score/backend/app/models/event.py

from sqlmodel import (
    Field,
    Relationship,
    Session,
    select,
)

from . import mixin
from .base import (
    BaseSQLModel,
    RowId,
)
from .user import (
    PermissionRight,
    User,
)

# region # Event ###############################################################


# Event auth
class EventUserLink(BaseSQLModel, table=True):
    event_id: RowId = Field(
        foreign_key="event.id",
        primary_key=True,
        nullable=False,
        ondelete="CASCADE",
    )

    user_id: RowId = Field(
        foreign_key="user.id",
        primary_key=True,
        nullable=False,
        ondelete="CASCADE",
    )

    rights: PermissionRight = Field(default=PermissionRight.READ, nullable=False)

    event: "Event" = Relationship(back_populates="user_links")
    user: "User" = Relationship(back_populates="event_links")


# ##############################################################################


# Shared properties
class EventBase(
    mixin.Name,
    mixin.Contact,
    mixin.StartEndDate,
    mixin.IsActive,
    mixin.Contact,
    BaseSQLModel,
):
    pass


# Properties to receive via API on creation
class EventCreate(EventBase):
    pass


# Properties to receive via API on update, all are optional
class EventUpdate(EventBase):
    pass


# Database model, database table inferred from class name
class Event(mixin.RowId, EventBase, table=True):
    # --- database only items --------------------------------------------------

    # --- back_populates links -------------------------------------------------

    # --- many-to-many links ---------------------------------------------------
    user_links: list[EventUserLink] = Relationship(back_populates="event")

    # --- CRUD actions ---------------------------------------------------------
    @classmethod
    def create(cls, *, session: Session, create_obj: EventCreate) -> "Event":
        data_obj = create_obj.model_dump(exclude_unset=True)

        db_obj = cls.model_validate(data_obj)
        session.add(db_obj)
        session.commit()
        session.refresh(db_obj)
        return db_obj

    @classmethod
    def update(
        cls, *, session: Session, db_obj: "Event", in_obj: EventUpdate
    ) -> "Event":
        data_obj = in_obj.model_dump(exclude_unset=True)

        db_obj.sqlmodel_update(data_obj)
        session.add(db_obj)
        session.commit()
        session.refresh(db_obj)
        return db_obj

    def add_user(
        self,
        user: User,
        rights: PermissionRight = PermissionRight.READ,
        *,
        session: Session,
    ) -> "Event":
        to_add = next((add for add in self.user_links if add.user == user), None)

        if to_add:
            to_add.rights = rights
            session.add(to_add)
        else:
            self.user_links.append(EventUserLink(event=self, user=user, rights=rights))
            session.add(self.user_links[-1])

        session.commit()

        return self

    def remove_user(self, user: User, *, session: Session) -> "Event":
        to_remove = next(
            (remove for remove in self.user_links if remove.user == user), None
        )
        if to_remove:
            statement = select(EventUserLink).where(
                EventUserLink.event_id == self.id, EventUserLink.user_id == user.id
            )
            link_to_remove = session.exec(statement).first()

            if link_to_remove:
                session.delete(link_to_remove)
                session.commit()

        return self

    def user_has_rights(
        self,
        user: User,
        rights: PermissionRight | None = None,
    ) -> bool:
        return any(
            (
                link.user == user
                and link.rights
                and (not rights or (link.rights & rights) == rights)
            )
            for link in self.user_links
        )


# Properties to return via API, id is always required
class EventPublic(mixin.RowIdPublic, EventBase):
    pass


class EventsPublic(BaseSQLModel):
    data: list[EventPublic]
    count: int


# endregion

# region EventTeam ############################################################


class EventTeamBase(mixin.ThemeName, mixin.CheckInCheckOut, mixin.Canceled, BaseSQLModel):
    # scouting_team_id: RowId | None = Field(
    #     foreign_key="ScoutingTeam.id", nullable=False, ondelete="CASCADE"
    # )
    pass


# Properties to receive via API on creation
class EventTeamCreate(EventTeamBase):
    pass


# Properties to receive via API on update, all are optional
class EventTeamUpdate(mixin.ThemeNameUpdate, EventTeamBase):
    pass


class EventTeam(mixin.RowId, EventTeamBase, table=True):
    # --- database only items --------------------------------------------------

    # --- read only items ------------------------------------------------------
    event_id: RowId = Field(
        foreign_key="event.id", nullable=False, ondelete="CASCADE"
    )

    # --- back_populates links -------------------------------------------------
    event: "Event" = Relationship(back_populates="team_links", cascade_delete=True)
    # team: "ScoutingTeam" = Relationship(back_populates="event_links", cascade_delete=True)

    # --- CRUD actions ---------------------------------------------------------
    @classmethod
    def create(cls, *, session: Session, create_obj: EventTeamCreate, event: Event) -> "EventTeam":
        data_obj = create_obj.model_dump(exclude_unset=True)

        db_obj = cls.model_validate(data_obj, update={"event_id": event.id})
        session.add(db_obj)
        session.commit()
        session.refresh(db_obj)
        return db_obj

    @classmethod
    def update(
        cls, *, session: Session, db_obj: "EventTeam", in_obj: EventTeamUpdate
    ) -> "EventTeam":
        data_obj = in_obj.model_dump(exclude_unset=True)
        db_obj.sqlmodel_update(data_obj)
        session.add(db_obj)
        session.commit()
        session.refresh(db_obj)
        return db_obj


# Properties to return via API, id is always required
class EventTeamPublic(mixin.RowIdPublic, EventTeamBase):
    event_id: RowId


class EventTeamsPublic(BaseSQLModel):
    data: list[EventTeamPublic]
    count: int


# endregion