41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
from datetime import UTC, datetime
|
|
from typing import TYPE_CHECKING
|
|
|
|
from app.models.model import UTCBaseModel
|
|
|
|
from sqlmodel import (
|
|
BigInteger,
|
|
Column,
|
|
DateTime,
|
|
Field,
|
|
ForeignKey,
|
|
Relationship,
|
|
SQLModel,
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
from .lazer_user import User
|
|
|
|
|
|
class UserAchievementBase(SQLModel, UTCBaseModel):
|
|
achievement_id: int = Field(primary_key=True)
|
|
achieved_at: datetime = Field(
|
|
default=datetime.now(UTC), sa_column=Column(DateTime(timezone=True))
|
|
)
|
|
|
|
|
|
class UserAchievement(UserAchievementBase, table=True):
|
|
__tablename__ = "lazer_user_achievements" # pyright: ignore[reportAssignmentType]
|
|
|
|
id: int | None = Field(default=None, primary_key=True, index=True)
|
|
user_id: int = Field(
|
|
sa_column=Column(BigInteger, ForeignKey("lazer_users.id")), exclude=True
|
|
)
|
|
user: "User" = Relationship(back_populates="achievement")
|
|
|
|
|
|
class UserAchievementResp(UserAchievementBase):
|
|
@classmethod
|
|
def from_db(cls, db_model: UserAchievement) -> "UserAchievementResp":
|
|
return cls.model_validate(db_model)
|