ahtlon
22e57996c2
- merge card.py, user.py and dbmodels.py into models.py - deduplicate - change responsemodels in the manager functions - correct inports
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
from sqlmodel import Field, Relationship, Session, SQLModel
|
|
from typing import List
|
|
|
|
class Base(SQLModel):
|
|
pass
|
|
|
|
#### User
|
|
class UserBase(Base):
|
|
name: str = Field(index=True)
|
|
email: str | None = None
|
|
is_admin: bool = False
|
|
|
|
class UserResponse(UserBase):
|
|
id: int
|
|
|
|
class UserCreate(UserBase):
|
|
password: str
|
|
|
|
class UserDB(UserBase, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
password: str
|
|
|
|
#### Special
|
|
class AaGroupLink(Base, table=True):
|
|
group_id: int | None = Field(default=None, foreign_key="group.id", primary_key=True)
|
|
accessauth_id: int | None = Field(default=None, foreign_key="accessauthorization.id", primary_key=True)
|
|
|
|
#### Group
|
|
class GroupBase(Base):
|
|
name: str = Field(index=True)
|
|
|
|
|
|
class GroupDB(GroupBase, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
cards: List["Card"] = Relationship(back_populates="group")
|
|
accessauths: List["AccessAuthorizationBase"] = Relationship(back_populates="groups", link_model=AaGroupLink)
|
|
|
|
class GroupResponse(GroupBase):
|
|
id: int
|
|
cards: List["Card"]
|
|
accessauths: List["AccessAuthorizationBase"]
|
|
|
|
#### AccessAuthorization
|
|
class AccessAuthorizationBase(Base):
|
|
name: str = Field(index=True)
|
|
is_active: bool
|
|
groups: List["GroupDB"] = Relationship(back_populates="accessauths", link_model=AaGroupLink)
|
|
timetables: List["Timetable"] = Relationship(back_populates="accessauth")
|
|
|
|
class AccessAuthorizationDB(Base, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
|
|
class AccessAuthorizationCreate(AccessAuthorizationBase):
|
|
pass
|
|
|
|
class AccessAuthorizationResponse(AccessAuthorizationBase):
|
|
id: int
|
|
|
|
#### Card
|
|
class Card(Base, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
uuid: str
|
|
group_id: int | None = Field(default=None, foreign_key="groupdb.id")
|
|
group: GroupDB | None = Relationship(back_populates="cards")
|
|
|
|
class Timetable(Base, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
weekday: int
|
|
starttime: str
|
|
duration: int
|
|
accessauth_id: int | None = Field(default=None, foreign_key="accessauthorizationdb.id")
|
|
accessauth: AccessAuthorizationDB | None = Relationship(back_populates="timetables") |