216 lines
8.3 KiB
Python
216 lines
8.3 KiB
Python
from sqlalchemy import delete, select, update
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy.orm import aliased
|
||
from module_admin.system.entity.do.test_job_do import TestJob
|
||
from module_admin.entity.do.user_do import SysUser
|
||
from module_admin.system.entity.vo.test_job_vo import Test_jobModel, Test_jobPageQueryModel
|
||
from utils.page_util import PageUtil
|
||
|
||
|
||
class Test_jobDao:
|
||
"""
|
||
作业模块数据库操作层
|
||
"""
|
||
|
||
@classmethod
|
||
async def get_test_job_detail_by_id(cls, db: AsyncSession, id: int):
|
||
"""
|
||
根据作业ID获取作业详细信息（包含关联用户名称）
|
||
|
||
:param db: orm对象
|
||
:param id: 作业ID
|
||
:return: 作业信息对象
|
||
"""
|
||
# 为同一个表创建多个别名，用于多次JOIN
|
||
TesterUser = aliased(SysUser)
|
||
ReviewerUser = aliased(SysUser)
|
||
SecondTesterUser = aliased(SysUser)
|
||
ThirdTesterUser = aliased(SysUser)
|
||
|
||
query_result = await db.execute(
|
||
select(
|
||
TestJob,
|
||
TesterUser.nick_name.label('tester_name'),
|
||
ReviewerUser.nick_name.label('reviewer_name'),
|
||
SecondTesterUser.nick_name.label('second_tester_name'),
|
||
ThirdTesterUser.nick_name.label('third_tester_name')
|
||
)
|
||
.outerjoin(TesterUser, TestJob.tester_id == TesterUser.user_id)
|
||
.outerjoin(ReviewerUser, TestJob.reviewer_id == ReviewerUser.user_id)
|
||
.outerjoin(SecondTesterUser, TestJob.second_tester_id == SecondTesterUser.user_id)
|
||
.outerjoin(ThirdTesterUser, TestJob.third_tester_id == ThirdTesterUser.user_id)
|
||
.where(TestJob.id == id)
|
||
)
|
||
|
||
row = query_result.first()
|
||
if row:
|
||
# 转换为dict格式，包含ID和名称
|
||
from utils.common_util import CamelCaseUtil
|
||
job = row[0] # TestJob对象
|
||
job_dict = CamelCaseUtil.transform_result(job)
|
||
job_dict['testerName'] = row[1]
|
||
job_dict['reviewerName'] = row[2]
|
||
job_dict['secondTesterName'] = row[3]
|
||
job_dict['thirdTesterName'] = row[4]
|
||
return job_dict
|
||
|
||
return None
|
||
|
||
@classmethod
|
||
async def get_test_job_detail_by_info(cls, db: AsyncSession, test_job: Test_jobModel):
|
||
"""
|
||
根据作业参数获取作业信息
|
||
|
||
:param db: orm对象
|
||
:param test_job: 作业参数对象
|
||
:return: 作业信息对象
|
||
"""
|
||
test_job_info = (
|
||
(
|
||
await db.execute(
|
||
select(TestJob).where(
|
||
)
|
||
)
|
||
)
|
||
.scalars()
|
||
.first()
|
||
)
|
||
|
||
return test_job_info
|
||
|
||
@classmethod
|
||
async def get_test_job_list(cls, db: AsyncSession, query_object: Test_jobPageQueryModel, is_page: bool = False):
|
||
"""
|
||
根据查询参数获取作业列表信息（包含关联用户名称）
|
||
|
||
:param db: orm对象
|
||
:param query_object: 查询参数对象
|
||
:param is_page: 是否开启分页
|
||
:return: 作业列表信息对象
|
||
"""
|
||
# 为同一个表创建多个别名，用于多次JOIN
|
||
TesterUser = aliased(SysUser)
|
||
ReviewerUser = aliased(SysUser)
|
||
SecondTesterUser = aliased(SysUser)
|
||
ThirdTesterUser = aliased(SysUser)
|
||
|
||
query = (
|
||
select(
|
||
TestJob,
|
||
TesterUser.nick_name.label('tester_name'),
|
||
ReviewerUser.nick_name.label('reviewer_name'),
|
||
SecondTesterUser.nick_name.label('second_tester_name'),
|
||
ThirdTesterUser.nick_name.label('third_tester_name')
|
||
)
|
||
.outerjoin(TesterUser, TestJob.tester_id == TesterUser.user_id)
|
||
.outerjoin(ReviewerUser, TestJob.reviewer_id == ReviewerUser.user_id)
|
||
.outerjoin(SecondTesterUser, TestJob.second_tester_id == SecondTesterUser.user_id)
|
||
.outerjoin(ThirdTesterUser, TestJob.third_tester_id == ThirdTesterUser.user_id)
|
||
.where(
|
||
TestJob.name.like(f'%{query_object.name}%') if query_object.name else True,
|
||
TestJob.test_item_id == query_object.test_item_id if query_object.test_item_id else True,
|
||
TestJob.tester_id == query_object.tester_id if query_object.tester_id else True,
|
||
TestJob.reviewer_id == query_object.reviewer_id if query_object.reviewer_id else True,
|
||
TestJob.second_tester_id == query_object.second_tester_id if query_object.second_tester_id else True,
|
||
TestJob.third_tester_id == query_object.third_tester_id if query_object.third_tester_id else True,
|
||
TestJob.memo == query_object.memo if query_object.memo else True,
|
||
)
|
||
.order_by(TestJob.id)
|
||
.distinct()
|
||
)
|
||
|
||
# 不使用PageUtil，自己处理JOIN后的数据
|
||
from sqlalchemy import func
|
||
from utils.common_util import CamelCaseUtil
|
||
import math
|
||
|
||
if is_page:
|
||
# 计算总数
|
||
total = (await db.execute(select(func.count('*')).select_from(query.subquery()))).scalar()
|
||
# 执行分页查询
|
||
query_result = await db.execute(query.offset((query_object.page_num - 1) * query_object.page_size).limit(query_object.page_size))
|
||
|
||
# 处理JOIN结果
|
||
processed_rows = []
|
||
for row in query_result:
|
||
job = row[0] # TestJob对象
|
||
job_dict = CamelCaseUtil.transform_result(job)
|
||
job_dict['testerName'] = row[1]
|
||
job_dict['reviewerName'] = row[2]
|
||
job_dict['secondTesterName'] = row[3]
|
||
job_dict['thirdTesterName'] = row[4]
|
||
processed_rows.append(job_dict)
|
||
|
||
has_next = math.ceil(total / query_object.page_size) > query_object.page_num
|
||
return {
|
||
'rows': processed_rows,
|
||
'total': total,
|
||
'pageNum': query_object.page_num,
|
||
'pageSize': query_object.page_size,
|
||
'hasNext': has_next
|
||
}
|
||
else:
|
||
# 非分页模式
|
||
query_result = await db.execute(query)
|
||
processed_rows = []
|
||
for row in query_result:
|
||
job = row[0]
|
||
job_dict = CamelCaseUtil.transform_result(job)
|
||
job_dict['testerName'] = row[1]
|
||
job_dict['reviewerName'] = row[2]
|
||
job_dict['secondTesterName'] = row[3]
|
||
job_dict['thirdTesterName'] = row[4]
|
||
processed_rows.append(job_dict)
|
||
return processed_rows
|
||
|
||
@classmethod
|
||
async def add_test_job_dao(cls, db: AsyncSession, test_job: Test_jobModel):
|
||
"""
|
||
新增作业数据库操作
|
||
|
||
:param db: orm对象
|
||
:param test_job: 作业对象
|
||
:return:
|
||
"""
|
||
# 调试信息
|
||
print(f"DEBUG add: test_job type = {type(test_job)}")
|
||
|
||
# 如果test_job已经是dict，直接使用
|
||
if isinstance(test_job, dict):
|
||
job_data = {k: v for k, v in test_job.items()
|
||
if k not in ['tester_name', 'reviewer_name', 'second_tester_name', 'third_tester_name']}
|
||
db_test_job = TestJob(**job_data)
|
||
else:
|
||
# 排除name字段（这些是JOIN查询返回的，不存在于数据库表中）
|
||
db_test_job = TestJob(**test_job.model_dump(
|
||
exclude={'tester_name', 'reviewer_name', 'second_tester_name', 'third_tester_name'},
|
||
exclude_unset=True
|
||
))
|
||
db.add(db_test_job)
|
||
await db.flush()
|
||
print("db_test_job.id",db_test_job.id)
|
||
return db_test_job
|
||
|
||
@classmethod
|
||
async def edit_test_job_dao(cls, db: AsyncSession, test_job: dict):
|
||
"""
|
||
编辑作业数据库操作
|
||
|
||
:param db: orm对象
|
||
:param test_job: 需要更新的作业字典
|
||
:return:
|
||
"""
|
||
await db.execute(update(TestJob), [test_job])
|
||
|
||
@classmethod
|
||
async def delete_test_job_dao(cls, db: AsyncSession, test_job: Test_jobModel):
|
||
"""
|
||
删除作业数据库操作
|
||
|
||
:param db: orm对象
|
||
:param test_job: 作业对象
|
||
:return:
|
||
"""
|
||
await db.execute(delete(TestJob).where(TestJob.id.in_([test_job.id])))
|
||
|