upload project source code

后端源码/yifan.action-ai.cn/app/api/v1/module_system/role/schema.py

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+
+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
+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: str | None = Field(default=None, max_length=40, description="角色编码")
+    order: int | None = Field(default=1, ge=1, description='显示排序')
+    data_scope: int | None = Field(default=1, description='数据权限范围(1:仅本人 2:本部门 3:本部门及以下 4:全部 5:自定义)')
+    status: str = Field(default="0", description="是否启用")
+    description: str | None = Field(default=None, max_length=255, description="描述")
+
+    @field_validator("code")
+    @classmethod
+    def validate_code(cls, value: str | None):
+        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, description='数据权限范围(1:仅本人 2:本部门 3:本部门及以下 4:全部 5:自定义)')
+    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):
+    """角色信息响应模型"""
+    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: str | None = Query(None, description="角色名称"),
+        status: str | None = Query(None, description="是否可用"),
+        created_time: list[DateTimeStr] | None = Query(None, description="创建时间范围", examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"]),
+        updated_time: list[DateTimeStr] | None = Query(None, description="更新时间范围", examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"]),
+    ) -> None:
+        
+        # 模糊查询字段
+        self.name = ("like", name)
+
+        # 精确查询字段
+        self.status = status
+        
+        # 时间范围查询
+        if created_time and len(created_time) == 2:
+            self.created_time = ("between", (created_time[0], created_time[1]))
+        
+        if updated_time and len(updated_time) == 2:
+            self.updated_time = ("between", (updated_time[0], updated_time[1]))
+