于小丘
6c96c43bea
Some checks failed
Test / test (push) Failing after 3m47s
- 替换 Field(max_length=X) 为 StrX/TextX 类型别名（21 个 sqlmodels 文件） - 替换 get + 404 检查为 get_exist_one()（17 个路由文件，约 50 处） - 替换 save + session.refresh 为 save(load=...) - 替换 session.add + commit 为 save()（dav/provider.py） - 更新所有依赖至最新版本 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
277 lines
6.9 KiB
Python
277 lines
6.9 KiB
Python
from typing import TYPE_CHECKING
|
||
from uuid import UUID
|
||
|
||
from enum import StrEnum
|
||
from sqlmodel import Field, Relationship, text
|
||
|
||
from sqlmodel_ext import SQLModelBase, UUIDTableBaseMixin, Str64, Str255
|
||
|
||
if TYPE_CHECKING:
|
||
from .object import Object
|
||
from .group import Group
|
||
|
||
|
||
class GroupPolicyLink(SQLModelBase, table=True):
|
||
"""用户组与存储策略的多对多关联表"""
|
||
|
||
group_id: UUID = Field(
|
||
foreign_key="group.id",
|
||
primary_key=True,
|
||
ondelete="CASCADE"
|
||
)
|
||
"""用户组UUID"""
|
||
|
||
policy_id: UUID = Field(
|
||
foreign_key="policy.id",
|
||
primary_key=True,
|
||
ondelete="CASCADE"
|
||
)
|
||
"""存储策略UUID"""
|
||
|
||
|
||
class PolicyType(StrEnum):
|
||
LOCAL = "local"
|
||
S3 = "s3"
|
||
|
||
|
||
class PolicyBase(SQLModelBase):
|
||
"""存储策略基础字段，供 DTO 和数据库模型共享"""
|
||
|
||
name: Str255
|
||
"""策略名称"""
|
||
|
||
type: PolicyType
|
||
"""存储策略类型"""
|
||
|
||
server: Str255 | None = None
|
||
"""服务器地址（本地策略为绝对路径）"""
|
||
|
||
bucket_name: Str255 | None = None
|
||
"""存储桶名称"""
|
||
|
||
is_private: bool = True
|
||
"""是否为私有空间"""
|
||
|
||
base_url: Str255 | None = None
|
||
"""访问文件的基础URL"""
|
||
|
||
access_key: str | None = None
|
||
"""Access Key"""
|
||
|
||
secret_key: str | None = None
|
||
"""Secret Key"""
|
||
|
||
max_size: int = Field(default=0, ge=0)
|
||
"""允许上传的最大文件尺寸（字节）"""
|
||
|
||
auto_rename: bool = False
|
||
"""是否自动重命名"""
|
||
|
||
dir_name_rule: Str255 | None = None
|
||
"""目录命名规则"""
|
||
|
||
file_name_rule: Str255 | None = None
|
||
"""文件命名规则"""
|
||
|
||
is_origin_link_enable: bool = False
|
||
"""是否开启源链接访问"""
|
||
|
||
|
||
# ==================== DTO 模型 ====================
|
||
|
||
|
||
class PolicySummary(SQLModelBase):
|
||
"""策略摘要，用于列表展示"""
|
||
|
||
id: UUID
|
||
"""策略UUID"""
|
||
|
||
name: str
|
||
"""策略名称"""
|
||
|
||
type: PolicyType
|
||
"""策略类型"""
|
||
|
||
server: str | None
|
||
"""服务器地址"""
|
||
|
||
max_size: int
|
||
"""最大文件尺寸"""
|
||
|
||
is_private: bool
|
||
"""是否私有"""
|
||
|
||
|
||
class PolicyCreateRequest(PolicyBase):
|
||
"""创建存储策略请求 DTO，包含 PolicyOptions 扁平字段"""
|
||
|
||
# PolicyOptions 字段（平铺到请求体中，与 GroupCreateRequest 模式一致）
|
||
token: str | None = None
|
||
"""访问令牌"""
|
||
|
||
file_type: str | None = None
|
||
"""允许的文件类型"""
|
||
|
||
mimetype: str | None = Field(default=None, max_length=127)
|
||
"""MIME类型"""
|
||
|
||
od_redirect: Str255 | None = None
|
||
"""OneDrive重定向地址"""
|
||
|
||
chunk_size: int = Field(default=52428800, ge=1)
|
||
"""分片上传大小（字节），默认50MB"""
|
||
|
||
s3_path_style: bool = False
|
||
"""是否使用S3路径风格"""
|
||
|
||
s3_region: Str64 = 'us-east-1'
|
||
"""S3 区域（如 us-east-1、ap-southeast-1），仅 S3 策略使用"""
|
||
|
||
|
||
class PolicyUpdateRequest(SQLModelBase):
|
||
"""更新存储策略请求 DTO（所有字段可选）"""
|
||
|
||
name: Str255 | None = None
|
||
"""策略名称"""
|
||
|
||
server: Str255 | None = None
|
||
"""服务器地址"""
|
||
|
||
bucket_name: Str255 | None = None
|
||
"""存储桶名称"""
|
||
|
||
is_private: bool | None = None
|
||
"""是否为私有空间"""
|
||
|
||
base_url: Str255 | None = None
|
||
"""访问文件的基础URL"""
|
||
|
||
access_key: str | None = None
|
||
"""Access Key"""
|
||
|
||
secret_key: str | None = None
|
||
"""Secret Key"""
|
||
|
||
max_size: int | None = Field(default=None, ge=0)
|
||
"""允许上传的最大文件尺寸（字节）"""
|
||
|
||
auto_rename: bool | None = None
|
||
"""是否自动重命名"""
|
||
|
||
dir_name_rule: Str255 | None = None
|
||
"""目录命名规则"""
|
||
|
||
file_name_rule: Str255 | None = None
|
||
"""文件命名规则"""
|
||
|
||
is_origin_link_enable: bool | None = None
|
||
"""是否开启源链接访问"""
|
||
|
||
# PolicyOptions 字段
|
||
token: str | None = None
|
||
"""访问令牌"""
|
||
|
||
file_type: str | None = None
|
||
"""允许的文件类型"""
|
||
|
||
mimetype: str | None = Field(default=None, max_length=127)
|
||
"""MIME类型"""
|
||
|
||
od_redirect: Str255 | None = None
|
||
"""OneDrive重定向地址"""
|
||
|
||
chunk_size: int | None = Field(default=None, ge=1)
|
||
"""分片上传大小（字节）"""
|
||
|
||
s3_path_style: bool | None = None
|
||
"""是否使用S3路径风格"""
|
||
|
||
s3_region: Str64 | None = None
|
||
"""S3 区域"""
|
||
|
||
|
||
# ==================== 数据库模型 ====================
|
||
|
||
|
||
class PolicyOptionsBase(SQLModelBase):
|
||
"""存储策略选项的基础模型"""
|
||
|
||
token: str | None = Field(default=None)
|
||
"""访问令牌"""
|
||
|
||
file_type: str | None = Field(default=None)
|
||
"""允许的文件类型"""
|
||
|
||
mimetype: str | None = Field(default=None, max_length=127)
|
||
"""MIME类型"""
|
||
|
||
od_redirect: Str255 | None = None
|
||
"""OneDrive重定向地址"""
|
||
|
||
chunk_size: int = Field(default=52428800, sa_column_kwargs={"server_default": "52428800"})
|
||
"""分片上传大小（字节），默认50MB"""
|
||
|
||
s3_path_style: bool = Field(default=False, sa_column_kwargs={"server_default": text("false")})
|
||
"""是否使用S3路径风格"""
|
||
|
||
s3_region: Str64 = Field(default='us-east-1', sa_column_kwargs={"server_default": "'us-east-1'"})
|
||
"""S3 区域（如 us-east-1、ap-southeast-1），仅 S3 策略使用"""
|
||
|
||
|
||
class PolicyOptions(PolicyOptionsBase, UUIDTableBaseMixin):
|
||
"""存储策略选项模型（与Policy一对一关联）"""
|
||
|
||
policy_id: UUID = Field(
|
||
foreign_key="policy.id",
|
||
unique=True,
|
||
ondelete="CASCADE"
|
||
)
|
||
"""关联的策略UUID"""
|
||
|
||
# 反向关系
|
||
policy: "Policy" = Relationship(back_populates="options")
|
||
"""关联的策略"""
|
||
|
||
|
||
class Policy(PolicyBase, UUIDTableBaseMixin):
|
||
"""存储策略模型"""
|
||
|
||
# 覆盖基类字段以添加数据库专有配置
|
||
name: Str255 = Field(unique=True)
|
||
"""策略名称"""
|
||
|
||
is_private: bool = Field(default=True, sa_column_kwargs={"server_default": text("true")})
|
||
"""是否为私有空间"""
|
||
|
||
max_size: int = Field(default=0, sa_column_kwargs={"server_default": "0"})
|
||
"""允许上传的最大文件尺寸（字节）"""
|
||
|
||
auto_rename: bool = Field(default=False, sa_column_kwargs={"server_default": text("false")})
|
||
"""是否自动重命名"""
|
||
|
||
is_origin_link_enable: bool = Field(default=False, sa_column_kwargs={"server_default": text("false")})
|
||
"""是否开启源链接访问"""
|
||
|
||
# 一对一关系：策略选项
|
||
options: PolicyOptions | None = Relationship(
|
||
back_populates="policy",
|
||
sa_relationship_kwargs={"uselist": False, "cascade": "all, delete-orphan"},
|
||
)
|
||
"""策略的扩展选项"""
|
||
|
||
# 关系
|
||
objects: list["Object"] = Relationship(back_populates="policy")
|
||
"""策略下的所有对象"""
|
||
|
||
# 多对多关系：策略可以被多个用户组使用
|
||
groups: list["Group"] = Relationship(
|
||
back_populates="policies",
|
||
link_model=GroupPolicyLink,
|
||
)
|
||
|
||
@staticmethod
|
||
async def create(
|
||
policy: 'Policy | None' = None,
|
||
**kwargs
|
||
):
|
||
pass |