126 lines
4.6 KiB
Python
126 lines
4.6 KiB
Python
from sqlalchemy import select, func, and_, or_, update, delete
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from module_admin.entity.do.warehouse_sample_do import WarehouseSample
|
|
from module_admin.entity.vo.warehouse_sample_vo import WarehouseSamplePageQueryModel, WarehouseSampleModel
|
|
from typing import List, Optional
|
|
|
|
|
|
class WarehouseSampleDao:
|
|
"""
|
|
样品数据访问层
|
|
"""
|
|
|
|
@classmethod
|
|
async def get_sample_by_id(cls, db: AsyncSession, sample_id: int):
|
|
"""
|
|
根据样品ID获取样品信息
|
|
"""
|
|
query = select(WarehouseSample).where(
|
|
WarehouseSample.sample_id == sample_id,
|
|
WarehouseSample.del_flag == '0'
|
|
)
|
|
result = await db.execute(query)
|
|
return result.scalars().first()
|
|
|
|
@classmethod
|
|
async def get_sample_list(cls, db: AsyncSession, query_object: WarehouseSamplePageQueryModel, is_page: bool = False):
|
|
"""
|
|
获取样品列表
|
|
"""
|
|
query = select(WarehouseSample).where(WarehouseSample.del_flag == '0')
|
|
|
|
# 条件筛选
|
|
if query_object.receipt_id:
|
|
query = query.where(WarehouseSample.receipt_id == query_object.receipt_id)
|
|
if query_object.receipt_no:
|
|
query = query.where(WarehouseSample.receipt_no.like(f'%{query_object.receipt_no}%'))
|
|
if query_object.sample_model:
|
|
query = query.where(WarehouseSample.sample_model.like(f'%{query_object.sample_model}%'))
|
|
if query_object.sample_sn:
|
|
query = query.where(WarehouseSample.sample_sn.like(f'%{query_object.sample_sn}%'))
|
|
if query_object.external_status:
|
|
query = query.where(WarehouseSample.external_status == query_object.external_status)
|
|
if query_object.status:
|
|
query = query.where(WarehouseSample.status == query_object.status)
|
|
|
|
# 排序
|
|
query = query.order_by(WarehouseSample.create_time.desc())
|
|
|
|
# 分页
|
|
if is_page:
|
|
offset = (query_object.page_num - 1) * query_object.page_size
|
|
query = query.offset(offset).limit(query_object.page_size)
|
|
|
|
result = await db.execute(query)
|
|
return result.scalars().all()
|
|
|
|
@classmethod
|
|
async def get_sample_count(cls, db: AsyncSession, query_object: WarehouseSamplePageQueryModel):
|
|
"""
|
|
获取样品总数
|
|
"""
|
|
query = select(func.count()).select_from(WarehouseSample).where(WarehouseSample.del_flag == '0')
|
|
|
|
# 条件筛选（与列表查询保持一致）
|
|
if query_object.receipt_id:
|
|
query = query.where(WarehouseSample.receipt_id == query_object.receipt_id)
|
|
if query_object.receipt_no:
|
|
query = query.where(WarehouseSample.receipt_no.like(f'%{query_object.receipt_no}%'))
|
|
if query_object.sample_model:
|
|
query = query.where(WarehouseSample.sample_model.like(f'%{query_object.sample_model}%'))
|
|
if query_object.sample_sn:
|
|
query = query.where(WarehouseSample.sample_sn.like(f'%{query_object.sample_sn}%'))
|
|
if query_object.external_status:
|
|
query = query.where(WarehouseSample.external_status == query_object.external_status)
|
|
if query_object.status:
|
|
query = query.where(WarehouseSample.status == query_object.status)
|
|
|
|
result = await db.execute(query)
|
|
return result.scalar()
|
|
|
|
@classmethod
|
|
async def add_sample(cls, db: AsyncSession, sample: WarehouseSample):
|
|
"""
|
|
新增样品
|
|
"""
|
|
db.add(sample)
|
|
await db.flush()
|
|
return sample
|
|
|
|
@classmethod
|
|
async def edit_sample(cls, db: AsyncSession, sample: WarehouseSampleModel):
|
|
"""
|
|
编辑样品
|
|
"""
|
|
update_data = sample.model_dump(exclude_unset=True, exclude={'sample_id'})
|
|
stmt = update(WarehouseSample).where(WarehouseSample.sample_id == sample.sample_id).values(**update_data)
|
|
await db.execute(stmt)
|
|
await db.flush()
|
|
|
|
@classmethod
|
|
async def delete_sample(cls, db: AsyncSession, sample_ids: List[int]):
|
|
"""
|
|
删除样品（逻辑删除）
|
|
"""
|
|
stmt = update(WarehouseSample).where(WarehouseSample.sample_id.in_(sample_ids)).values(del_flag='1')
|
|
await db.execute(stmt)
|
|
await db.flush()
|
|
|
|
@classmethod
|
|
async def get_sample_count_by_receipt(cls, db: AsyncSession, receipt_id: int):
|
|
"""
|
|
获取入库单下的样品数量
|
|
"""
|
|
query = select(func.count()).select_from(WarehouseSample).where(
|
|
WarehouseSample.receipt_id == receipt_id,
|
|
WarehouseSample.del_flag == '0'
|
|
)
|
|
result = await db.execute(query)
|
|
return result.scalar()
|
|
|
|
|
|
|
|
|
|
|
|
|