feat: work shifts by QR codes
This commit is contained in:
@@ -5,6 +5,7 @@ from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from enums.user import UserRole
|
||||
from models.base import BaseModel
|
||||
from models.work_shifts import WorkShift
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from models.payroll import PayRate, PaymentRecord
|
||||
@@ -91,6 +92,12 @@ class User(BaseModel):
|
||||
uselist=True,
|
||||
foreign_keys="PaymentRecord.user_id"
|
||||
)
|
||||
work_shifts: Mapped[list["WorkShift"]] = relationship(
|
||||
"WorkShift",
|
||||
back_populates="user",
|
||||
uselist=True,
|
||||
foreign_keys="WorkShift.user_id"
|
||||
)
|
||||
|
||||
|
||||
class Position(BaseModel):
|
||||
|
||||
30
models/work_shifts.py
Normal file
30
models/work_shifts.py
Normal file
@@ -0,0 +1,30 @@
|
||||
from datetime import datetime
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from models.base import BaseModel
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from models import User
|
||||
|
||||
|
||||
class WorkShift(BaseModel):
|
||||
__tablename__ = "work_shifts"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
started_at: Mapped[datetime] = mapped_column(
|
||||
nullable=False,
|
||||
)
|
||||
finished_at: Mapped[datetime] = mapped_column(
|
||||
nullable=True,
|
||||
)
|
||||
|
||||
user_id: Mapped[int] = mapped_column(
|
||||
ForeignKey("users.id"),
|
||||
)
|
||||
user: Mapped["User"] = relationship(
|
||||
"User",
|
||||
back_populates="work_shifts",
|
||||
)
|
||||
Reference in New Issue
Block a user