feat: 添加代码生成功能

2025-03-04 00:54:33 +08:00
parent c790233aee
commit bd13f1cfdc
16 changed files with 1758 additions and 1 deletions
--- a/templates/python/api.py.jinja
+++ b/templates/python/api.py.jinja
@@ -0,0 +1,135 @@
+# _*_ coding : UTF-8 _*_
+# @Time : {{ current_time }}
+# @UpdateTime : {{ current_time }}
+# @Author : {{ author }}
+# @File : {{ table_name }}.py
+# @Comment : 本程序用于生成{{ table_comment }}增删改查接口
+
+from datetime import datetime
+from typing import Optional
+
+from fastapi import APIRouter, Depends, Path, Request, Query
+from fastapi.responses import JSONResponse
+
+from annotation.auth import Auth
+from annotation.log import Log
+from config.constant import BusinessType
+from controller.login import LoginController
+from models import {{ class_name }}
+from schemas.common import BaseResponse, DeleteListParams
+from schemas.{{ name }} import Add{{ class_name }}Params, Update{{ class_name }}Params, Get{{ class_name }}InfoResponse, Get{{ class_name }}ListResponse
+from utils.response import Response
+
+{{ table_name }}API= APIRouter(
+    prefix="{{ prefix }}",
+    dependencies=[Depends(LoginController.get_current_user)],
+)
+
+@{{ table_name }}API.post("/add", response_class=JSONResponse, response_model=BaseResponse, summary="新增{{ description }}")
+@Log(title="新增{{ description }}", business_type=BusinessType.INSERT)
+@Auth(permission_list=["{{ name }}:btn:add"])
+async def add_{{ name }}(request: Request, params: Add{{ class_name }}Params):
+    if await {{ class_name }}.get_or_none(
+    {% for column in columns if column.is_insert %}
+    {{ column.python_name }} = params.{{ column.python_name }},
+    {% endfor %}
+    del_flag=1
+    ):
+        return Response.error(msg="{{ description }}已存在!")
+    {{ name }} = await {{ class_name }}.create(
+        {% for column in columns if column.is_insert %}
+        {{ column.python_name }} = params.{{ column.python_name }},
+        {% endfor %}
+    )
+    if {{ name }}:
+        return Response.success(msg="新增成功!")
+    else:
+        return Response.error(msg="新增失败")
+
+
+@{{ table_name }}API.delete("/delete/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="删除{{ description }}")
+@{{ table_name }}API.post("/delete/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="删除{{ description }}")
+@Log(title="删除{{ description }}", business_type=BusinessType.DELETE)
+@Auth(permission_list=["{{ name }}:btn:delete"])
+async def delete_{{ name }}(request: Request, id: str = Path(description="{{ description }}ID")):
+    if {{ name }} := await {{ class_name }}.get_or_none(id=id, del_flag=1):
+        {{ name }}.del_flag = 0
+        await {{ name }}.save()
+        return Response.success(msg="删除成功")
+    else:
+        return Response.error(msg="{{ description }}不存在!")
+
+
+@{{ table_name }}API.delete("/deleteList", response_class=JSONResponse, response_model=BaseResponse, summary="批量删除{{ description }}")
+@{{ table_name }}API.post("/deleteList", response_class=JSONResponse, response_model=BaseResponse, summary="批量删除{{ description }}")
+@Log(title="批量删除{{ description }}", business_type=BusinessType.DELETE)
+@Auth(permission_list=["{{ name }}:btn:delete"])
+async def delete_{{ name }}_list(request: Request, params: DeleteListParams):
+    for id in set(params.ids):
+        if {{ name }} := await {{ class_name }}.get_or_none(id=id, del_flag=1):
+            {{ name }}.del_flag = 0
+            await {{ name }}.save()
+    return Response.success(msg="删除成功")
+
+
+@{{ table_name }}API.put("/update/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="修改{{ description }}")
+@{{ table_name }}API.post("/update/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="修改{{ description }}")
+@Log(title="修改{{ description }}", business_type=BusinessType.UPDATE)
+@Auth(permission_list=["{{ name }}:btn:update"])
+async def update_{{ name }}(request: Request, params: Update{{ class_name }}Params, id: str = Path(description="{{ description }}ID")):
+    if {{ name }} := await {{ class_name }}.get_or_none(id=id, del_flag=1):
+        {% for column in columns if column.is_edit %}
+        {{ name }}.{{ column.python_name }} = params.{{ column.python_name }},
+        {% endfor %}
+        await {{ name }}.save()
+        return Response.success(msg="修改成功")
+    else:
+        return Response.error(msg="{{ description }}不存在")
+
+
+@{{ table_name }}API.get("/info/{id}", response_class=JSONResponse, response_model=Get{{ class_name }}InfoResponse, summary="获取{{ description }}信息")
+@Log(title="获取{{ description }}信息", business_type=BusinessType.SELECT)
+@Auth(permission_list=["{{ name }}:btn:info"])
+async def get_{{ name }}_info(request: Request, id: str = Path(description="{{ description }}ID")):
+    if {{ name }} := await {{ class_name }}.get_or_none(id=id, del_flag=1):
+        data = {
+            {% for column in columns if column.is_list %}
+            "{{ column.python_name }}":{{ name }}.{{ column.python_name }},
+            {% endfor %}
+        }
+        return Response.success(data=data)
+    else:
+        return Response.error(msg="{{ description }}不存在")
+
+
+@{{ table_name }}API.get("/list", response_class=JSONResponse, response_model=Get{{ class_name }}ListResponse, summary="获取{{ description }}列表")
+@Log(title="获取{{ description }}列表", business_type=BusinessType.SELECT)
+@Auth(permission_list=["{{ name }}:btn:list"])
+async def get_{{ name }}_list(
+        request: Request,
+
+        page: int = Query(default=1, description="当前页码"),
+
+        pageSize: int = Query(default=10, description="每页数量"),
+        {% for column in columns if column.is_query %}
+        {{ column.python_name }}: Optional[str] = Query(default=None, description="{{ column.column_comment }}"),
+        {% endfor %}
+        ):
+    filterArgs={
+        {% for column in columns if column.is_query %}
+        "{{ column.python_name }}{{ column.query_way }}": {{ column.python_name }},
+        {% endfor %}
+    }
+    filterArgs = {k: v for k, v in filterArgs.items() if v is not None}
+    total = await {{ class_name }}.filter(**filterArgs, del_flag=1).count()
+    data = await {{ class_name }}.filter(**filterArgs, del_flag=1).offset((page - 1) * pageSize).limit(pageSize).values(
+        {% for column in columns if column.is_list %}
+        {{ column.python_name }} = "{{ column.python_name }}",
+        {% endfor %}
+    )
+    return Response.success(data={
+        "total": total,
+        "result": data,
+        "page": page,
+        "pageSize": pageSize,
+    })
--- a/templates/python/model.py.jinja
+++ b/templates/python/model.py.jinja
@@ -0,0 +1,44 @@
+# _*_ coding : UTF-8 _*_
+# @Time : {{ current_time }}
+# @UpdateTime : {{ current_time }}
+# @Author : {{ author }}
+# @File : {{ table_name }}.py
+# @Comment : 本程序用于{{ table_comment }}模型
+
+from tortoise import fields
+
+from models.common import BaseModel
+
+
+class {{ class_name }}(BaseModel):
+    """
+    {{ table_comment }}模型
+    """
+    {% for column in columns %}
+    {%- set params = [] %}
+    {%- if column.max_length is not none %}{% set params = params + ["max_length=" ~ column.max_length] %}{% endif %}
+    {%- if column.is_nullable %}{% set params = params + ["null=True"] %}{% endif %}
+    {%- if column.is_unique %}{% set params = params + ["unique=True"] %}{% endif %}
+    {%- if column.default is not none %}{% set params = params + ["default=" ~ column.default] %}{% endif %}
+    {%- if column.column_comment %}{% set params = params + ['description="' ~ column.column_comment ~ '"'] %}{% endif %}
+    {%- if column.column_name %}{% set params = params + ['source_field="' ~ column.column_name ~ '"'] %}{% endif %}
+
+    {{ column.python_name }} = fields.{{ column.field_type }}({{ params | join(", ") }})
+    """
+    {{ column.column_comment }}。
+    {%- if column.max_length is not none %}
+    - 最大长度为 {{ column.max_length }} 个字符
+    {%- endif %}
+    - 映射到数据库字段 {{ column.column_name }}
+    {%- if column.is_nullable %}
+    - 可为空
+    {%- endif %}
+    {%- if column.default is not none %}
+    - 默认值：{{ column.default }}
+    {%- endif %}
+    """
+    {% endfor %}
+
+    class Meta:
+        table = "{{ table_name }}"
+        table_description = "{{ table_comment }}"
--- a/templates/python/schemas.py.jinja
+++ b/templates/python/schemas.py.jinja
@@ -0,0 +1,51 @@
+# _*_ coding : UTF-8 _*_
+# @Time : {{ current_time }}
+# @UpdateTime : {{ current_time }}
+# @Author : {{ author }}
+# @File : {{ table_name }}.py
+# @Comment : 本程序用于生成{{ table_comment }}参数和响应模型
+
+from datetime import datetime
+from typing import Optional, List
+from pydantic import BaseModel, Field, ConfigDict
+from pydantic.alias_generators import to_snake
+from schemas.common import BaseResponse, ListQueryResult
+
+class {{ class_name }}Info(BaseModel):
+    """{{ description }}信息"""
+    model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
+    {% for column in columns %}
+    {{ column.python_name }}: {% if not column.is_required %}Optional[{% endif %}{{ column.python_type }}{% if not column.is_required %}]{% endif %} = Field(
+        {% if column.default is not none %}default={{ column.default }}, {% endif %}title="{{ column.column_comment }}"
+    )
+    {% endfor %}
+
+class Add{{ class_name }}Params(BaseModel):
+    """新增{{ description }}参数"""
+    model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
+    {% for column in columns if column.is_insert %}
+    {{ column.python_name }}: {% if not column.is_required %}Optional[{% endif %}{{ column.python_type }}{% if not column.is_required %}]{% endif %} = Field(
+        {% if column.default is not none %}default={{ column.default }}, {% endif %}title="{{ column.column_comment }}"
+    )
+    {% endfor %}
+
+class Update{{ class_name }}Params(BaseModel):
+    """更新{{ description }}参数"""
+    model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
+    {% for column in columns if column.is_edit %}
+    {{ column.python_name }}: {% if not column.is_required %}Optional[{% endif %}{{ column.python_type }}{% if not column.is_required %}]{% endif %} = Field(
+        {% if column.default is not none %}default={{ column.default }}, {% endif %}title="{{ column.column_comment }}"
+    )
+    {% endfor %}
+
+class Get{{ class_name }}InfoResponse(BaseResponse):
+    """获取{{ description }}信息响应"""
+    data: {{ class_name }}Info = Field(None, title="{{ table_comment }}信息")
+
+class Get{{ class_name }}InfoListResult(ListQueryResult):
+    """获取{{ description }}信息列表响应结果"""
+    result: List[{{ class_name }}Info] = Field(None, title="{{ table_comment }}信息列表")
+
+class Get{{ class_name }}InfoListResponse(BaseResponse):
+    """获取{{ description }}信息列表响应"""
+    data: Get{{ class_name }}InfoListResult = Field(None, title="{{ table_comment }}信息列表")