Files
MoviePilot/app/schemas/initialization.py

57 lines
2.3 KiB
Python

"""
首次初始化接口使用的请求与状态模型。
"""
import re
from pydantic import BaseModel, Field, field_validator, model_validator
class InitializationStatus(BaseModel):
"""
描述 MoviePilot 的首次初始化状态。
未初始化且部署配置已提供 SUPERUSER 时,同时返回固定的管理员用户名,供初始化页面锁定输入。
"""
initialized: bool = Field(description="是否已经完成首次初始化")
configured_username: str | None = Field(default=None, description="部署配置指定的管理员名,未指定时为空")
class InitializationRequest(BaseModel):
"""
首次初始化超级管理员、密码和 API Key 所需的提交数据。
"""
username: str = Field(min_length=1, max_length=50, description="首次创建的超级管理员用户名,长度为 1 至 50 个字符")
password: str = Field(min_length=6, max_length=50, description="管理员密码,长度为 6 至 50 个字符,并包含字母、数字、特殊字符中的至少两类")
confirm_password: str = Field(min_length=6, max_length=50, description="再次输入的管理员密码,必须与 password 完全一致")
api_key: str = Field(min_length=16, max_length=256, description="接口访问密钥,长度为 16 至 256 个字符")
@field_validator("username", "api_key", mode="before")
@classmethod
def strip_text(cls, value: str) -> str:
"""
去除用户名和 API Key 的首尾空白,避免产生不可见凭据差异。
"""
return value.strip() if isinstance(value, str) else value
@field_validator("password", "confirm_password")
@classmethod
def validate_password_shape(cls, value: str) -> str:
"""
复用用户管理中的密码复杂度约束。
"""
if not re.match(r"^(?![a-zA-Z]+$)(?!\d+$)(?![^\da-zA-Z\s]+$).{6,50}$", value):
raise ValueError("密码需要同时包含字母、数字、特殊字符中的至少两项,且长度为 6-50 位")
return value
@model_validator(mode="after")
def passwords_match(self) -> "InitializationRequest":
"""
确保两次输入的密码完全一致。
"""
if self.password != self.confirm_password:
raise ValueError("两次输入的密码不一致")
return self