Yuerchu
b364b740ca
- Updated README to include KodBox in project vision. - Added model descriptions for clarity in Download, File, Folder, Group, Node, Order, Policy, Redeem, Report, Request, Response, Setting, Share, SourceLink, StoragePack, Tag, Task, User, and WebDAV. - Changed optional fields from Optional[...] to the new union type syntax (e.g., str | None). - Improved foreign key references in models for consistency. - Refactored relationships in models to use singular forms where appropriate. - Updated login service to reflect changes in request model types.
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
|
|
from typing import Optional, List, TYPE_CHECKING
|
|
from sqlmodel import Field, Relationship, UniqueConstraint, CheckConstraint
|
|
from .base import TableBase
|
|
from datetime import datetime
|
|
|
|
if TYPE_CHECKING:
|
|
from .user import User
|
|
from .policy import Policy
|
|
from .file import File
|
|
|
|
class Folder(TableBase, table=True):
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"owner_id",
|
|
"parent_id",
|
|
"name",
|
|
name="uq_folder_name_parent",
|
|
),
|
|
CheckConstraint(
|
|
"name NOT LIKE '%/%' AND name NOT LIKE '%\\%'",
|
|
name="ck_folder_name_no_slash",
|
|
),
|
|
)
|
|
|
|
name: str = Field(max_length=255, nullable=False, description="目录名")
|
|
|
|
# 外键
|
|
parent_id: int | None = Field(default=None, foreign_key="folder.id", index=True, description="父目录ID")
|
|
owner_id: int = Field(foreign_key="user.id", index=True, description="所有者用户ID")
|
|
policy_id: int = Field(foreign_key="policy.id", index=True, description="所属存储策略ID")
|
|
|
|
# 关系
|
|
owner: "User" = Relationship(back_populates="folders")
|
|
policy: "Policy" = Relationship(back_populates="folders")
|
|
|
|
# 自我引用关系
|
|
parent: Optional["Folder"] = Relationship(back_populates="children", sa_relationship_kwargs={"remote_side": "Folder.id"})
|
|
children: List["Folder"] = Relationship(back_populates="parent")
|
|
|
|
files: List["File"] = Relationship(back_populates="folder") |