feat: 初始化项目架构

ruoyi-fastapi-backend/module_admin/annotation/pydantic_annotation.py

@@ -0,0 +1,82 @@
+import inspect
+from typing import Type
+
+from fastapi import Query, Form
+from pydantic import BaseModel
+from pydantic.fields import FieldInfo
+
+
+def as_query(cls: Type[BaseModel]):
+    """
+    pydantic模型查询参数装饰器，将pydantic模型用于接收查询参数
+    """
+    new_parameters = []
+
+    for field_name, model_field in cls.model_fields.items():
+        model_field: FieldInfo  # type: ignore
+
+        if not model_field.is_required():
+            new_parameters.append(
+                inspect.Parameter(
+                    model_field.alias,
+                    inspect.Parameter.POSITIONAL_ONLY,
+                    default=Query(model_field.default),
+                    annotation=model_field.annotation
+                )
+            )
+        else:
+            new_parameters.append(
+                inspect.Parameter(
+                    model_field.alias,
+                    inspect.Parameter.POSITIONAL_ONLY,
+                    default=Query(...),
+                    annotation=model_field.annotation
+                )
+            )
+
+    async def as_query_func(**data):
+        return cls(**data)
+
+    sig = inspect.signature(as_query_func)
+    sig = sig.replace(parameters=new_parameters)
+    as_query_func.__signature__ = sig  # type: ignore
+    setattr(cls, 'as_query', as_query_func)
+    return cls
+
+
+def as_form(cls: Type[BaseModel]):
+    """
+    pydantic模型表单参数装饰器，将pydantic模型用于接收表单参数
+    """
+    new_parameters = []
+
+    for field_name, model_field in cls.model_fields.items():
+        model_field: FieldInfo  # type: ignore
+
+        if not model_field.is_required():
+            new_parameters.append(
+                inspect.Parameter(
+                    model_field.alias,
+                    inspect.Parameter.POSITIONAL_ONLY,
+                    default=Form(model_field.default),
+                    annotation=model_field.annotation
+                )
+            )
+        else:
+            new_parameters.append(
+                inspect.Parameter(
+                    model_field.alias,
+                    inspect.Parameter.POSITIONAL_ONLY,
+                    default=Form(...),
+                    annotation=model_field.annotation
+                )
+            )
+
+    async def as_form_func(**data):
+        return cls(**data)
+
+    sig = inspect.signature(as_form_func)
+    sig = sig.replace(parameters=new_parameters)
+    as_form_func.__signature__ = sig  # type: ignore
+    setattr(cls, 'as_form', as_form_func)
+    return cls