upload project source code

后端源码/yifan.action-ai.cn/app/core/validator.py

@@ -0,0 +1,167 @@
+# -*- coding: utf-8 -*-
+
+import re
+from datetime import datetime
+from typing import Annotated
+from pydantic import AfterValidator, PlainSerializer, WithJsonSchema
+
+from app.common.constant import RET
+from app.core.exceptions import CustomException
+
+
+# 自定义日期时间字符串类型
+DateTimeStr = Annotated[
+    datetime,
+    AfterValidator(lambda x: datetime_validator(x)),
+    PlainSerializer(lambda x: x.strftime('%Y-%m-%d %H:%M:%S') if isinstance(x, datetime) else str(x), return_type=str),
+    WithJsonSchema({'type': 'string'}, mode='serialization')
+]
+
+# 自定义手机号类型
+Telephone = Annotated[
+    str,
+    AfterValidator(lambda x: mobile_validator(x)),
+    PlainSerializer(lambda x: x, return_type=str),
+    WithJsonSchema({'type': 'string'}, mode='serialization')
+]
+
+# 自定义邮箱类型
+Email = Annotated[
+    str,
+    AfterValidator(lambda x: email_validator(x)),
+    PlainSerializer(lambda x: x, return_type=str),
+    WithJsonSchema({'type': 'string'}, mode='serialization')
+]
+
+def datetime_validator(value: str | datetime) -> datetime:
+    """
+    日期格式验证器。
+    
+    参数:
+    - value (str | datetime): 日期值。
+    
+    返回:
+    - datetime: 格式化后的日期。
+    
+    异常:
+    - CustomException: 日期格式无效时抛出。
+    """
+    pattern = "%Y-%m-%d %H:%M:%S"
+    try:
+        if isinstance(value, str):
+            return datetime.strptime(value, pattern)
+        elif isinstance(value, datetime):
+            return value
+    except Exception:
+        raise CustomException(code=RET.ERROR.code, msg="无效的日期格式")
+
+
+def email_validator(value: str) -> str:
+    """
+    邮箱地址验证器。
+    
+    参数:
+    - value (str): 邮箱地址。
+    
+    返回:
+    - str: 验证后的邮箱地址。
+    
+    异常:
+    - CustomException: 邮箱格式无效时抛出。
+    """
+    if not value:
+        raise CustomException(code=RET.ERROR.code, msg="邮箱地址不能为空")
+
+    regex = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
+
+    if not re.match(regex, value):
+        raise CustomException(code=RET.ERROR.code, msg="邮箱地址格式不正确")
+
+    return value
+
+
+def mobile_validator(value: str | None) -> str | None:
+    """
+    手机号验证器。
+    
+    参数:
+    - value (str | None): 手机号。
+    
+    返回:
+    - str | None: 验证后的手机号。
+    
+    异常:
+    - CustomException: 手机号格式无效时抛出。
+    """
+    if not value:
+        return value
+
+    if len(value) != 11 or not value.isdigit():
+        raise CustomException(code=RET.ERROR.code, msg="手机号格式不正确")
+
+    regex = r'^1(3\d|4[4-9]|5[0-35-9]|6[67]|7[013-8]|8[0-9]|9[0-9])\d{8}$'
+
+    if not re.match(regex, value):
+        raise CustomException(code=RET.ERROR.code, msg="手机号格式不正确")
+
+    return value
+
+
+def menu_request_validator(data):
+    """
+    菜单请求数据验证器。
+    
+    参数:
+    - data (Any): 请求数据。
+    
+    返回:
+    - Any: 验证后的请求数据。
+    
+    异常:
+    - CustomException: 请求数据无效时抛出。
+    """
+    menu_types = {1: "目录", 2: "功能", 3: "权限", 4: "外链"}
+
+    if data.type not in menu_types:
+        raise CustomException(code=RET.ERROR.code, msg=f"菜单类型必须为: {','.join(map(str, menu_types.keys()))}")
+    
+    if data.type in [1, 2]:
+        if not data.route_name:
+            raise CustomException(code=RET.ERROR.code, msg="路由名称不能为空")
+        if not data.route_path:
+            raise CustomException(code=RET.ERROR.code, msg="路由路径不能为空")
+            
+    if data.type == 2 and not data.component_path:
+        raise CustomException(code=RET.ERROR.code, msg="组件路径不能为空")
+        
+    return data
+
+
+def role_permission_request_validator(data):
+    """
+    角色权限设置数据验证器。
+    
+    参数:
+    - data (Any): 请求数据。
+    
+    返回:
+    - Any: 验证后的请求数据。
+    
+    异常:
+    - CustomException: 请求数据无效时抛出。
+    """
+    data_scopes = {
+        1: "仅本人数据权限",
+        2: "本部门数据权限", 
+        3: "本部门及以下数据权限",
+        4: "全部数据权限",
+        5: "自定义数据权限"
+    }
+    
+    if data.data_scope not in data_scopes:
+        raise CustomException(code=RET.ERROR.code, msg=f"数据权限范围必须为: {','.join(map(str, data_scopes.keys()))}")
+        
+    if not data.role_ids:
+        raise CustomException(code=RET.ERROR.code, msg="角色不能为空")
+        
+    return data