diff --git a/app/models.py b/app/models.py index ca42b27..2f0ad6e 100644 --- a/app/models.py +++ b/app/models.py @@ -5,6 +5,7 @@ from enum import Enum from typing import Optional from pydantic import BaseModel +from app.database import LazerUserAchievement # 添加数据库模型导入 class GameMode(str, Enum): @@ -83,6 +84,14 @@ class UserAchievement(BaseModel): achieved_at: datetime achievement_id: int + # 添加数据库模型转换方法 + def to_db_model(self, user_id: int) -> LazerUserAchievement: + return LazerUserAchievement( + user_id=user_id, + achievement_id=self.achievement_id, + achieved_at=self.achieved_at + ) + class RankHighest(BaseModel): rank: int