|
1 | 1 | {% set pkField = pkColumn.python_field %} |
2 | 2 | {% set pk_field = pkColumn.python_field | camel_to_snake %} |
| 3 | +{% set need_import_datetime = namespace(has_datetime=False) %} |
3 | 4 | {% 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 %} |
6 | 7 | {% endif %} |
7 | 8 | {% endfor %} |
| 9 | +{% if need_import_datetime.has_datetime %} |
| 10 | +from datetime import datetime |
| 11 | +{% endif %} |
8 | 12 | from fastapi import APIRouter, Depends, Form, Request |
9 | 13 | from pydantic_validation_decorator import ValidateFields |
10 | 14 | from sqlalchemy.ext.asyncio import AsyncSession |
@@ -82,8 +86,13 @@ async def edit_{{ moduleName }}_{{ businessName }}( |
82 | 86 | query_db: AsyncSession = Depends(get_db), |
83 | 87 | current_user: CurrentUserModel = Depends(LoginService.get_current_user), |
84 | 88 | ): |
| 89 | + {% for column in columns %} |
| 90 | + {% if column.python_field == "updateBy" %} |
85 | 91 | edit_{{ businessName }}.update_by = current_user.user.user_name |
| 92 | + {% elif column.python_field == "updateTime" %} |
86 | 93 | edit_{{ businessName }}.update_time = datetime.now() |
| 94 | + {% endif %} |
| 95 | + {% endfor %} |
87 | 96 | edit_{{ businessName }}_result = await {{ BusinessName }}Service.edit_{{ businessName }}_services(query_db, edit_{{ businessName }}) |
88 | 97 | logger.info(edit_{{ businessName }}_result.message) |
89 | 98 |
|
|
0 commit comments