226 lines
8.6 KiB
Python
226 lines
8.6 KiB
Python
from sqlalchemy import delete, select, update
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from module_admin.system.controller import test_flow_controller
|
||
from module_admin.system.entity.do.test_eut_do import TestEut
|
||
from module_admin.system.entity.vo.test_eut_vo import Test_eutModel, Test_eutPageQueryModel
|
||
from utils.page_util import PageUtil
|
||
|
||
|
||
class Test_eutDao:
|
||
"""
|
||
产品管理模块数据库操作层
|
||
"""
|
||
|
||
@classmethod
|
||
async def get_test_eut_detail_by_id(cls, db: AsyncSession, id: int):
|
||
"""
|
||
根据产品ID获取产品管理详细信息
|
||
|
||
:param db: orm对象
|
||
:param id: 产品ID
|
||
:return: 产品管理信息对象
|
||
"""
|
||
test_eut_info = (
|
||
(
|
||
await db.execute(
|
||
select(TestEut)
|
||
.where(
|
||
TestEut.id == id
|
||
)
|
||
)
|
||
)
|
||
.scalars()
|
||
.first()
|
||
)
|
||
|
||
return test_eut_info
|
||
|
||
@classmethod
|
||
async def get_test_eut_detail_by_info(cls, db: AsyncSession, test_eut: Test_eutModel):
|
||
"""
|
||
根据产品管理参数获取产品管理信息
|
||
|
||
:param db: orm对象
|
||
:param test_eut: 产品管理参数对象
|
||
:return: 产品管理信息对象
|
||
"""
|
||
test_eut_info = (
|
||
(
|
||
await db.execute(
|
||
select(TestEut).where(
|
||
)
|
||
)
|
||
)
|
||
.scalars()
|
||
.first()
|
||
)
|
||
|
||
return test_eut_info
|
||
|
||
@classmethod
|
||
async def get_test_eut_list(cls, db: AsyncSession, query_object: Test_eutPageQueryModel, is_page: bool = False, order_id_from_work_order: int = None):
|
||
"""
|
||
根据查询参数获取产品管理列表信息
|
||
|
||
:param db: orm对象
|
||
:param query_object: 查询参数对象
|
||
:param is_page: 是否开启分页
|
||
:param order_id_from_work_order: 通过工单的order_id查询样品（用于从工单生成的订单）
|
||
:return: 产品管理列表信息对象
|
||
"""
|
||
from module_admin.system.entity.do.eut_type_do import EutType
|
||
from module_admin.system.entity.do.test_flow_do import TestFlow
|
||
from module_admin.system.entity.do.test_work_order_do import TestWorkOrder
|
||
from sqlalchemy import func
|
||
from utils.common_util import CamelCaseUtil
|
||
import math
|
||
|
||
# 如果指定了 order_id_from_work_order，通过工单查询样品
|
||
if order_id_from_work_order:
|
||
query = (
|
||
select(
|
||
TestEut,
|
||
EutType.name.label('eut_type_name'),
|
||
TestFlow.name.label('test_flow_name')
|
||
)
|
||
.join(TestWorkOrder, TestEut.id == TestWorkOrder.test_eut_id)
|
||
.outerjoin(EutType, TestEut.test_eut_type_id == EutType.id)
|
||
.outerjoin(TestFlow, TestEut.test_flow_id == TestFlow.id)
|
||
.where(
|
||
TestWorkOrder.order_id == order_id_from_work_order,
|
||
TestEut.test_eut_type_id == query_object.test_eut_type_id if query_object.test_eut_type_id else True,
|
||
TestEut.test_flow_id == query_object.test_flow_id if query_object.test_flow_id else True,
|
||
TestEut.sn == query_object.sn if query_object.sn else True,
|
||
TestEut.batch_no == query_object.batch_no if query_object.batch_no else True,
|
||
TestEut.version == query_object.version if query_object.version else True,
|
||
TestEut.sample_appearance == query_object.sample_appearance if query_object.sample_appearance else True,
|
||
TestEut.memo == query_object.memo if query_object.memo else True,
|
||
)
|
||
.order_by(TestEut.id)
|
||
.distinct()
|
||
)
|
||
else:
|
||
# 原有的查询逻辑（通过 test_order_id）
|
||
query = (
|
||
select(
|
||
TestEut,
|
||
EutType.name.label('eut_type_name'),
|
||
TestFlow.name.label('test_flow_name')
|
||
)
|
||
.outerjoin(EutType, TestEut.test_eut_type_id == EutType.id)
|
||
.outerjoin(TestFlow, TestEut.test_flow_id == TestFlow.id)
|
||
.where(
|
||
TestEut.test_order_id == query_object.test_order_id if query_object.test_order_id else True,
|
||
TestEut.test_eut_type_id == query_object.test_eut_type_id if query_object.test_eut_type_id else True,
|
||
TestEut.test_flow_id == query_object.test_flow_id if query_object.test_flow_id else True,
|
||
TestEut.sn == query_object.sn if query_object.sn else True,
|
||
TestEut.batch_no == query_object.batch_no if query_object.batch_no else True,
|
||
TestEut.version == query_object.version if query_object.version else True,
|
||
TestEut.sample_appearance == query_object.sample_appearance if query_object.sample_appearance else True,
|
||
TestEut.memo == query_object.memo if query_object.memo else True,
|
||
)
|
||
.order_by(TestEut.id)
|
||
.distinct()
|
||
)
|
||
|
||
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:
|
||
eut = row[0] # TestEut对象
|
||
eut_dict = CamelCaseUtil.transform_result(eut)
|
||
eut_dict['eutTypeName'] = row[1]
|
||
eut_dict['testFlowName'] = row[2]
|
||
processed_rows.append(eut_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:
|
||
eut = row[0]
|
||
eut_dict = CamelCaseUtil.transform_result(eut)
|
||
eut_dict['eutTypeName'] = row[1]
|
||
eut_dict['testFlowName'] = row[2]
|
||
processed_rows.append(eut_dict)
|
||
|
||
return processed_rows
|
||
|
||
@classmethod
|
||
async def add_test_eut_dao(cls, db: AsyncSession, test_eut: Test_eutModel):
|
||
"""
|
||
新增产品管理数据库操作
|
||
|
||
:param db: orm对象
|
||
:param test_eut: 产品管理对象
|
||
:return:
|
||
"""
|
||
db_test_eut = TestEut(**test_eut.model_dump(
|
||
exclude={'eut_type_name', 'test_flow_name'},
|
||
exclude_unset=True
|
||
))
|
||
db.add(db_test_eut)
|
||
await db.flush()
|
||
|
||
return db_test_eut
|
||
|
||
@classmethod
|
||
async def edit_test_eut_dao(cls, db: AsyncSession, test_eut: dict):
|
||
"""
|
||
编辑产品管理数据库操作
|
||
|
||
:param db: orm对象
|
||
:param test_eut: 需要更新的产品管理字典
|
||
:return:
|
||
"""
|
||
await db.execute(update(TestEut), [test_eut])
|
||
|
||
@classmethod
|
||
async def delete_test_eut_dao(cls, db: AsyncSession, test_eut: Test_eutModel):
|
||
"""
|
||
删除产品管理数据库操作
|
||
|
||
:param db: orm对象
|
||
:param test_eut: 产品管理对象
|
||
:return:
|
||
"""
|
||
await db.execute(delete(TestEut).where(TestEut.id.in_([test_eut.id])))
|
||
|
||
@classmethod
|
||
async def get_test_flow_list(cls, db: AsyncSession):
|
||
"""
|
||
获取测试流程列表
|
||
|
||
:param db: orm对象
|
||
:return: 测试流程列表
|
||
"""
|
||
from module_admin.system.entity.do.test_flow_do import TestFlow
|
||
test_flows = await db.execute(select(TestFlow.id, TestFlow.name))
|
||
return [{"id": item[0], "name": item[1]} for item in test_flows.all()]
|
||
|
||
@classmethod
|
||
async def get_eut_type_list(cls, db: AsyncSession):
|
||
"""
|
||
获取产品类别列表
|
||
|
||
:param db: orm对象
|
||
:return: 产品列表
|
||
"""
|
||
from module_admin.system.entity.do.eut_type_do import EutType
|
||
eut_list = await db.execute(select(EutType.id, EutType.name))
|
||
return [{"id": item[0], "name": item[1]} for item in eut_list.all()]
|
||
|