fastgenerateapi 1.2.11__tar.gz → 1.2.12__tar.gz

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastgenerateapi
-Version: 1.2.11
+Version: 1.2.12
 Summary: FastAPIView Class View
 Author: ShiLiang
 Author-email: 2509144896@qq.com

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/fastgenerateapi/__version__.py

@@ -8,7 +8,7 @@
 #                                                  d8888P
 
 
-VERSION = (1, 2, 11)
+VERSION = (1, 2, 12)
 
 __version__ = '.'.join(map(str, VERSION))
 

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/fastgenerateapi/api_view/update_view.py

@@ -59,8 +59,8 @@ class UpdateView(BaseView, SaveMixin):
     async def set_update_model(self, model: Model, data_dict: dict, *args, **kwargs) -> Model:
         """
         在数据模型校验前后修改值:
-        - 【不常使用】：这里修改将不做唯一字段校验
-        - 可用于全局修改
+        - 这里修改将不做唯一字段校验
+        - 可用于例如租户id等全局修改，或者字段变更记录等
         data_dict.update({
             "user_id": request.user.id,
         })

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/fastgenerateapi/schemas_factory/common_function.py

@@ -1,13 +1,14 @@
 import time
 from datetime import datetime
-from typing import Type, Optional, Union, List
+from typing import Type, Optional, Union, List, Callable
 
+from fastapi import HTTPException
 from pydantic import field_validator, model_validator
 from pydantic.fields import FieldInfo
 from tortoise import Model
 
 from fastgenerateapi.api_view.mixin.dbmodel_mixin import DBModelMixin
-from fastgenerateapi.pydantic_utils.base_model import alias_generator
+from fastgenerateapi.pydantic_utils.base_model import alias_generator, BaseModel
 from fastgenerateapi.settings.all_settings import settings
 
 
@@ -165,6 +166,35 @@ def get_validate_dict_from_fields(fields_info: dict) -> dict:
     return validator_dict
 
 
+class FieldValidateInfo(BaseModel):
+    name: str
+    is_required: bool
+    description: str
+    func: Callable
+
+
+def get_field_validate_dict(infos: List[FieldValidateInfo]) -> dict:
+    """
+    字段做校验
+    :param infos:
+    :return:
+    """
+    if not infos:
+        return {}
+
+    def _func(i: FieldValidateInfo):
+        def func(cls, value):
+            is_empty = value is None or value == ''
+            if i.is_required and is_empty:
+                raise HTTPException(status_code=422, detail=i.description+"不能为空")
+            if not is_empty and not i.func(value):
+                raise HTTPException(status_code=422, detail=i.description+"校验不通过")
+            return value
+        return func
+
+    return {"check_"+info.name: field_validator(info.name)(_func(info)) for info in infos}
+
+
 def get_model_validate_text_to_list(fields: Optional[List[str]]) -> dict:
     """
     详情列表用于返回

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/fastgenerateapi/schemas_factory/create_schema_factory.py

@@ -8,7 +8,7 @@ from tortoise import Model
 
 from fastgenerateapi.data_type.data_type import T
 from fastgenerateapi.schemas_factory.common_function import get_dict_from_model_fields, get_dict_from_pydanticmeta, \
