mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-10-12 20:27:19 +00:00
83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from typing import List, Optional
|
|
from fastapi import Query
|
|
from pydantic import BaseModel, ConfigDict, Field, model_validator, field_validator
|
|
|
|
from app.core.validator import DateTimeStr
|
|
from app.core.base_schema import BaseSchema, UserBySchema, TenantSchema
|
|
from app.core.validator import role_permission_request_validator
|
|
|
|
from ..dept.schema import DeptOutSchema
|
|
from ..menu.schema import MenuOutSchema
|
|
|
|
|
|
class RoleCreateSchema(BaseModel):
|
|
"""角色创建模型"""
|
|
name: str = Field(..., max_length=40, description="角色名称")
|
|
code: Optional[str] = Field(default=None, max_length=40, description="角色编码")
|
|
order: Optional[int] = Field(default=1, ge=1, description='显示排序')
|
|
data_scope: Optional[int] = Field(default=1, ge=1, le=5, description='数据权限范围')
|
|
status: str = Field(default="0", description="是否启用")
|
|
description: Optional[str] = Field(default=None, max_length=255, description="描述")
|
|
|
|
@field_validator("code")
|
|
@classmethod
|
|
def validate_code(cls, value: Optional[str]):
|
|
if value is None:
|
|
return value
|
|
import re
|
|
v = value.strip()
|
|
if not re.match(r"^[A-Za-z][A-Za-z0-9_]{1,39}$", v):
|
|
raise ValueError("角色编码需字母开头，允许字母/数字/下划线，长度2-40")
|
|
return v
|
|
|
|
|
|
class RolePermissionSettingSchema(BaseModel):
|
|
"""角色权限配置模型"""
|
|
data_scope: int = Field(default=1, ge=1, le=5, description='数据权限范围')
|
|
role_ids: List[int] = Field(default_factory=list, description='角色ID列表')
|
|
menu_ids: List[int] = Field(default_factory=list, description='菜单ID列表')
|
|
dept_ids: List[int] = Field(default_factory=list, description='部门ID列表')
|
|
|
|
@model_validator(mode='after')
|
|
def validate_fields(self):
|
|
"""验证权限配置字段"""
|
|
return role_permission_request_validator(self)
|
|
|
|
|
|
class RoleUpdateSchema(RoleCreateSchema):
|
|
"""角色更新模型"""
|
|
...
|
|
|
|
|
|
class RoleOutSchema(RoleCreateSchema, BaseSchema, UserBySchema, TenantSchema):
|
|
"""角色信息响应模型"""
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
menus: List[MenuOutSchema] = Field(default_factory=list, description='角色菜单列表')
|
|
depts: List[DeptOutSchema] = Field(default_factory=list, description='角色部门列表')
|
|
|
|
|
|
class RoleQueryParam:
|
|
"""角色管理查询参数"""
|
|
|
|
def __init__(
|
|
self,
|
|
name: Optional[str] = Query(None, description="角色名称"),
|
|
status: Optional[str] = Query(None, description="是否可用"),
|
|
created_id: Optional[int] = Query(None, description="创建人"),
|
|
created_time: Optional[list[DateTimeStr]] = Query(None, description="创建时间范围", example=["2025-01-01 00:00:00", "2025-12-31 23:59:59"]),
|
|
) -> None:
|
|
|
|
# 模糊查询字段
|
|
self.name = ("like", name)
|
|
|
|
# 精确查询字段
|
|
self.created_id = created_id
|
|
self.status = status
|
|
|
|
# 时间范围查询
|
|
if created_time and len(created_time) == 2:
|
|
self.created_time = ("between", (created_time[0], created_time[1]))
|