154 lines
3.6 KiB
Python
154 lines
3.6 KiB
Python
|
||
from typing import TYPE_CHECKING
|
||
from datetime import datetime
|
||
from uuid import UUID
|
||
|
||
from sqlmodel import Field, Relationship, text, UniqueConstraint, Index
|
||
|
||
from .base import SQLModelBase
|
||
from .mixin import TableBaseMixin
|
||
|
||
if TYPE_CHECKING:
|
||
from .user import User
|
||
from .report import Report
|
||
from .object import Object
|
||
|
||
|
||
class Share(SQLModelBase, TableBaseMixin):
|
||
"""分享模型"""
|
||
|
||
__table_args__ = (
|
||
UniqueConstraint("code", name="uq_share_code"),
|
||
Index("ix_share_source_name", "source_name"),
|
||
Index("ix_share_user_created", "user_id", "created_at"),
|
||
Index("ix_share_object", "object_id"),
|
||
)
|
||
|
||
code: str = Field(max_length=64, nullable=False, index=True)
|
||
"""分享码"""
|
||
|
||
password: str | None = Field(default=None, max_length=255)
|
||
"""分享密码（加密后）"""
|
||
|
||
object_id: UUID = Field(
|
||
foreign_key="object.id",
|
||
index=True,
|
||
ondelete="CASCADE"
|
||
)
|
||
"""关联的对象UUID"""
|
||
|
||
views: int = Field(default=0, sa_column_kwargs={"server_default": "0"})
|
||
"""浏览次数"""
|
||
|
||
downloads: int = Field(default=0, sa_column_kwargs={"server_default": "0"})
|
||
"""下载次数"""
|
||
|
||
remain_downloads: int | None = Field(default=None)
|
||
"""剩余下载次数 (NULL为不限制)"""
|
||
|
||
expires: datetime | None = Field(default=None)
|
||
"""过期时间 (NULL为永不过期)"""
|
||
|
||
preview_enabled: bool = Field(default=True, sa_column_kwargs={"server_default": text("true")})
|
||
"""是否允许预览"""
|
||
|
||
source_name: str | None = Field(default=None, max_length=255)
|
||
"""源名称（冗余字段，便于展示）"""
|
||
|
||
score: int = Field(default=0, sa_column_kwargs={"server_default": "0"})
|
||
"""兑换此分享所需的积分"""
|
||
|
||
# 外键
|
||
user_id: UUID = Field(
|
||
foreign_key="user.id",
|
||
index=True,
|
||
ondelete="CASCADE"
|
||
)
|
||
"""创建分享的用户UUID"""
|
||
|
||
# 关系
|
||
user: "User" = Relationship(back_populates="shares")
|
||
"""分享创建者"""
|
||
|
||
object: "Object" = Relationship(back_populates="shares")
|
||
"""关联的对象"""
|
||
|
||
reports: list["Report"] = Relationship(
|
||
back_populates="share",
|
||
sa_relationship_kwargs={"cascade": "all, delete-orphan"}
|
||
)
|
||
"""举报列表"""
|
||
|
||
@property
|
||
def is_dir(self) -> bool:
|
||
"""是否为目录分享（向后兼容属性）"""
|
||
from .object import ObjectType
|
||
return self.object.type == ObjectType.FOLDER if self.object else False
|
||
|
||
|
||
# ==================== DTO 模型 ====================
|
||
|
||
class ShareCreateRequest(SQLModelBase):
|
||
"""创建分享请求 DTO"""
|
||
|
||
object_id: UUID
|
||
"""要分享的对象UUID"""
|
||
|
||
password: str | None = None
|
||
"""分享密码（可选）"""
|
||
|
||
expires: datetime | None = None
|
||
"""过期时间（可选，NULL为永不过期）"""
|
||
|
||
remain_downloads: int | None = None
|
||
"""剩余下载次数（可选，NULL为不限制）"""
|
||
|
||
preview_enabled: bool = True
|
||
"""是否允许预览"""
|
||
|
||
score: int = 0
|
||
"""兑换此分享所需的积分"""
|
||
|
||
|
||
class ShareResponse(SQLModelBase):
|
||
"""分享响应 DTO"""
|
||
|
||
id: int
|
||
"""分享ID"""
|
||
|
||
code: str
|
||
"""分享码"""
|
||
|
||
object_id: UUID
|
||
"""关联对象UUID"""
|
||
|
||
source_name: str | None
|
||
"""源名称"""
|
||
|
||
views: int
|
||
"""浏览次数"""
|
||
|
||
downloads: int
|
||
"""下载次数"""
|
||
|
||
remain_downloads: int | None
|
||
"""剩余下载次数"""
|
||
|
||
expires: datetime | None
|
||
"""过期时间"""
|
||
|
||
preview_enabled: bool
|
||
"""是否允许预览"""
|
||
|
||
score: int
|
||
"""积分"""
|
||
|
||
created_at: datetime
|
||
"""创建时间"""
|
||
|
||
is_expired: bool
|
||
"""是否已过期"""
|
||
|
||
has_password: bool
|
||
"""是否有密码"""
|