-    get_validate_dict_from_fields, get_model_validate_list_to_text
+    get_validate_dict_from_fields, get_model_validate_list_to_text, FieldValidateInfo, get_field_validate_dict
 
 
 def create_schema_factory(
@@ -23,6 +23,8 @@ def create_schema_factory(
         pydantic_meta_save_exclude: Optional[bool] = True,
         pydantic_meta_create_include: Optional[bool] = True,
         pydantic_meta_create_exclude: Optional[bool] = True,
+        pydantic_meta_save_validate: Optional[bool] = True,
+        pydantic_meta_create_validate: Optional[bool] = True,
         pydantic_meta_text_to_list: Optional[bool] = True,
         name: Optional[str] = None
 ) -> Type[T]:
@@ -36,6 +38,8 @@ def create_schema_factory(
     - save_exclude: 增改通用，指定排除字段
     - create_include: 增使用，额外添加字段
     - create_exclude: 增使用，指定排除字段
+    - save_validate: 增改通用，指定字段做校验
+    - create_validate: 增使用，指定字段做校验
 
     逻辑顺序：合并所有include和exclude，排除include下包含的exclude字段
 
@@ -65,14 +69,17 @@ def create_schema_factory(
                 ("item_id_list", List[str], FieldInfo(description="多对多关联id列表")),
             ]
             save_exclude = ["ignore_field"]
+            save_validate = [("phone", phone_validate), ("id_card", id_card_validate)]
             create_include = ["create_user"]
             create_exclude = []
+            create_validate = []
     """
     all_fields_info = get_dict_from_model_fields(model_class)
 
     include_fields = set()
     exclude_fields = set()
     text_to_list_fields = set()
+    validate_field_info = []
     if include:
         include_fields_dict = get_dict_from_pydanticmeta(model_class, include)
         all_fields_info.update(include_fields_dict)
@@ -104,6 +111,24 @@ def create_schema_factory(
                 exclude_fields.update(model_class.PydanticMeta.save_exclude)
             if pydantic_meta_create_exclude and hasattr(model_class.PydanticMeta, "create_exclude"):
                 exclude_fields.update(model_class.PydanticMeta.create_exclude)
+            if pydantic_meta_save_validate and hasattr(model_class.PydanticMeta, "save_validate"):
+                for validate in model_class.PydanticMeta.save_validate:
+                    file_type, file_info = all_fields_info.get(validate[0])
+                    validate_field_info.append(FieldValidateInfo(
+                        name=validate[0],
+                        is_required=file_info.is_required(),
+                        description=file_info.description,
+                        func=validate[1],
+                    ))
+            if pydantic_meta_create_validate and hasattr(model_class.PydanticMeta, "create_validate"):
+                for validate in model_class.PydanticMeta.create_validate:
+                    file_type, file_info = all_fields_info.get(validate[0])
+                    validate_field_info.append(FieldValidateInfo(
+                        name=validate[0],
+                        is_required=file_info.is_required(),
+                        description=file_info.description,
+                        func=validate[1],
+                    ))
             if pydantic_meta_text_to_list and hasattr(model_class.PydanticMeta, "text_to_list"):
                 text_to_list_fields = getattr(model_class.PydanticMeta, "text_to_list")
                 for field in text_to_list_fields:
@@ -136,12 +161,13 @@ def create_schema_factory(
     schema_name = name if name else model_class.__name__ + "CreateSchema"
     schema_field_dict = {field: all_fields_info[field] for field in all_fields}
     validators_dict = get_validate_dict_from_fields(schema_field_dict)
+    field_validators = get_field_validate_dict(validate_field_info)
     text_to_list_validators_dict = get_model_validate_list_to_text(text_to_list_fields)
     schema: Type[T] = create_model(
         schema_name,
         **schema_field_dict,
         __config__=model_config,
-        __validators__={**validators_dict, **text_to_list_validators_dict},
+        __validators__={**validators_dict, **field_validators, **text_to_list_validators_dict},
     )
     return schema
 

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/fastgenerateapi/schemas_factory/update_schema_factory.py

@@ -10,7 +10,7 @@ from tortoise import Model
 
 from fastgenerateapi.data_type.data_type import T
 from fastgenerateapi.schemas_factory.common_function import get_dict_from_model_fields, get_dict_from_pydanticmeta, \
-    get_validate_dict_from_fields, get_model_validate_list_to_text
+    get_validate_dict_from_fields, get_model_validate_list_to_text, FieldValidateInfo, get_field_validate_dict
 
 
 def update_schema_factory(
@@ -25,6 +25,8 @@ def update_schema_factory(
         pydantic_meta_save_exclude: Optional[bool] = True,
         pydantic_meta_update_include: Optional[bool] = True,
         pydantic_meta_update_exclude: Optional[bool] = True,
+        pydantic_meta_save_validate: Optional[bool] = True,
+        pydantic_meta_update_validate: Optional[bool] = True,
         pydantic_meta_text_to_list: Optional[bool] = True,
         name: Optional[str] = None
 ) -> Type[T]:
@@ -36,8 +38,10 @@ def update_schema_factory(
     - exclude： 增改详情列表通用，指定排除字段
     - save_include: 增改通用，额外添加字段
     - save_exclude: 增改通用，指定排除字段
-    - update_include: 增使用，额外添加字段
-    - update_exclude: 增使用，指定排除字段
+    - update_include: 改使用，额外添加字段
+    - update_exclude: 改使用，指定排除字段
+    - save_validate: 增改通用，指定字段做校验
+    - update_validate: 改使用，指定字段做校验
 
     逻辑顺序：合并所有include和exclude，排除include下包含的exclude字段
 
@@ -67,14 +71,17 @@ def update_schema_factory(
                 ("item_id_list", List[str], FieldInfo(description="多对多关联id列表")),
             ]
             save_exclude = ["ignore_field"]
+            save_validate = [("phone", phone_validate), ("id_card", id_card_validate)]
             update_include = ["update_user"]
             update_exclude = []
+            update_validate = []
     """
     all_fields_info = get_dict_from_model_fields(model_class)
 
     include_fields = set()
     exclude_fields = set()
     text_to_list_fields = set()
+    validate_field_info = []
     if include:
         include_fields_dict = get_dict_from_pydanticmeta(model_class, include)
         all_fields_info.update(include_fields_dict)
@@ -106,6 +113,24 @@ def update_schema_factory(
                 exclude_fields.update(model_class.PydanticMeta.save_exclude)
             if pydantic_meta_update_exclude and hasattr(model_class.PydanticMeta, "update_exclude"):
                 exclude_fields.update(model_class.PydanticMeta.update_exclude)
+            if pydantic_meta_save_validate and hasattr(model_class.PydanticMeta, "save_validate"):
+                for validate in model_class.PydanticMeta.save_validate:
+                    file_type, file_info = all_fields_info.get(validate[0])
+                    validate_field_info.append(FieldValidateInfo(
+                        name=validate[0],
+                        is_required=file_info.is_required(),
+                        description=file_info.description,
+                        func=validate[1],
+                    ))
+            if pydantic_meta_update_validate and hasattr(model_class.PydanticMeta, "update_validate"):
+                for validate in model_class.PydanticMeta.create_validate:
+                    file_type, file_info = all_fields_info.get(validate[0])
+                    validate_field_info.append(FieldValidateInfo(
+                        name=validate[0],
+                        is_required=file_info.is_required(),
+                        description=file_info.description,
+                        func=validate[1],
+                    ))
             if pydantic_meta_text_to_list and hasattr(model_class.PydanticMeta, "text_to_list"):
                 text_to_list_fields = getattr(model_class.PydanticMeta, "text_to_list")
                 for field in text_to_list_fields:
@@ -138,11 +163,12 @@ def update_schema_factory(
     schema_name = name if name else model_class.__name__ + "UpdateSchema"
     schema_field_dict = {field: all_fields_info[field] for field in all_fields}
     validators_dict = get_validate_dict_from_fields(schema_field_dict)
+    field_validators = get_field_validate_dict(validate_field_info)
     text_to_list_validators_dict = get_model_validate_list_to_text(text_to_list_fields)
     schema: Type[T] = create_model(
         schema_name,
         **schema_field_dict,
         __config__=model_config,
-        __validators__={**validators_dict, **text_to_list_validators_dict},
+        __validators__={**validators_dict, **field_validators, **text_to_list_validators_dict},
     )
     return schema

fastgenerateapi-1.2.12/fastgenerateapi/validator/validator.py

@@ -0,0 +1,61 @@
+import re
+from datetime import datetime, date
+from typing import Optional
+
+
+def phone_validate(phone: Optional[str]) -> bool:
+    """
+    手机号码校验
+    :param phone:
+    :return:
+    """
+    if not phone:
+        return False
+    phone_pattern = re.compile(r'^(?:(?:\+|00)86)?1[3-9]\d{9}$')
+
+    return phone_pattern.fullmatch(phone) is not None
+
+
+def id_card_validate(id_card: Optional[str]) -> bool:
+    """
+    身份证号码校验
+    :param id_card:
+    :return:
+    """
+    if not id_card or len(id_card) != 18:
+        return False
+    id_card = id_card.upper()
+    if not id_card[:17].isdigit() or id_card[-1] not in '0123456789X':
+        return False
+    date_of_birth = datetime(
+        year=int(id_card[6:10]),
+        month=int(id_card[10:12]),
+        day=int(id_card[12:14])
+    )
+    current_date = date.today()
+    if date_of_birth > current_date or date_of_birth.year < 1900:
+        return False
+
+    weights = [7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2]
+    check_code_map = {0: '1', 1: '0', 2: 'X', 3: '9', 4: '8', 5: '7', 6: '6', 7: '5', 8: '4', 9: '3', 10: '2'}
+
+    total = sum(int(id_card[i]) * weights[i] for i in range(17))
+
+    remainder = total % 11
+    expected_check_code = check_code_map[remainder]
+    return id_card[-1] == expected_check_code
+
+
+def email_validate(email: Optional[str]) -> bool:
+    """
+    验证邮箱格式合法性（符合RFC 5322简化标准，覆盖99%+日常/开发场景）
+    :param email: 待校验的邮箱字符串
+    :return: 合法返回True，否则False
+    """
+    if not email:
+        return False
+
+    email_pattern = r'^[a-zA-Z0-9_%+-]+(?:\.[a-zA-Z0-9_%+-]+)*@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
+
+    return re.fullmatch(email_pattern, email) is not None
+

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/fastgenerateapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastgenerateapi
-Version: 1.2.11
+Version: 1.2.12
 Summary: FastAPIView Class View
 Author: ShiLiang
 Author-email: 2509144896@qq.com

{fastgenerateapi-1.2.11 → fastgenerateapi-1.2.12}/fastgenerateapi.egg-info/SOURCES.txt

@@ -111,4 +111,6 @@ fastgenerateapi/utils/ramdom_utils.py
 fastgenerateapi/utils/snowflake.py
 fastgenerateapi/utils/str_util.py
 fastgenerateapi/utils/swagger_to_js.py
+fastgenerateapi/validator/__init__.py
+fastgenerateapi/validator/validator.py
 script/__init__.py