2025-02-14 17:42:36 +08:00
{% set pkField = pkColumn.python_field %}
{% set pk_field = pkColumn.python_field | camel_to_snake %}
2025-02-16 23:29:15 +08:00
{% set pkParentheseIndex = pkColumn.column_comment.find("（
{% set pk_field_comment = pkColumn.column_comment[:pkParentheseIndex] if pkParentheseIndex != -1 else pkColumn.column_comment %}
2025-02-17 17:43:29 +08:00
{% for column in columns %}
2025-02-18 11:36:33 +08:00
{% if column.query and column.query_type == 'BETWEEN' and column.python_field == "createTime" %}
2025-02-14 17:42:36 +08:00
from datetime import datetime, time
2025-02-17 17:43:29 +08:00
{% endif %}
{% endfor %}
2025-02-14 17:42:36 +08:00
from sqlalchemy import delete, select, update
from sqlalchemy.ext.asyncio import AsyncSession
2025-02-16 23:29:15 +08:00
{% if table.sub %}
from sqlalchemy.orm import selectinload
{% endif %}
2025-02-18 11:36:33 +08:00
{% if table.sub %}
from {{ packageName }}.entity.do.{{ businessName }}_do import {{ ClassName }}, {{ subClassName }}
from {{ packageName }}.entity.vo.{{ businessName }}_vo import {{ BusinessName }}Model, {{ BusinessName }}PageQueryModel, {{ subTable.business_name | capitalize }}Model
{% else %}
2025-02-14 17:42:36 +08:00
from {{ packageName }}.entity.do.{{ businessName }}_do import {{ ClassName }}
from {{ packageName }}.entity.vo.{{ businessName }}_vo import {{ BusinessName }}Model, {{ BusinessName }}PageQueryModel
2025-02-18 11:36:33 +08:00
{% endif %}
2025-02-14 17:42:36 +08:00
from utils.page_util import PageUtil
class {{ BusinessName }}Dao:
"""
{{ functionName }}模块数据库操作层
"""
@classmethod
async def get_{{ businessName }}_detail_by_id(cls, db: AsyncSession, {{ pk_field }}: int):
"""
2025-02-16 23:29:15 +08:00
根据{{ pk_field_comment }}获取{{ functionName }}详细信息
2025-02-14 17:42:36 +08:00
:param db: orm对象
2025-02-16 23:29:15 +08:00
:param {{ pk_field }}: {{ pk_field_comment }}
2025-02-14 17:42:36 +08:00
:return: {{ functionName }}信息对象
"""
2025-02-17 17:43:29 +08:00
{{ businessName }}_info = (
(
await db.execute(
{% if table.sub %}
select({{ ClassName }})
.options(selectinload({{ ClassName }}.{{ subclassName }}_list))
{% else %}
select({{ ClassName }})
{% endif %}
.where(
{{ ClassName }}.{{ pk_field }} == {{ pk_field }}
)
)
)
.scalars()
.first()
)
2025-02-14 17:42:36 +08:00
return {{ businessName }}_info
@classmethod
async def get_{{ businessName }}_detail_by_info(cls, db: AsyncSession, {{ businessName }}: {{ BusinessName }}Model):
"""
根据{{ functionName }}参数获取{{ functionName }}信息
:param db: orm对象
:param {{ businessName }}: {{ functionName }}参数对象
:return: {{ functionName }}信息对象
"""
{{ businessName }}_info = (
(
await db.execute(
select({{ ClassName }}).where(
{% for column in columns %}
2025-02-26 09:00:35 +08:00
{% if column.unique %}
2025-02-18 16:10:57 +08:00
{{ ClassName }}.{{ column.python_field | camel_to_snake }} == {{ businessName }}.{{ column.python_field | camel_to_snake }} if {{ businessName }}.{{ column.python_field | camel_to_snake }} else True,
2025-02-14 17:42:36 +08:00
{% endif %}
{% endfor %}
)
)
)
.scalars()
.first()
)
return {{ businessName }}_info
@classmethod
async def get_{{ businessName }}_list(cls, db: AsyncSession, query_object: {{ BusinessName }}PageQueryModel, is_page: bool = False):
"""
根据查询参数获取{{ functionName }}列表信息
:param db: orm对象
:param query_object: 查询参数对象
:param is_page: 是否开启分页
:return: {{ functionName }}列表信息对象
"""
query = (
2025-02-17 17:43:29 +08:00
{% if table.sub %}
2025-02-14 17:42:36 +08:00
select({{ ClassName }})
2025-02-17 17:43:29 +08:00
.options(selectinload({{ ClassName }}.{{ subclassName }}_list))
2025-02-16 23:29:15 +08:00
{% else %}
select({{ ClassName }})
{% endif %}
2025-02-14 17:42:36 +08:00
.where(
{% for column in columns %}
{% set field = column.python_field | camel_to_snake %}
{% if column.query %}
{% if column.query_type == "EQ" %}
{{ ClassName }}.{{ field }} == query_object.{{ field }} if query_object.{{ field }} else True,
{% elif column.query_type == "NE" %}
{{ ClassName }}.{{ field }} != query_object.{{ field }} if query_object.{{ field }} else True,
{% elif column.query_type == "GT" %}
{{ ClassName }}.{{ field }} > query_object.{{ field }} if query_object.{{ field }} else True,
{% elif column.query_type == "GTE" %}
{{ ClassName }}.{{ field }} >= query_object.{{ field }} if query_object.{{ field }} else True,
{% elif column.query_type == "LT" %}
{{ ClassName }}.{{ field }} < query_object.{{ field }} if query_object.{{ field }} else True,
{% elif column.query_type == "LTE" %}
{{ ClassName }}.{{ field }} <= query_object.{{ field }} if query_object.{{ field }} else True,
{% elif column.query_type == "LIKE" %}
{{ ClassName }}.{{ field }}.like(f'%{% raw %}{{% endraw %}query_object.{{ field }}{% raw %}}{% endraw %}%') if query_object.{{ field }} else True,
{% elif column.query_type == "BETWEEN" %}
{{ ClassName }}.{{ field }}.between(
datetime.combine(datetime.strptime(query_object.begin_time, '%Y-%m-%d'), time(00, 00, 00)),
datetime.combine(datetime.strptime(query_object.end_time, '%Y-%m-%d'), time(23, 59, 59)),
)
if query_object.begin_time and query_object.end_time
else True,
{% endif %}
{% endif %}
{% endfor %}
)
.order_by({{ ClassName }}.{{ pk_field }})
.distinct()
)
{{ businessName }}_list = await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page)
return {{ businessName }}_list
@classmethod
async def add_{{ businessName }}_dao(cls, db: AsyncSession, {{ businessName }}: {{ BusinessName }}Model):
"""
新增{{ functionName }}数据库操作
:param db: orm对象
2025-02-17 17:43:29 +08:00
:param {{ businessName }}: {{ functionName }}对象
2025-02-14 17:42:36 +08:00
:return:
"""
2025-03-17 09:51:05 +08:00
db_{{ businessName }} = {{ ClassName }}(**{{ businessName }}.model_dump(exclude={% raw %}{{% endraw %}{% if table.sub %}'{{ subclassName }}_list', {% endif %}{% for column in columns %}{% if not column.insert and column.column_name not in column_not_add_show + column_not_edit_show %}'{{ column.python_field | camel_to_snake }}'{% if not loop.last %}, {% endif %}{% endif %}{% endfor %}{% raw %}}{% endraw %}))
2025-02-14 17:42:36 +08:00
db.add(db_{{ businessName }})
await db.flush()
return db_{{ businessName }}
@classmethod
async def edit_{{ businessName }}_dao(cls, db: AsyncSession, {{ businessName }}: dict):
"""
编辑{{ functionName }}数据库操作
:param db: orm对象
2025-02-17 17:43:29 +08:00
:param {{ businessName }}: 需要更新的{{ functionName }}字典
2025-02-14 17:42:36 +08:00
:return:
"""
await db.execute(update({{ ClassName }}), [{{ businessName }}])
@classmethod
async def delete_{{ businessName }}_dao(cls, db: AsyncSession, {{ businessName }}: {{ BusinessName }}Model):
"""
删除{{ functionName }}数据库操作
:param db: orm对象
2025-02-17 17:43:29 +08:00
:param {{ businessName }}: {{ functionName }}对象
2025-02-14 17:42:36 +08:00
:return:
"""
await db.execute(delete({{ ClassName }}).where({{ ClassName }}.{{ pk_field }}.in_([{{ businessName }}.{{ pk_field }}])))
2025-02-17 17:43:29 +08:00
{% if table.sub %}
@classmethod
async def add_{{ subTable.business_name }}_dao(cls, db: AsyncSession, {{ subTable.business_name }}: {{ subTable.business_name | capitalize }}Model):
"""
新增{{ subTable.function_name }}数据库操作
:param db: orm对象
:param {{ subTable.business_name }}: {{ subTable.function_name }}对象
:return:
"""
db_{{ subTable.business_name }} = {{ subClassName }}(**{{ subTable.business_name }}.model_dump())
db.add(db_{{ subTable.business_name }})
await db.flush()
return db_{{ subTable.business_name }}
@classmethod
async def edit_{{ subTable.business_name }}_dao(cls, db: AsyncSession, {{ subTable.business_name }}: dict):
"""
编辑{{ subTable.function_name }}数据库操作
:param db: orm对象
:param {{ subTable.business_name }}: 需要更新的{{ subTable.function_name }}字典
:return:
"""
await db.execute(update({{ subClassName }}), [{{ subTable.business_name }}])
@classmethod
async def delete_{{ subTable.business_name }}_dao(cls, db: AsyncSession, {{ subTable.business_name }}: {{ subTable.business_name | capitalize }}Model):
"""
删除{{ subTable.function_name }}数据库操作
:param db: orm对象
:param {{ subTable.business_name }}: {{ subTable.function_name }}对象
:return:
"""
await db.execute(delete({{ subClassName }}).where({{ subClassName }}.{{ subTable.pk_column.python_field | camel_to_snake }}.in_([{{ subTable.business_name }}.{{ subTable.pk_column.python_field | camel_to_snake }}])))
{% endif %}
