优化数据库模型与关系

This commit is contained in:
2025-09-27 22:40:10 +08:00
parent cf5539f3c2
commit 0d45a07ba7
10 changed files with 81 additions and 23 deletions

View File

@@ -1,7 +1,7 @@
# my_project/models/task.py
from typing import Optional, TYPE_CHECKING
from sqlmodel import Field, Relationship, Column, func, DateTime
from sqlmodel import Field, Relationship, CheckConstraint
from .base import TableBase
from datetime import datetime
@@ -11,6 +11,10 @@ if TYPE_CHECKING:
class Task(TableBase, table=True):
__tablename__ = 'tasks'
__table_args__ = (
CheckConstraint("progress BETWEEN 0 AND 100", name="ck_task_progress_range"),
)
status: int = Field(default=0, sa_column_kwargs={"server_default": "0"}, description="任务状态: 0=排队中, 1=处理中, 2=完成, 3=错误")
type: int = Field(description="任务类型")
@@ -19,7 +23,7 @@ class Task(TableBase, table=True):
props: Optional[str] = Field(default=None, description="任务属性 (JSON格式)")
# 外键
user_id: "int" = Field(foreign_key="users.id", index=True, description="所属用户ID")
user_id: int = Field(foreign_key="users.id", index=True, description="所属用户ID")
# 关系
user: "User" = Relationship(back_populates="tasks")