feat: 更新用户组和选项模型,优化存储信息获取逻辑

This commit is contained in:
2025-12-18 14:27:41 +08:00
parent 8d2658b166
commit d271c81de7
3 changed files with 117 additions and 46 deletions

View File

@@ -1,44 +1,94 @@
from typing import Optional, List, TYPE_CHECKING
from sqlmodel import Field, Relationship, text, Column, JSON
from typing import TYPE_CHECKING
from sqlmodel import Field, Relationship, text
from .base import TableBase
from sqlmodel import SQLModel
if TYPE_CHECKING:
from .user import User
class GroupOptions(SQLModel):
archive_download: bool | None = False
archive_task: bool | None = False
share_download: bool | None = False
share_free: bool | None = False
webdav_proxy: bool | None = False
aria2: bool | None = False
relocate: bool | None = False
source_batch: int | None = 10
redirected_source: bool | None = False
available_nodes: List[int] | None = []
select_node: bool | None = False
advance_delete: bool | None = False
class GroupOptions(TableBase, table=True):
"""用户组选项模型"""
group_id: int = Field(foreign_key="group.id", unique=True)
"""关联的用户组ID"""
archive_download: bool = False
"""是否允许打包下载"""
archive_task: bool = False
"""是否允许创建打包任务"""
share_download: bool = False
"""是否允许分享下载"""
share_free: bool = False
"""是否免积分分享"""
webdav_proxy: bool = False
"""是否允许WebDAV代理"""
aria2: bool = False
"""是否允许使用aria2"""
relocate: bool = False
"""是否允许文件重定位"""
source_batch: int = 10
"""批量获取源地址数量"""
redirected_source: bool = False
"""是否使用重定向源"""
available_nodes: str = "[]"
"""可用节点ID列表JSON数组"""
select_node: bool = False
"""是否允许选择节点"""
advance_delete: bool = False
"""是否允许高级删除"""
# 反向关系
group: "Group" = Relationship(back_populates="options")
class Group(TableBase, table=True):
"""用户组模型"""
name: str = Field(max_length=255, unique=True, description="用户组名")
policies: str | None = Field(default=None, max_length=255, description="允许的策略ID列表逗号分隔")
max_storage: int = Field(default=0, sa_column_kwargs={"server_default": "0"}, description="最大存储空间(字节)")
share_enabled: bool = Field(default=False, sa_column_kwargs={"server_default": text("false")}, description="是否允许创建分享")
web_dav_enabled: bool = Field(default=False, sa_column_kwargs={"server_default": text("false")}, description="是否允许使用WebDAV")
admin: bool = Field(default=False, description="是否为管理员组")
speed_limit: int = Field(default=0, sa_column_kwargs={"server_default": "0"}, description="速度限制 (KB/s), 0为不限制")
options: GroupOptions = Field(default=GroupOptions, sa_column=Column(JSON), description="其他选项")
name: str = Field(max_length=255, unique=True)
"""用户组名"""
policies: str | None = Field(default=None, max_length=255)
"""允许的策略ID列表逗号分隔"""
max_storage: int = Field(default=0, sa_column_kwargs={"server_default": "0"})
"""最大存储空间(字节)"""
share_enabled: bool = Field(default=False, sa_column_kwargs={"server_default": text("false")})
"""是否允许创建分享"""
web_dav_enabled: bool = Field(default=False, sa_column_kwargs={"server_default": text("false")})
"""是否允许使用WebDAV"""
admin: bool = False
"""是否为管理员组"""
speed_limit: int = Field(default=0, sa_column_kwargs={"server_default": "0"})
"""速度限制 (KB/s), 0为不限制"""
# 一对一关系:用户组选项
options: GroupOptions | None = Relationship(
back_populates="group",
sa_relationship_kwargs={"uselist": False}
)
# 关系:一个组可以有多个用户
user: List["User"] = Relationship(
user: list["User"] = Relationship(
back_populates="group",
sa_relationship_kwargs={"foreign_keys": "User.group_id"}
)
previous_user: List["User"] = Relationship(
previous_user: list["User"] = Relationship(
back_populates="previous_group",
sa_relationship_kwargs={"foreign_keys": "User.previous_group_id"}
)
)