Added series model
This commit is contained in:
3
.idea/league-manager.iml
generated
3
.idea/league-manager.iml
generated
@@ -3,8 +3,9 @@
|
|||||||
<component name="NewModuleRootManager">
|
<component name="NewModuleRootManager">
|
||||||
<content url="file://$MODULE_DIR$">
|
<content url="file://$MODULE_DIR$">
|
||||||
<sourceFolder url="file://$MODULE_DIR$/backend" isTestSource="false" />
|
<sourceFolder url="file://$MODULE_DIR$/backend" isTestSource="false" />
|
||||||
|
<excludeFolder url="file://$MODULE_DIR$/.venv" />
|
||||||
</content>
|
</content>
|
||||||
<orderEntry type="jdk" jdkName="Python 3.13" jdkType="Python SDK" />
|
<orderEntry type="jdk" jdkName="Python 3.14 (RSW_puntensysteem)" jdkType="Python SDK" />
|
||||||
<orderEntry type="sourceFolder" forTests="false" />
|
<orderEntry type="sourceFolder" forTests="false" />
|
||||||
</component>
|
</component>
|
||||||
<component name="PyDocumentationSettings">
|
<component name="PyDocumentationSettings">
|
||||||
|
|||||||
2
.idea/misc.xml
generated
2
.idea/misc.xml
generated
@@ -3,5 +3,5 @@
|
|||||||
<component name="Black">
|
<component name="Black">
|
||||||
<option name="sdkName" value="Python 3.13" />
|
<option name="sdkName" value="Python 3.13" />
|
||||||
</component>
|
</component>
|
||||||
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.13" project-jdk-type="Python SDK" />
|
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.14 (RSW_puntensysteem)" project-jdk-type="Python SDK" />
|
||||||
</project>
|
</project>
|
||||||
173
backend/app/models/serie.py
Normal file
173
backend/app/models/serie.py
Normal file
@@ -0,0 +1,173 @@
|
|||||||
|
# app/models/serie.py
|
||||||
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
|
from sqlmodel import Field, Relationship, Session, select
|
||||||
|
|
||||||
|
from . import mixin
|
||||||
|
from .base import BaseSQLModel, RowId
|
||||||
|
from .user import PermissionRight, User
|
||||||
|
|
||||||
|
if TYPE_CHECKING:
|
||||||
|
from .team import Team
|
||||||
|
|
||||||
|
# region # Serie ##############################################################
|
||||||
|
|
||||||
|
# Shared properties
|
||||||
|
class SerieUserLinkBase(BaseSQLModel):
|
||||||
|
rights: PermissionRight = Field(default=PermissionRight.READ, nullable=False)
|
||||||
|
|
||||||
|
# Properties to receive via API on creation
|
||||||
|
class SerieUserLinkCreate(SerieUserLinkBase):
|
||||||
|
user_id: RowId = Field(nullable=False)
|
||||||
|
|
||||||
|
# Properties to receive via API on update, all are optional
|
||||||
|
class SerieUserLinkUpdate(SerieUserLinkBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Database model (link tussen serie en user)
|
||||||
|
class SerieUserLink(SerieUserLinkBase, table=True):
|
||||||
|
serie_id: RowId = Field(
|
||||||
|
foreign_key="serie.id",
|
||||||
|
primary_key=True,
|
||||||
|
nullable=False,
|
||||||
|
ondelete="CASCADE",
|
||||||
|
)
|
||||||
|
user_id: RowId = Field(
|
||||||
|
foreign_key="user.id",
|
||||||
|
primary_key=True,
|
||||||
|
nullable=False,
|
||||||
|
ondelete="CASCADE",
|
||||||
|
)
|
||||||
|
|
||||||
|
serie: "Serie" = Relationship(back_populates="user_links")
|
||||||
|
user: "User" = Relationship(back_populates="serie_links")
|
||||||
|
|
||||||
|
# Properties to return via API
|
||||||
|
class SerieUserLinkPublic(SerieUserLinkBase):
|
||||||
|
user_id: RowId
|
||||||
|
serie_id: RowId
|
||||||
|
|
||||||
|
class SerieUserLinksPublic(BaseSQLModel):
|
||||||
|
data: list[SerieUserLinkPublic]
|
||||||
|
count: int
|
||||||
|
|
||||||
|
# Shared properties
|
||||||
|
class SerieBase(
|
||||||
|
mixin.Name,
|
||||||
|
mixin.Contact,
|
||||||
|
mixin.IsActive,
|
||||||
|
BaseSQLModel,
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Properties to receive via API on creation
|
||||||
|
class SerieCreate(SerieBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Properties to receive via API on update, all are optional
|
||||||
|
class SerieUpdate(SerieBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Database model
|
||||||
|
class Serie(mixin.RowId, SerieBase, table=True):
|
||||||
|
|
||||||
|
user_links: list["SerieUserLink"] = Relationship(back_populates="serie", cascade_delete=True)
|
||||||
|
#teams: list["Team"] = Relationship(back_populates="serie", cascade_delete=True)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def create(cls, *, session: Session, create_obj: SerieCreate) -> "Serie":
|
||||||
|
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: "Serie", in_obj: SerieUpdate) -> "Serie":
|
||||||
|
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 get_user_link(self, user: User) -> "SerieUserLink | None":
|
||||||
|
return next((link for link in self.user_links if link.user == user), None)
|
||||||
|
|
||||||
|
def add_user(
|
||||||
|
self,
|
||||||
|
user: User,
|
||||||
|
rights: PermissionRight = PermissionRight.READ,
|
||||||
|
*,
|
||||||
|
session: Session,
|
||||||
|
) -> "SerieUserLink | None":
|
||||||
|
existing = self.get_user_link(user=user)
|
||||||
|
if existing is None:
|
||||||
|
new_link = SerieUserLink(serie=self, user=user, rights=rights)
|
||||||
|
self.user_links.append(new_link)
|
||||||
|
session.add(new_link)
|
||||||
|
session.commit()
|
||||||
|
return new_link
|
||||||
|
return None #TODO aanpassen??
|
||||||
|
|
||||||
|
def update_user(
|
||||||
|
self,
|
||||||
|
user: User,
|
||||||
|
rights: PermissionRight = PermissionRight.READ,
|
||||||
|
*,
|
||||||
|
session: Session,
|
||||||
|
) -> "SerieUserLink | None":
|
||||||
|
link = self.get_user_link(user=user)
|
||||||
|
if link:
|
||||||
|
link.rights = rights
|
||||||
|
session.add(link)
|
||||||
|
session.commit()
|
||||||
|
return link
|
||||||
|
return None
|
||||||
|
|
||||||
|
def remove_user(self, user: User, *, session: Session) -> None:
|
||||||
|
link = self.get_user_link(user=user)
|
||||||
|
if link:
|
||||||
|
session.delete(link)
|
||||||
|
session.commit()
|
||||||
|
|
||||||
|
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
|
||||||
|
)
|
||||||
|
|
||||||
|
def user_has_right(
|
||||||
|
self,
|
||||||
|
user: User,
|
||||||
|
rights: PermissionRight | None = None,
|
||||||
|
) -> bool:
|
||||||
|
return any(
|
||||||
|
(
|
||||||
|
link.user == user
|
||||||
|
and link.rights
|
||||||
|
and (not rights or (link.rights & rights))
|
||||||
|
)
|
||||||
|
for link in self.user_links
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# API output models
|
||||||
|
class SeriePublic(mixin.RowIdPublic, SerieBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class SeriesPublic(BaseSQLModel):
|
||||||
|
data: list[SeriePublic]
|
||||||
|
count: int
|
||||||
|
|
||||||
|
# endregion
|
||||||
Reference in New Issue
Block a user