Skip to content

Commit 2ea6347

Browse files
committed
fix: 修复代码生成controller模板编辑接口异常生成字段的问题
1 parent 0d03a03 commit 2ea6347

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

ruoyi-fastapi-backend/module_generator/templates/python/controller.py.jinja2

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,14 @@
11
{% set pkField = pkColumn.python_field %}
22
{% set pk_field = pkColumn.python_field | camel_to_snake %}
3+
{% set need_import_datetime = namespace(has_datetime=False) %}
34
{% for column in columns %}
4-
{% if column.python_field == "createTime" %}
5-
from datetime import datetime
5+
{% if column.python_field in ["createTime", "updatetime"] %}
6+
{% set need_import_datetime.has_datetime = True %}
67
{% endif %}
78
{% endfor %}
9+
{% if need_import_datetime.has_datetime %}
10+
from datetime import datetime
11+
{% endif %}
812
from fastapi import APIRouter, Depends, Form, Request
913
from pydantic_validation_decorator import ValidateFields
1014
from sqlalchemy.ext.asyncio import AsyncSession
@@ -82,8 +86,13 @@ async def edit_{{ moduleName }}_{{ businessName }}(
8286
query_db: AsyncSession = Depends(get_db),
8387
current_user: CurrentUserModel = Depends(LoginService.get_current_user),
8488
):
89+
{% for column in columns %}
90+
{% if column.python_field == "updateBy" %}
8591
edit_{{ businessName }}.update_by = current_user.user.user_name
92+
{% elif column.python_field == "updateTime" %}
8693
edit_{{ businessName }}.update_time = datetime.now()
94+
{% endif %}
95+
{% endfor %}
8796
edit_{{ businessName }}_result = await {{ BusinessName }}Service.edit_{{ businessName }}_services(query_db, edit_{{ businessName }})
8897
logger.info(edit_{{ businessName }}_result.message)
8998

0 commit comments

Comments
 (0)