于小丘
16cec42181
- Added DownloadTokenManager for creating and verifying JWT download tokens. - Introduced new download routes for creating download tokens and downloading files using tokens. - Restructured file upload routes into a dedicated sub-router. - Updated file upload session management with improved error handling and response structures. - Created a new MCP (Microservice Communication Protocol) router with basic request and response models. - Added base models for MCP requests and responses, including method enumeration.
109 lines
2.5 KiB
Python
109 lines
2.5 KiB
Python
from typing import Literal
|
||
from enum import StrEnum
|
||
|
||
from sqlmodel import Field, UniqueConstraint
|
||
|
||
from .base import SQLModelBase
|
||
from .mixin import TableBaseMixin
|
||
|
||
|
||
# ==================== DTO 模型 ====================
|
||
|
||
class SiteConfigResponse(SQLModelBase):
|
||
"""站点配置响应 DTO"""
|
||
|
||
title: str = "DiskNext"
|
||
"""网站标题"""
|
||
|
||
themes: dict[str, str] = {}
|
||
"""网站主题配置"""
|
||
|
||
default_theme: dict[str, str] = {}
|
||
"""默认主题RGB色号"""
|
||
|
||
site_notice: str | None = None
|
||
"""网站公告"""
|
||
|
||
user: dict[str, str | int | bool] = {}
|
||
"""用户信息"""
|
||
|
||
logo_light: str | None = None
|
||
"""网站Logo URL"""
|
||
|
||
logo_dark: str | None = None
|
||
"""网站Logo URL（深色模式）"""
|
||
|
||
captcha_type: Literal["none", "default", "gcaptcha", "cloudflare turnstile"] = "none"
|
||
"""验证码类型"""
|
||
|
||
captcha_key: str | None = None
|
||
"""验证码密钥"""
|
||
|
||
|
||
# ==================== 管理员设置 DTO ====================
|
||
|
||
class SettingItem(SQLModelBase):
|
||
"""设置项 DTO"""
|
||
|
||
name: str
|
||
"""设置项名称"""
|
||
|
||
value: str | None = None
|
||
"""设置值"""
|
||
|
||
|
||
class SettingsUpdateRequest(SQLModelBase):
|
||
"""更新设置请求 DTO"""
|
||
|
||
settings: dict[str, dict[str, str | None]]
|
||
"""按类型分组的设置项，格式: {"basic": {"siteName": "xxx", ...}, ...}"""
|
||
|
||
|
||
class SettingsGetResponse(SQLModelBase):
|
||
"""获取设置响应 DTO"""
|
||
|
||
settings: dict[str, dict[str, str | None]] = {}
|
||
"""按类型分组的设置项"""
|
||
|
||
|
||
# ==================== 数据库模型 ====================
|
||
|
||
class SettingsType(StrEnum):
|
||
"""设置类型枚举"""
|
||
|
||
ARIA2 = "aria2"
|
||
AUTH = "auth"
|
||
AUTHN = "authn"
|
||
AVATAR = "avatar"
|
||
BASIC = "basic"
|
||
CAPTCHA = "captcha"
|
||
CRON = "cron"
|
||
FILE_EDIT = "file_edit"
|
||
LOGIN = "login"
|
||
MAIL = "mail"
|
||
MAIL_TEMPLATE = "mail_template"
|
||
MOBILE = "mobile"
|
||
PATH = "path"
|
||
PREVIEW = "preview"
|
||
PWA = "pwa"
|
||
REGISTER = "register"
|
||
RETRY = "retry"
|
||
SHARE = "share"
|
||
SLAVE = "slave"
|
||
TASK = "task"
|
||
THUMB = "thumb"
|
||
TIMEOUT = "timeout"
|
||
UPLOAD = "upload"
|
||
VERSION = "version"
|
||
VIEW = "view"
|
||
WOPI = "wopi"
|
||
|
||
# 数据库模型
|
||
class Setting(SQLModelBase, TableBaseMixin):
|
||
"""设置模型"""
|
||
|
||
__table_args__ = (UniqueConstraint("type", "name", name="uq_setting_type_name"),)
|
||
|
||
type: SettingsType = Field(max_length=255, description="设置类型/分组")
|
||
name: str = Field(max_length=255, description="设置项名称")
|
||
value: str | None = Field(default=None, description="设置值") |