1531 lines
55 KiB
Python
1531 lines
55 KiB
Python
"""样本接收 CRUD（质量检测）"""
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Any, Dict, List, Optional, Tuple
|
|
|
|
from sqlalchemy import and_, func, or_
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.utils import generate_uuid
|
|
from app.crud.collection_protocol import urls_to_attachment_list
|
|
from app.utils.sample_special import sample_special_to_dict
|
|
from app.crud import sample_reception_check_result as check_result_svc
|
|
from app.crud import sample_production_reception as prod_reception_svc
|
|
from app.models.models import (
|
|
BizCellSampleType,
|
|
BizCollectionProtocol,
|
|
BizExpressCompany,
|
|
BizProtocolCollectionInfo,
|
|
BizProtocolPartyCustomer,
|
|
BizProtocolSample,
|
|
BizProtocolSampleTransport,
|
|
BizSampleReception,
|
|
SysUser,
|
|
)
|
|
from app.models.research_donation import (
|
|
BizResearchDonation,
|
|
BizResearchDonationCollection,
|
|
BizResearchDonationParty,
|
|
BizResearchDonationSample,
|
|
BizResearchDonationSampleTransport,
|
|
)
|
|
from app.core.protocol_status_display import INACTIVE_PROTOCOL_STATUSES
|
|
from app.services.web_auth_roles import get_user_role_names
|
|
|
|
SAMPLE_RECEPTION_EDIT_ROLE = "质量主管"
|
|
|
|
# 科研捐赠表默认 general_ci，样本接收等为 unicode_ci，跨表比较需统一
|
|
_UNICODE_CI = "utf8mb4_unicode_ci"
|
|
|
|
|
|
def _donation_sample_id_col(col):
|
|
"""捐赠样本 ID 与 biz_sample_reception.donation_sample_id 比较时统一排序规则。"""
|
|
return col.collate(_UNICODE_CI)
|
|
|
|
|
|
def _fmt_dt(t: Optional[datetime]) -> str:
|
|
return t.strftime("%Y-%m-%d %H:%M:%S") if t else ""
|
|
|
|
|
|
def _normalize_date_bound(value: str, *, end_of_day: bool = False) -> str:
|
|
val = (value or "").strip()
|
|
if len(val) == 10:
|
|
return f"{val} 23:59:59" if end_of_day else f"{val} 00:00:00"
|
|
return val
|
|
|
|
|
|
def _dec(v: Any) -> Optional[float]:
|
|
if v is None:
|
|
return None
|
|
return float(v)
|
|
|
|
|
|
def _reception_to_dict(
|
|
r: BizSampleReception,
|
|
*,
|
|
check_results: Optional[List[Dict[str, Any]]] = None,
|
|
) -> Dict[str, Any]:
|
|
return {
|
|
"id": r.id,
|
|
"protocolSampleId": r.protocol_sample_id,
|
|
"collectionProtocolId": r.collection_protocol_id,
|
|
"receptionTime": _fmt_dt(r.reception_time),
|
|
"handlerUserId": r.handler_user_id or "",
|
|
"handlerName": r.handler_name or "",
|
|
"receiveTemperature": r.receive_temperature or "",
|
|
"checkResults": check_results or [],
|
|
"acceptConclusion": r.accept_conclusion or "",
|
|
"labStorageCode": r.lab_storage_code or "",
|
|
"receiveAttachments": r.receive_attachments or "",
|
|
"remark": r.remark or "",
|
|
"createTime": _fmt_dt(r.create_time),
|
|
"updateTime": _fmt_dt(r.update_time),
|
|
}
|
|
|
|
|
|
def _sample_detail_rows_query(db: Session):
|
|
"""协议样本 + 协议单 + 客户 + 运输（显式 select_from 避免多表 join 歧义）。"""
|
|
return (
|
|
db.query(
|
|
BizProtocolSample,
|
|
BizCollectionProtocol,
|
|
BizProtocolPartyCustomer,
|
|
BizProtocolSampleTransport,
|
|
)
|
|
.select_from(BizProtocolSample)
|
|
.join(
|
|
BizCollectionProtocol,
|
|
and_(
|
|
BizCollectionProtocol.id == BizProtocolSample.collection_protocol_id,
|
|
BizCollectionProtocol.del_flag == 0,
|
|
),
|
|
)
|
|
.outerjoin(
|
|
BizProtocolPartyCustomer,
|
|
and_(
|
|
BizProtocolPartyCustomer.collection_protocol_id == BizCollectionProtocol.id,
|
|
BizProtocolPartyCustomer.del_flag == 0,
|
|
),
|
|
)
|
|
.outerjoin(
|
|
BizProtocolSampleTransport,
|
|
and_(
|
|
BizProtocolSampleTransport.sample_id == BizProtocolSample.id,
|
|
BizProtocolSampleTransport.del_flag == 0,
|
|
),
|
|
)
|
|
)
|
|
|
|
|
|
def _donation_sample_detail_rows_query(db: Session):
|
|
"""捐赠样本 + 捐赠单 + 当事人 + 运输（四表 JOIN）。"""
|
|
return (
|
|
db.query(
|
|
BizResearchDonationSample,
|
|
BizResearchDonation,
|
|
BizResearchDonationParty,
|
|
BizResearchDonationSampleTransport,
|
|
)
|
|
.select_from(BizResearchDonationSample)
|
|
.join(
|
|
BizResearchDonation,
|
|
and_(
|
|
BizResearchDonation.id == BizResearchDonationSample.research_donation_id,
|
|
BizResearchDonation.del_flag == 0,
|
|
),
|
|
)
|
|
.outerjoin(
|
|
BizResearchDonationParty,
|
|
and_(
|
|
BizResearchDonationParty.research_donation_id == BizResearchDonation.id,
|
|
BizResearchDonationParty.del_flag == 0,
|
|
),
|
|
)
|
|
.outerjoin(
|
|
BizResearchDonationSampleTransport,
|
|
and_(
|
|
BizResearchDonationSampleTransport.sample_id == BizResearchDonationSample.id,
|
|
BizResearchDonationSampleTransport.del_flag == 0,
|
|
),
|
|
)
|
|
)
|
|
|
|
|
|
def _donation_sample_pending_query(db: Session, tenant_no: str, keyword: Optional[str] = None):
|
|
"""待接收：捐赠样本且尚无有效接收记录（按租户 + 样本编码判定，与细胞存储一致）；排除已作废捐赠单。"""
|
|
rec = BizSampleReception
|
|
received_exists = (
|
|
db.query(rec.id)
|
|
.filter(
|
|
rec.tenant_no == tenant_no,
|
|
rec.del_flag == 0,
|
|
rec.reception_source == "research_donation",
|
|
rec.sample_code.isnot(None),
|
|
rec.sample_code != "",
|
|
rec.sample_code == BizResearchDonationSample.sample_code,
|
|
)
|
|
.exists()
|
|
)
|
|
q = _donation_sample_detail_rows_query(db).filter(
|
|
BizResearchDonationSample.del_flag == 0,
|
|
BizResearchDonationSample.tenant_no == tenant_no,
|
|
BizResearchDonation.tenant_no == tenant_no,
|
|
BizResearchDonation.donation_status.notin_(INACTIVE_PROTOCOL_STATUSES),
|
|
~received_exists,
|
|
)
|
|
if keyword and keyword.strip():
|
|
kw = f"%{keyword.strip()}%"
|
|
kw_conds = [
|
|
BizResearchDonationSample.sample_code.like(kw),
|
|
BizResearchDonationParty.mother_name.like(kw),
|
|
BizResearchDonationParty.newborn_name.like(kw),
|
|
]
|
|
lab_codes = sample_codes_matching_lab_storage_keyword(
|
|
db, tenant_no, keyword, reception_source="research_donation"
|
|
)
|
|
if lab_codes:
|
|
kw_conds.append(BizResearchDonationSample.sample_code.in_(lab_codes))
|
|
q = q.filter(or_(*kw_conds))
|
|
return q
|
|
|
|
|
|
def _donation_rows_to_sample_list_items(
|
|
db: Session,
|
|
rows: list,
|
|
tenant_no: str,
|
|
*,
|
|
reception_map: Optional[Dict[str, BizSampleReception]] = None,
|
|
) -> List[Dict[str, Any]]:
|
|
"""将捐赠样本行转为统一列表项字典（字段名与细胞存储保持一致）。"""
|
|
type_ids = list({s.sample_type_id for s, *_ in rows if s.sample_type_id})
|
|
type_map: Dict[str, str] = {}
|
|
if type_ids:
|
|
for t in (
|
|
db.query(BizCellSampleType)
|
|
.filter(
|
|
BizCellSampleType.id.in_(type_ids),
|
|
BizCellSampleType.del_flag == 0,
|
|
)
|
|
.all()
|
|
):
|
|
type_map[t.id] = t.type_name or ""
|
|
|
|
express_ids = list(
|
|
{tr.express_company_id for *_, tr in rows if tr and tr.express_company_id}
|
|
)
|
|
express_map: Dict[str, str] = {}
|
|
if express_ids:
|
|
for e in (
|
|
db.query(BizExpressCompany)
|
|
.filter(BizExpressCompany.id.in_(express_ids), BizExpressCompany.del_flag == 0)
|
|
.all()
|
|
):
|
|
express_map[e.id] = e.express_name or ""
|
|
|
|
donation_ids = list({donation.id for _, donation, _, _ in rows if donation and donation.id})
|
|
collect_info_map: Dict[str, Dict[str, str]] = {}
|
|
if donation_ids:
|
|
for coll in (
|
|
db.query(BizResearchDonationCollection)
|
|
.filter(
|
|
BizResearchDonationCollection.research_donation_id.in_(donation_ids),
|
|
BizResearchDonationCollection.del_flag == 0,
|
|
)
|
|
.all()
|
|
):
|
|
collect_info_map[coll.research_donation_id] = {
|
|
"sampleCollectTime": _fmt_dt(coll.sample_collect_time),
|
|
"maternalBloodCollectTime": _fmt_dt(coll.maternal_blood_collect_time),
|
|
}
|
|
|
|
reception_ids = (
|
|
[reception_map[s.id].id for s, *_ in rows if s.id in reception_map]
|
|
if reception_map
|
|
else []
|
|
)
|
|
results_map = (
|
|
check_result_svc.list_results_by_reception_ids(db, reception_ids, tenant_no)
|
|
if reception_ids
|
|
else {}
|
|
)
|
|
|
|
items: List[Dict[str, Any]] = []
|
|
for sample, donation, party, transport in rows:
|
|
item: Dict[str, Any] = {
|
|
"protocolSampleId": sample.id,
|
|
"collectionProtocolId": donation.id,
|
|
"protocolNo": donation.donation_no or "",
|
|
"protocolStatus": donation.donation_status or "",
|
|
"sampleCode": sample.sample_code or "",
|
|
"sampleTypeId": sample.sample_type_id or "",
|
|
"sampleTypeName": type_map.get(sample.sample_type_id or "", ""),
|
|
"collectAmount": _dec(sample.collect_amount),
|
|
"motherName": (party.mother_name if party else "") or "",
|
|
"newbornName": (party.newborn_name if party else "") or "",
|
|
"sampleCollectTime": (
|
|
collect_info_map.get(donation.id, {}).get("sampleCollectTime", "")
|
|
if donation
|
|
else ""
|
|
),
|
|
"maternalBloodCollectTime": (
|
|
collect_info_map.get(donation.id, {}).get("maternalBloodCollectTime", "")
|
|
if donation
|
|
else ""
|
|
),
|
|
"sourceType": "research_donation",
|
|
"sourceTypeLabel": "科研捐赠",
|
|
"infectiousDiseaseReportAttachmentFiles": urls_to_attachment_list(
|
|
donation.infectious_disease_report_urls if donation else None, db, tenant_no
|
|
),
|
|
"etaTime": _fmt_dt(transport.eta_time) if transport and transport.eta_time else "",
|
|
"transportType": (transport.transport_type if transport else "") or "",
|
|
"waybillNo": (transport.waybill_no if transport else "") or "",
|
|
"expressCompanyId": (transport.express_company_id if transport else "") or "",
|
|
"expressCompanyName": express_map.get(
|
|
(transport.express_company_id if transport else "") or "", ""
|
|
),
|
|
"carrierName": (transport.carrier_name if transport else "") or "",
|
|
"carrierPhone": (transport.carrier_phone if transport else "") or "",
|
|
**sample_special_to_dict(sample),
|
|
}
|
|
if reception_map and sample.id in reception_map:
|
|
rec = reception_map[sample.id]
|
|
cr = results_map.get(rec.id)
|
|
if not cr:
|
|
cr = check_result_svc.get_check_results_for_reception(db, rec, tenant_no)
|
|
item.update(_reception_to_dict(rec, check_results=cr))
|
|
item["receiveAttachmentFiles"] = urls_to_attachment_list(
|
|
rec.receive_attachments, db, tenant_no
|
|
)
|
|
items.append(item)
|
|
rec_ids = [str(item.get("id") or "") for item in items if item.get("id")]
|
|
if rec_ids:
|
|
production_map = prod_reception_svc.get_map_by_reception_ids(db, rec_ids, tenant_no)
|
|
for item in items:
|
|
rid = str(item.get("id") or "")
|
|
item["productionReception"] = production_map.get(rid) if rid else None
|
|
return items
|
|
|
|
|
|
def _sample_pending_query(db: Session, tenant_no: str, keyword: Optional[str] = None):
|
|
"""待接收：协议样本且尚无有效接收记录（仅按租户 + 样本编码判定）；排除已作废采集协议单。"""
|
|
rec = BizSampleReception
|
|
received_exists = (
|
|
db.query(rec.id)
|
|
.filter(
|
|
rec.tenant_no == tenant_no,
|
|
rec.del_flag == 0,
|
|
rec.reception_source == "cell_storage",
|
|
rec.sample_code.isnot(None),
|
|
rec.sample_code != "",
|
|
rec.sample_code == BizProtocolSample.sample_code,
|
|
)
|
|
.exists()
|
|
)
|
|
q = _sample_detail_rows_query(db).filter(
|
|
BizProtocolSample.del_flag == 0,
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
BizCollectionProtocol.tenant_no == tenant_no,
|
|
BizCollectionProtocol.protocol_status.notin_(INACTIVE_PROTOCOL_STATUSES),
|
|
~received_exists,
|
|
)
|
|
if keyword and keyword.strip():
|
|
kw = f"%{keyword.strip()}%"
|
|
kw_conds = [
|
|
BizProtocolSample.sample_code.like(kw),
|
|
BizProtocolPartyCustomer.mother_name.like(kw),
|
|
BizProtocolPartyCustomer.newborn_name.like(kw),
|
|
]
|
|
lab_codes = sample_codes_matching_lab_storage_keyword(
|
|
db, tenant_no, keyword, reception_source="cell_storage"
|
|
)
|
|
if lab_codes:
|
|
kw_conds.append(BizProtocolSample.sample_code.in_(lab_codes))
|
|
q = q.filter(or_(*kw_conds))
|
|
return q
|
|
|
|
|
|
def _rows_to_sample_list_items(
|
|
db: Session,
|
|
rows: list,
|
|
tenant_no: str,
|
|
*,
|
|
reception_map: Optional[Dict[str, BizSampleReception]] = None,
|
|
) -> List[Dict[str, Any]]:
|
|
type_ids = list({s.sample_type_id for s, *_ in rows if s.sample_type_id})
|
|
type_map: Dict[str, str] = {}
|
|
if type_ids:
|
|
for t in (
|
|
db.query(BizCellSampleType)
|
|
.filter(
|
|
BizCellSampleType.id.in_(type_ids),
|
|
BizCellSampleType.del_flag == 0,
|
|
)
|
|
.all()
|
|
):
|
|
type_map[t.id] = t.type_name or ""
|
|
|
|
express_ids = list(
|
|
{tr.express_company_id for *_, tr in rows if tr and tr.express_company_id}
|
|
)
|
|
express_map: Dict[str, str] = {}
|
|
if express_ids:
|
|
for e in (
|
|
db.query(BizExpressCompany)
|
|
.filter(BizExpressCompany.id.in_(express_ids), BizExpressCompany.del_flag == 0)
|
|
.all()
|
|
):
|
|
express_map[e.id] = e.express_name or ""
|
|
|
|
protocol_ids = list({protocol.id for _, protocol, _, _ in rows if protocol and protocol.id})
|
|
collect_info_map: Dict[str, Dict[str, str]] = {}
|
|
payment_summary_map: Dict[str, Dict[str, Any]] = {}
|
|
if protocol_ids:
|
|
from app.services.qc_work_order_fee_summary import build_protocol_payment_summaries
|
|
|
|
payment_summary_map = build_protocol_payment_summaries(db, protocol_ids, tenant_no)
|
|
for coll in (
|
|
db.query(BizProtocolCollectionInfo)
|
|
.filter(
|
|
BizProtocolCollectionInfo.collection_protocol_id.in_(protocol_ids),
|
|
BizProtocolCollectionInfo.del_flag == 0,
|
|
)
|
|
.all()
|
|
):
|
|
collect_info_map[coll.collection_protocol_id] = {
|
|
"sampleCollectTime": _fmt_dt(coll.sample_collect_time),
|
|
"maternalBloodCollectTime": _fmt_dt(coll.maternal_blood_collect_time),
|
|
}
|
|
|
|
def _map_rec(sample: BizProtocolSample) -> Optional[BizSampleReception]:
|
|
if not reception_map:
|
|
return None
|
|
code = (sample.sample_code or "").strip()
|
|
if code and code in reception_map:
|
|
return reception_map[code]
|
|
return None
|
|
|
|
matched_recs = [_map_rec(s) for s, *_ in rows]
|
|
reception_ids = [r.id for r in matched_recs if r]
|
|
results_map = (
|
|
check_result_svc.list_results_by_reception_ids(db, reception_ids, tenant_no)
|
|
if reception_ids
|
|
else {}
|
|
)
|
|
|
|
items: List[Dict[str, Any]] = []
|
|
for sample, protocol, party, transport in rows:
|
|
item: Dict[str, Any] = {
|
|
"protocolSampleId": sample.id,
|
|
"collectionProtocolId": protocol.id,
|
|
"protocolNo": protocol.protocol_no or "",
|
|
"protocolStatus": protocol.protocol_status or "",
|
|
"sampleCode": sample.sample_code or "",
|
|
"sampleTypeId": sample.sample_type_id or "",
|
|
"sampleTypeName": type_map.get(sample.sample_type_id or "", ""),
|
|
"collectAmount": _dec(sample.collect_amount),
|
|
"sampleStatus": sample.sample_status or "",
|
|
"motherName": (party.mother_name if party else "") or "",
|
|
"newbornName": (party.newborn_name if party else "") or "",
|
|
"sampleCollectTime": (
|
|
collect_info_map.get(protocol.id, {}).get("sampleCollectTime", "")
|
|
if protocol
|
|
else ""
|
|
),
|
|
"maternalBloodCollectTime": (
|
|
collect_info_map.get(protocol.id, {}).get("maternalBloodCollectTime", "")
|
|
if protocol
|
|
else ""
|
|
),
|
|
"etaTime": _fmt_dt(transport.eta_time) if transport and transport.eta_time else "",
|
|
"transportType": (transport.transport_type if transport else "") or "",
|
|
"waybillNo": (transport.waybill_no if transport else "") or "",
|
|
"expressCompanyId": (transport.express_company_id if transport else "") or "",
|
|
"expressCompanyName": express_map.get(
|
|
(transport.express_company_id if transport else "") or "", ""
|
|
),
|
|
"carrierName": (transport.carrier_name if transport else "") or "",
|
|
"carrierPhone": (transport.carrier_phone if transport else "") or "",
|
|
**sample_special_to_dict(sample),
|
|
"sourceType": "cell_storage",
|
|
"sourceTypeLabel": "细胞存储",
|
|
"healthSurveyAttachmentFiles": urls_to_attachment_list(
|
|
protocol.health_survey_form_urls if protocol else None, db, tenant_no
|
|
),
|
|
"paymentSummary": payment_summary_map.get(protocol.id) if protocol else None,
|
|
}
|
|
rec = _map_rec(sample)
|
|
if rec:
|
|
cr = results_map.get(rec.id)
|
|
if not cr:
|
|
cr = check_result_svc.get_check_results_for_reception(db, rec, tenant_no)
|
|
item.update(_reception_to_dict(rec, check_results=cr))
|
|
item["receiveAttachmentFiles"] = urls_to_attachment_list(
|
|
rec.receive_attachments, db, tenant_no
|
|
)
|
|
items.append(item)
|
|
reception_ids = [str(item.get("id") or "") for item in items if item.get("id")]
|
|
if reception_ids:
|
|
production_map = prod_reception_svc.get_map_by_reception_ids(db, reception_ids, tenant_no)
|
|
for item in items:
|
|
rid = str(item.get("id") or "")
|
|
item["productionReception"] = production_map.get(rid) if rid else None
|
|
return items
|
|
|
|
|
|
def list_pending_protocol_samples(
|
|
db: Session,
|
|
*,
|
|
page: int = 1,
|
|
page_size: int = 10,
|
|
tenant_no: Optional[str] = None,
|
|
keyword: Optional[str] = None,
|
|
source_type: Optional[str] = None,
|
|
eta_time_start: Optional[str] = None,
|
|
eta_time_end: Optional[str] = None,
|
|
user_info=None,
|
|
) -> Tuple[List[Dict[str, Any]], int]:
|
|
if not tenant_no:
|
|
return [], 0
|
|
|
|
all_items: List[Dict[str, Any]] = []
|
|
|
|
if not source_type or source_type == "cell_storage":
|
|
q_base = _sample_pending_query(db, tenant_no, keyword)
|
|
if user_info:
|
|
from app.core.data_scope import apply_data_scope
|
|
q_base = apply_data_scope(q_base, BizProtocolSample, user_info, db, tenant_no)
|
|
cell_rows = (
|
|
q_base.order_by(
|
|
BizProtocolSampleTransport.eta_time.is_(None),
|
|
BizProtocolSampleTransport.eta_time.asc(),
|
|
BizProtocolSample.create_time.asc(),
|
|
)
|
|
.all()
|
|
)
|
|
all_items.extend(_rows_to_sample_list_items(db, cell_rows, tenant_no))
|
|
|
|
if not source_type or source_type == "research_donation":
|
|
q_donation = _donation_sample_pending_query(db, tenant_no, keyword)
|
|
if user_info:
|
|
from app.core.data_scope import apply_data_scope
|
|
q_donation = apply_data_scope(q_donation, BizResearchDonationSample, user_info, db, tenant_no)
|
|
donation_rows = (
|
|
q_donation.order_by(
|
|
BizResearchDonationSampleTransport.eta_time.is_(None),
|
|
BizResearchDonationSampleTransport.eta_time.asc(),
|
|
BizResearchDonationSample.create_time.asc(),
|
|
)
|
|
.all()
|
|
)
|
|
all_items.extend(_donation_rows_to_sample_list_items(db, donation_rows, tenant_no))
|
|
|
|
# 统一按 etaTime 排序（空值排后）
|
|
all_items.sort(key=lambda x: (not x.get("etaTime"), x.get("etaTime") or ""))
|
|
|
|
start_eta = _normalize_date_bound(eta_time_start) if eta_time_start else ""
|
|
end_eta = _normalize_date_bound(eta_time_end, end_of_day=True) if eta_time_end else ""
|
|
if start_eta or end_eta:
|
|
filtered: List[Dict[str, Any]] = []
|
|
for item in all_items:
|
|
eta = (item.get("etaTime") or "").strip()
|
|
if not eta:
|
|
continue
|
|
if start_eta and eta < start_eta:
|
|
continue
|
|
if end_eta and eta > end_eta:
|
|
continue
|
|
filtered.append(item)
|
|
all_items = filtered
|
|
|
|
total = len(all_items)
|
|
page = max(1, page)
|
|
page_size = min(max(1, page_size), 100)
|
|
start = (page - 1) * page_size
|
|
page_items = all_items[start:start + page_size]
|
|
return page_items, int(total)
|
|
|
|
|
|
def list_received_protocol_samples(
|
|
db: Session,
|
|
*,
|
|
page: int = 1,
|
|
page_size: int = 10,
|
|
tenant_no: Optional[str] = None,
|
|
keyword: Optional[str] = None,
|
|
source_type: Optional[str] = None,
|
|
reception_time_start: Optional[str] = None,
|
|
reception_time_end: Optional[str] = None,
|
|
user_info=None,
|
|
) -> Tuple[List[Dict[str, Any]], int]:
|
|
if not tenant_no:
|
|
return [], 0
|
|
|
|
q_rec = db.query(BizSampleReception).filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
)
|
|
if source_type:
|
|
q_rec = q_rec.filter(BizSampleReception.reception_source == source_type)
|
|
|
|
if keyword and keyword.strip():
|
|
kw = f"%{keyword.strip()}%"
|
|
cell_sample_ids_kw = (
|
|
_sample_detail_rows_query(db)
|
|
.with_entities(BizProtocolSample.id)
|
|
.filter(
|
|
BizProtocolSample.del_flag == 0,
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
BizCollectionProtocol.tenant_no == tenant_no,
|
|
or_(
|
|
BizProtocolSample.sample_code.like(kw),
|
|
BizProtocolPartyCustomer.mother_name.like(kw),
|
|
BizProtocolPartyCustomer.newborn_name.like(kw),
|
|
),
|
|
)
|
|
.subquery()
|
|
)
|
|
donation_sample_ids_kw = (
|
|
_donation_sample_detail_rows_query(db)
|
|
.with_entities(_donation_sample_id_col(BizResearchDonationSample.id))
|
|
.filter(
|
|
BizResearchDonationSample.del_flag == 0,
|
|
BizResearchDonationSample.tenant_no == tenant_no,
|
|
BizResearchDonation.tenant_no == tenant_no,
|
|
or_(
|
|
BizResearchDonationSample.sample_code.like(kw),
|
|
BizResearchDonationParty.mother_name.like(kw),
|
|
BizResearchDonationParty.newborn_name.like(kw),
|
|
),
|
|
)
|
|
.subquery()
|
|
)
|
|
q_rec = q_rec.filter(
|
|
or_(
|
|
BizSampleReception.handler_name.like(kw),
|
|
BizSampleReception.lab_storage_code.like(kw),
|
|
BizSampleReception.protocol_sample_id.in_(cell_sample_ids_kw),
|
|
BizSampleReception.donation_sample_id.in_(donation_sample_ids_kw),
|
|
)
|
|
)
|
|
|
|
if user_info:
|
|
from app.core.data_scope import apply_data_scope
|
|
q_rec = apply_data_scope(q_rec, BizSampleReception, user_info, db, tenant_no)
|
|
|
|
if reception_time_start:
|
|
q_rec = q_rec.filter(
|
|
BizSampleReception.reception_time >= _normalize_date_bound(reception_time_start)
|
|
)
|
|
if reception_time_end:
|
|
q_rec = q_rec.filter(
|
|
BizSampleReception.reception_time
|
|
<= _normalize_date_bound(reception_time_end, end_of_day=True)
|
|
)
|
|
|
|
total = q_rec.count()
|
|
page = max(1, page)
|
|
page_size = min(max(1, page_size), 100)
|
|
receptions = (
|
|
q_rec.order_by(BizSampleReception.reception_time.desc())
|
|
.offset((page - 1) * page_size)
|
|
.limit(page_size)
|
|
.all()
|
|
)
|
|
if not receptions:
|
|
return [], int(total)
|
|
|
|
# 按 reception_source 分组查询详情
|
|
cell_receptions = [r for r in receptions if r.reception_source == "cell_storage"]
|
|
donation_receptions = [r for r in receptions if r.reception_source == "research_donation"]
|
|
|
|
items: List[Dict[str, Any]] = []
|
|
|
|
if cell_receptions:
|
|
cell_reception_map: Dict[str, BizSampleReception] = {}
|
|
seen_codes: set = set()
|
|
cell_rows: list = []
|
|
for rec in cell_receptions:
|
|
code = _reception_sample_code(db, rec)
|
|
if not code:
|
|
continue
|
|
cell_reception_map[code] = rec
|
|
if code in seen_codes:
|
|
continue
|
|
seen_codes.add(code)
|
|
rows = (
|
|
_sample_detail_rows_query(db)
|
|
.filter(
|
|
BizProtocolSample.sample_code == code,
|
|
BizProtocolSample.del_flag == 0,
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
BizCollectionProtocol.tenant_no == tenant_no,
|
|
)
|
|
.all()
|
|
)
|
|
cell_rows.extend(rows)
|
|
items.extend(
|
|
_rows_to_sample_list_items(db, cell_rows, tenant_no, reception_map=cell_reception_map)
|
|
)
|
|
|
|
if donation_receptions:
|
|
donation_sample_ids = [r.donation_sample_id for r in donation_receptions]
|
|
donation_reception_map = {r.donation_sample_id: r for r in donation_receptions}
|
|
donation_rows = (
|
|
_donation_sample_detail_rows_query(db)
|
|
.filter(_donation_sample_id_col(BizResearchDonationSample.id).in_(donation_sample_ids))
|
|
.all()
|
|
)
|
|
donation_row_by_sample = {row[0].id: row for row in donation_rows}
|
|
donation_ordered = [donation_row_by_sample[sid] for sid in donation_sample_ids if sid in donation_row_by_sample]
|
|
items.extend(_donation_rows_to_sample_list_items(db, donation_ordered, tenant_no, reception_map=donation_reception_map))
|
|
|
|
# 按 reception_time DESC 排序
|
|
items.sort(key=lambda x: x.get("receptionTime") or "", reverse=True)
|
|
return items, int(total)
|
|
|
|
|
|
def _finalize_reception_detail(db: Session, item: Optional[Dict[str, Any]], tenant_no: str) -> Optional[Dict[str, Any]]:
|
|
if not item:
|
|
return item
|
|
return _attach_production_reception(db, item, tenant_no)
|
|
|
|
|
|
def get_reception_by_id(
|
|
db: Session, reception_id: str, tenant_no: str
|
|
) -> Optional[Dict[str, Any]]:
|
|
rec = (
|
|
db.query(BizSampleReception)
|
|
.filter(
|
|
BizSampleReception.id == reception_id,
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if not rec:
|
|
return None
|
|
if (rec.reception_source or "cell_storage") == "research_donation":
|
|
dsid = rec.donation_sample_id or ""
|
|
if not dsid:
|
|
cr = check_result_svc.get_check_results_for_reception(db, rec, tenant_no)
|
|
out = _reception_to_dict(rec, check_results=cr)
|
|
out["sourceType"] = "research_donation"
|
|
out["sourceTypeLabel"] = "科研捐赠"
|
|
out["receiveAttachmentFiles"] = urls_to_attachment_list(
|
|
rec.receive_attachments, db, tenant_no
|
|
)
|
|
return _finalize_reception_detail(db, out, tenant_no)
|
|
rows = (
|
|
_donation_sample_detail_rows_query(db)
|
|
.filter(_donation_sample_id_col(BizResearchDonationSample.id) == dsid)
|
|
.all()
|
|
)
|
|
if not rows:
|
|
cr = check_result_svc.get_check_results_for_reception(db, rec, tenant_no)
|
|
out = _reception_to_dict(rec, check_results=cr)
|
|
out["sourceType"] = "research_donation"
|
|
out["sourceTypeLabel"] = "科研捐赠"
|
|
out["receiveAttachmentFiles"] = urls_to_attachment_list(
|
|
rec.receive_attachments, db, tenant_no
|
|
)
|
|
return _finalize_reception_detail(db, out, tenant_no)
|
|
items = _donation_rows_to_sample_list_items(
|
|
db, rows, tenant_no, reception_map={dsid: rec}
|
|
)
|
|
return items[0] if items else None
|
|
|
|
code = _reception_sample_code(db, rec)
|
|
if code:
|
|
rows = (
|
|
_sample_detail_rows_query(db)
|
|
.filter(
|
|
BizProtocolSample.sample_code == code,
|
|
BizProtocolSample.del_flag == 0,
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
)
|
|
.all()
|
|
)
|
|
if rows:
|
|
items = _rows_to_sample_list_items(
|
|
db, rows, tenant_no, reception_map={code: rec}
|
|
)
|
|
return items[0] if items else None
|
|
cr = check_result_svc.get_check_results_for_reception(db, rec, tenant_no)
|
|
out = _reception_to_dict(rec, check_results=cr)
|
|
out["receiveAttachmentFiles"] = urls_to_attachment_list(
|
|
rec.receive_attachments, db, tenant_no
|
|
)
|
|
return _finalize_reception_detail(db, out, tenant_no)
|
|
|
|
|
|
def create_sample_reception(
|
|
db: Session,
|
|
*,
|
|
reception_source: str = "cell_storage",
|
|
protocol_sample_id: Optional[str] = None,
|
|
donation_sample_id: Optional[str] = None,
|
|
research_donation_id: Optional[str] = None,
|
|
tenant_no: str,
|
|
user_id: Optional[str],
|
|
create_dept_id: str = "",
|
|
reception_time: datetime,
|
|
handler_user_id: Optional[str],
|
|
handler_name: Optional[str],
|
|
receive_temperature: Optional[str],
|
|
volume_ml: Optional[float] = None,
|
|
net_weight_g: Optional[float] = None,
|
|
checks: List[Dict[str, Any]],
|
|
accept_conclusion: str,
|
|
receive_attachments: Optional[str],
|
|
remark: Optional[str],
|
|
lab_storage_code: Optional[str] = None,
|
|
) -> BizSampleReception:
|
|
source = (reception_source or "").strip()
|
|
if source not in ("cell_storage", "research_donation"):
|
|
raise ValueError("来源类型无效")
|
|
|
|
# 按来源校验样本并获取关联 ID
|
|
if source == "cell_storage":
|
|
if not protocol_sample_id:
|
|
raise ValueError("细胞存储来源必须传入 protocolSampleId")
|
|
sample = (
|
|
db.query(BizProtocolSample)
|
|
.filter(
|
|
BizProtocolSample.id == protocol_sample_id,
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
BizProtocolSample.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if not sample:
|
|
raise ValueError("样本不存在")
|
|
_sample_code = (sample.sample_code or "").strip()
|
|
if not _sample_code:
|
|
raise ValueError("样本编码不能为空")
|
|
exists = (
|
|
db.query(BizSampleReception)
|
|
.filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.reception_source == "cell_storage",
|
|
BizSampleReception.sample_code == _sample_code,
|
|
)
|
|
.first()
|
|
)
|
|
if exists:
|
|
raise ValueError("该样本已登记接收，请勿重复操作")
|
|
collection_protocol_id = sample.collection_protocol_id
|
|
_donation_sample_id = None
|
|
_research_donation_id = None
|
|
_sample_code = sample.sample_code or ""
|
|
else:
|
|
# research_donation
|
|
if not donation_sample_id:
|
|
raise ValueError("科研捐赠来源必须传入 donationSampleId")
|
|
dsample = (
|
|
db.query(BizResearchDonationSample)
|
|
.filter(
|
|
_donation_sample_id_col(BizResearchDonationSample.id) == donation_sample_id,
|
|
BizResearchDonationSample.tenant_no == tenant_no,
|
|
BizResearchDonationSample.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if not dsample:
|
|
raise ValueError("捐赠样本不存在")
|
|
_sample_code = (dsample.sample_code or "").strip()
|
|
if not _sample_code:
|
|
raise ValueError("样本编码不能为空")
|
|
exists = (
|
|
db.query(BizSampleReception)
|
|
.filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.reception_source == "research_donation",
|
|
BizSampleReception.sample_code == _sample_code,
|
|
)
|
|
.first()
|
|
)
|
|
if exists:
|
|
raise ValueError("该样本已登记接收，请勿重复操作")
|
|
collection_protocol_id = ""
|
|
protocol_sample_id = ""
|
|
_donation_sample_id = donation_sample_id
|
|
_research_donation_id = dsample.research_donation_id
|
|
|
|
conclusion = (accept_conclusion or "").strip()
|
|
if conclusion not in ("js", "bj"):
|
|
raise ValueError("接收结论无效")
|
|
|
|
from app.crud import sample_reception_lab_code as lab_code_svc
|
|
|
|
resolved_lab_code: Optional[str] = None
|
|
code_raw = (lab_storage_code or "").strip()
|
|
if conclusion == "js":
|
|
resolved_lab_code = lab_code_svc.validate_lab_storage_code(code_raw)
|
|
lab_code_svc.assert_lab_storage_code_available(
|
|
db, tenant_no=tenant_no, code=resolved_lab_code
|
|
)
|
|
elif code_raw:
|
|
resolved_lab_code = lab_code_svc.validate_lab_storage_code(code_raw)
|
|
lab_code_svc.assert_lab_storage_code_available(
|
|
db, tenant_no=tenant_no, code=resolved_lab_code
|
|
)
|
|
|
|
handler_display = (handler_name or "").strip()
|
|
if handler_user_id and not handler_display:
|
|
u = db.query(SysUser).filter(SysUser.id == handler_user_id).first()
|
|
if u:
|
|
handler_display = u.nick_name or ""
|
|
|
|
obj = BizSampleReception(
|
|
id=generate_uuid(),
|
|
protocol_sample_id=protocol_sample_id or "",
|
|
collection_protocol_id=collection_protocol_id or "",
|
|
reception_source=source,
|
|
donation_sample_id=_donation_sample_id,
|
|
research_donation_id=_research_donation_id,
|
|
sample_code=_sample_code or None,
|
|
lab_storage_code=resolved_lab_code,
|
|
reception_time=reception_time,
|
|
handler_user_id=handler_user_id,
|
|
handler_name=handler_display or None,
|
|
receive_temperature=(receive_temperature or "").strip() or None,
|
|
check_packaging=0,
|
|
check_sealing=0,
|
|
check_no_damage=0,
|
|
check_no_leakage=0,
|
|
check_color=0,
|
|
check_tying_rope=0,
|
|
check_no_stratification=0,
|
|
accept_conclusion=conclusion,
|
|
receive_attachments=(receive_attachments or "").strip() or None,
|
|
tenant_no=tenant_no,
|
|
remark=(remark or "").strip() or None,
|
|
create_by=user_id,
|
|
create_dept_id=create_dept_id or "",
|
|
update_by=user_id,
|
|
)
|
|
db.add(obj)
|
|
db.flush()
|
|
check_result_svc.save_reception_check_results(
|
|
db,
|
|
reception_id=obj.id,
|
|
tenant_no=tenant_no,
|
|
user_id=user_id,
|
|
create_dept_id=create_dept_id or "",
|
|
checks=checks,
|
|
)
|
|
from app.models.models import BizSampleReceptionCheckResult
|
|
|
|
saved_rows = (
|
|
db.query(BizSampleReceptionCheckResult)
|
|
.filter(
|
|
BizSampleReceptionCheckResult.reception_id == obj.id,
|
|
BizSampleReceptionCheckResult.del_flag == 0,
|
|
)
|
|
.all()
|
|
)
|
|
check_result_svc.sync_legacy_columns_from_results(db, obj, saved_rows)
|
|
db.add(obj)
|
|
if conclusion == "js":
|
|
from app.crud import qc_work_order as qc_wo_svc
|
|
|
|
qc_wo_svc.create_work_order_from_reception(
|
|
db,
|
|
reception=obj,
|
|
tenant_no=tenant_no,
|
|
user_id=user_id,
|
|
create_dept_id=create_dept_id or "",
|
|
)
|
|
from app.crud import production_work_order as prod_wo_svc
|
|
|
|
prod_wo_svc.create_production_work_order_from_reception(
|
|
db,
|
|
reception=obj,
|
|
tenant_no=tenant_no,
|
|
user_id=user_id,
|
|
create_dept_id=create_dept_id or "",
|
|
)
|
|
from app.crud import storage_release_review as srr_svc
|
|
|
|
_srr_protocol_no = ""
|
|
_srr_sample_name = ""
|
|
_src = getattr(obj, "reception_source", "cell_storage") or "cell_storage"
|
|
if _src == "research_donation":
|
|
_d = (
|
|
db.query(BizResearchDonation)
|
|
.filter(
|
|
BizResearchDonation.id == (obj.research_donation_id or ""),
|
|
BizResearchDonation.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if _d:
|
|
_srr_protocol_no = _d.donation_no or ""
|
|
_ds = (
|
|
db.query(BizResearchDonationSample)
|
|
.filter(
|
|
BizResearchDonationSample.id == (obj.donation_sample_id or ""),
|
|
BizResearchDonationSample.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if _ds:
|
|
_srr_sample_name = _ds.sample_code or ""
|
|
else:
|
|
_p = (
|
|
db.query(BizCollectionProtocol)
|
|
.filter(
|
|
BizCollectionProtocol.id == obj.collection_protocol_id,
|
|
BizCollectionProtocol.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if _p:
|
|
_srr_protocol_no = _p.protocol_no or ""
|
|
_s = (
|
|
db.query(BizProtocolSample)
|
|
.filter(
|
|
BizProtocolSample.id == obj.protocol_sample_id,
|
|
BizProtocolSample.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if _s:
|
|
_srr_sample_name = _s.sample_code or ""
|
|
srr_svc.create_review_from_reception(
|
|
db,
|
|
reception=obj,
|
|
protocol_no=_srr_protocol_no,
|
|
sample_name=_srr_sample_name,
|
|
tenant_no=tenant_no,
|
|
user_id=user_id,
|
|
create_dept_id=create_dept_id or "",
|
|
)
|
|
db.commit()
|
|
db.refresh(obj)
|
|
return obj
|
|
|
|
|
|
def _can_edit_sample_reception(
|
|
db: Session,
|
|
user_id: Optional[str],
|
|
rec: BizSampleReception,
|
|
) -> bool:
|
|
"""已接收记录：仅经手人或「质量主管」角色可编辑。"""
|
|
uid = (user_id or "").strip()
|
|
if not uid:
|
|
return False
|
|
handler_id = (rec.handler_user_id or "").strip()
|
|
if handler_id and handler_id == uid:
|
|
return True
|
|
return SAMPLE_RECEPTION_EDIT_ROLE in get_user_role_names(db, uid)
|
|
|
|
|
|
def update_sample_reception(
|
|
db: Session,
|
|
*,
|
|
reception_id: str,
|
|
tenant_no: str,
|
|
user_id: Optional[str],
|
|
reception_time: datetime,
|
|
handler_user_id: Optional[str],
|
|
handler_name: Optional[str],
|
|
receive_temperature: Optional[str],
|
|
volume_ml: Optional[float] = None,
|
|
net_weight_g: Optional[float] = None,
|
|
checks: List[Dict[str, Any]],
|
|
accept_conclusion: str,
|
|
receive_attachments: Optional[str],
|
|
remark: Optional[str],
|
|
lab_storage_code: Optional[str] = None,
|
|
) -> BizSampleReception:
|
|
"""更新已登记样本接收信息（不重复创建质检/生产工单）。"""
|
|
rec = (
|
|
db.query(BizSampleReception)
|
|
.filter(
|
|
BizSampleReception.id == reception_id,
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
if not rec:
|
|
raise ValueError("接收记录不存在")
|
|
|
|
if not _can_edit_sample_reception(db, user_id, rec):
|
|
raise PermissionError("仅该接收记录的经手人或质量主管可编辑")
|
|
|
|
conclusion = (accept_conclusion or "").strip()
|
|
if conclusion not in ("js", "bj"):
|
|
raise ValueError("接收结论无效")
|
|
|
|
from app.crud import sample_reception_lab_code as lab_code_svc
|
|
|
|
code_raw = (lab_storage_code or "").strip()
|
|
if conclusion == "js":
|
|
parsed_code = lab_code_svc.validate_lab_storage_code(code_raw)
|
|
lab_code_svc.assert_lab_storage_code_available(
|
|
db,
|
|
tenant_no=tenant_no,
|
|
code=parsed_code,
|
|
exclude_reception_id=reception_id,
|
|
)
|
|
rec.lab_storage_code = parsed_code
|
|
elif code_raw:
|
|
parsed_code = lab_code_svc.validate_lab_storage_code(code_raw)
|
|
lab_code_svc.assert_lab_storage_code_available(
|
|
db,
|
|
tenant_no=tenant_no,
|
|
code=parsed_code,
|
|
exclude_reception_id=reception_id,
|
|
)
|
|
rec.lab_storage_code = parsed_code
|
|
else:
|
|
rec.lab_storage_code = None
|
|
|
|
handler_display = (handler_name or "").strip()
|
|
if handler_user_id and not handler_display:
|
|
u = db.query(SysUser).filter(SysUser.id == handler_user_id).first()
|
|
if u:
|
|
handler_display = u.nick_name or ""
|
|
|
|
rec.reception_time = reception_time
|
|
rec.handler_user_id = handler_user_id
|
|
rec.handler_name = handler_display or None
|
|
rec.receive_temperature = (receive_temperature or "").strip() or None
|
|
rec.accept_conclusion = conclusion
|
|
rec.receive_attachments = (receive_attachments or "").strip() or None
|
|
rec.remark = (remark or "").strip() or None
|
|
rec.update_by = user_id
|
|
rec.update_time = datetime.now()
|
|
|
|
check_result_svc.soft_delete_reception_check_results(
|
|
db,
|
|
reception_id=reception_id,
|
|
tenant_no=tenant_no,
|
|
user_id=user_id,
|
|
)
|
|
check_result_svc.save_reception_check_results(
|
|
db,
|
|
reception_id=reception_id,
|
|
tenant_no=tenant_no,
|
|
user_id=user_id,
|
|
create_dept_id=rec.create_dept_id or "",
|
|
checks=checks,
|
|
)
|
|
from app.models.models import BizSampleReceptionCheckResult
|
|
|
|
saved_rows = (
|
|
db.query(BizSampleReceptionCheckResult)
|
|
.filter(
|
|
BizSampleReceptionCheckResult.reception_id == reception_id,
|
|
BizSampleReceptionCheckResult.del_flag == 0,
|
|
)
|
|
.all()
|
|
)
|
|
check_result_svc.sync_legacy_columns_from_results(db, rec, saved_rows)
|
|
db.add(rec)
|
|
db.commit()
|
|
db.refresh(rec)
|
|
return rec
|
|
|
|
|
|
def _attach_production_reception(
|
|
db: Session,
|
|
item: Dict[str, Any],
|
|
tenant_no: str,
|
|
*,
|
|
production_map: Optional[Dict[str, Dict[str, Any]]] = None,
|
|
) -> Dict[str, Any]:
|
|
rid = str(item.get("id") or "").strip()
|
|
if not rid:
|
|
item["productionReception"] = None
|
|
return item
|
|
if production_map is not None:
|
|
item["productionReception"] = production_map.get(rid)
|
|
else:
|
|
item["productionReception"] = prod_reception_svc.get_by_reception_id(db, rid, tenant_no)
|
|
return item
|
|
|
|
|
|
def _reception_detail_item(
|
|
db: Session,
|
|
rec: BizSampleReception,
|
|
tenant_no: str,
|
|
*,
|
|
results_map: Optional[Dict[str, List[Dict[str, Any]]]] = None,
|
|
) -> Dict[str, Any]:
|
|
cr = (results_map or {}).get(rec.id)
|
|
if not cr:
|
|
cr = check_result_svc.get_check_results_for_reception(db, rec, tenant_no)
|
|
item = _reception_to_dict(rec, check_results=cr)
|
|
item["receiveAttachmentFiles"] = urls_to_attachment_list(
|
|
rec.receive_attachments, db, tenant_no
|
|
)
|
|
return item
|
|
|
|
|
|
def _reception_detail_item_with_production(
|
|
db: Session,
|
|
rec: BizSampleReception,
|
|
tenant_no: str,
|
|
*,
|
|
results_map: Optional[Dict[str, List[Dict[str, Any]]]] = None,
|
|
production_map: Optional[Dict[str, Dict[str, Any]]] = None,
|
|
) -> Dict[str, Any]:
|
|
item = _reception_detail_item(db, rec, tenant_no, results_map=results_map)
|
|
return _attach_production_reception(db, item, tenant_no, production_map=production_map)
|
|
|
|
|
|
def _reception_sample_code(db: Session, rec: BizSampleReception) -> str:
|
|
code = (rec.sample_code or "").strip()
|
|
if code:
|
|
return code
|
|
sid = (rec.protocol_sample_id or "").strip()
|
|
if not sid:
|
|
return ""
|
|
row = (
|
|
db.query(BizProtocolSample.sample_code)
|
|
.filter(BizProtocolSample.id == sid)
|
|
.first()
|
|
)
|
|
return (row.sample_code or "").strip() if row else ""
|
|
|
|
|
|
def get_receptions_map_for_samples(
|
|
db: Session,
|
|
tenant_no: str,
|
|
*,
|
|
sample_code_by_id: Dict[str, str],
|
|
) -> Dict[str, Dict[str, Any]]:
|
|
"""按租户 + 样本编码关联接收记录（key 为当前协议样本 id），供协议单查看弹窗使用。"""
|
|
if not tenant_no or not sample_code_by_id:
|
|
return {}
|
|
codes = list({(c or "").strip() for c in sample_code_by_id.values() if (c or "").strip()})
|
|
if not codes:
|
|
return {}
|
|
rows = (
|
|
db.query(BizSampleReception)
|
|
.filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.reception_source == "cell_storage",
|
|
BizSampleReception.sample_code.in_(codes),
|
|
)
|
|
.all()
|
|
)
|
|
if not rows:
|
|
return {}
|
|
code_to_rec: Dict[str, BizSampleReception] = {}
|
|
for rec in rows:
|
|
code = _reception_sample_code(db, rec)
|
|
if code and code not in code_to_rec:
|
|
code_to_rec[code] = rec
|
|
reception_ids = [r.id for r in code_to_rec.values()]
|
|
results_map = check_result_svc.list_results_by_reception_ids(
|
|
db, reception_ids, tenant_no
|
|
)
|
|
production_map = prod_reception_svc.get_map_by_reception_ids(db, reception_ids, tenant_no)
|
|
out: Dict[str, Dict[str, Any]] = {}
|
|
for sid, raw_code in sample_code_by_id.items():
|
|
code = (raw_code or "").strip()
|
|
if not code:
|
|
continue
|
|
rec = code_to_rec.get(code)
|
|
if not rec:
|
|
continue
|
|
out[sid] = _reception_detail_item_with_production(
|
|
db, rec, tenant_no, results_map=results_map, production_map=production_map
|
|
)
|
|
return out
|
|
|
|
|
|
def get_receptions_map_for_donation_samples(
|
|
db: Session,
|
|
sample_ids: List[str],
|
|
tenant_no: str,
|
|
) -> Dict[str, Dict[str, Any]]:
|
|
"""按捐赠样本 ID 批量查询接收记录，供捐赠协议单查看弹窗使用。"""
|
|
if not sample_ids or not tenant_no:
|
|
return {}
|
|
rows = (
|
|
db.query(BizSampleReception)
|
|
.filter(
|
|
BizSampleReception.donation_sample_id.in_(sample_ids),
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
)
|
|
.all()
|
|
)
|
|
if not rows:
|
|
return {}
|
|
reception_ids = [r.id for r in rows]
|
|
results_map = check_result_svc.list_results_by_reception_ids(db, reception_ids, tenant_no)
|
|
production_map = prod_reception_svc.get_map_by_reception_ids(db, reception_ids, tenant_no)
|
|
out: Dict[str, Dict[str, Any]] = {}
|
|
for rec in rows:
|
|
sid = rec.donation_sample_id
|
|
if not sid:
|
|
continue
|
|
cr = results_map.get(rec.id)
|
|
if not cr:
|
|
cr = check_result_svc.get_check_results_for_reception(db, rec, tenant_no)
|
|
item = _reception_to_dict(rec, check_results=cr)
|
|
item["receiveAttachmentFiles"] = urls_to_attachment_list(
|
|
rec.receive_attachments, db, tenant_no
|
|
)
|
|
_attach_production_reception(db, item, tenant_no, production_map=production_map)
|
|
out[sid] = item
|
|
return out
|
|
|
|
|
|
def get_lab_storage_code_map_by_sample_codes(
|
|
db: Session,
|
|
tenant_no: str,
|
|
sample_codes: List[str],
|
|
*,
|
|
reception_source: Optional[str] = None,
|
|
) -> Dict[str, str]:
|
|
"""按样本编码查询已接收(js)记录的实验室入库编码；同码多条时取最新接收时间。"""
|
|
codes = list({(c or "").strip() for c in sample_codes if (c or "").strip()})
|
|
if not codes or not tenant_no:
|
|
return {}
|
|
q = db.query(BizSampleReception.sample_code, BizSampleReception.lab_storage_code).filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.sample_code.in_(codes),
|
|
BizSampleReception.accept_conclusion == "js",
|
|
BizSampleReception.lab_storage_code.isnot(None),
|
|
BizSampleReception.lab_storage_code != "",
|
|
)
|
|
if reception_source:
|
|
q = q.filter(BizSampleReception.reception_source == reception_source)
|
|
result: Dict[str, str] = {}
|
|
for row in q.order_by(BizSampleReception.reception_time.desc()).all():
|
|
sc = (row.sample_code or "").strip()
|
|
lk = (row.lab_storage_code or "").strip()
|
|
if sc and lk and sc not in result:
|
|
result[sc] = lk
|
|
|
|
missing = [c for c in codes if c not in result]
|
|
if missing:
|
|
join_q = (
|
|
db.query(BizProtocolSample.sample_code, BizSampleReception.lab_storage_code)
|
|
.join(
|
|
BizSampleReception,
|
|
and_(
|
|
BizSampleReception.protocol_sample_id == BizProtocolSample.id,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.accept_conclusion == "js",
|
|
BizSampleReception.lab_storage_code.isnot(None),
|
|
BizSampleReception.lab_storage_code != "",
|
|
),
|
|
)
|
|
.filter(
|
|
BizProtocolSample.sample_code.in_(missing),
|
|
BizProtocolSample.del_flag == 0,
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
)
|
|
)
|
|
if reception_source:
|
|
join_q = join_q.filter(BizSampleReception.reception_source == reception_source)
|
|
for row in join_q.order_by(BizSampleReception.reception_time.desc()).all():
|
|
sc = (row.sample_code or "").strip()
|
|
lk = (row.lab_storage_code or "").strip()
|
|
if sc and lk and sc not in result:
|
|
result[sc] = lk
|
|
return result
|
|
|
|
|
|
def format_lab_storage_codes_csv(
|
|
sample_codes: List[str],
|
|
code_map: Dict[str, str],
|
|
) -> str:
|
|
"""与 sampleCodes 顺序对齐，逗号拼接入库编码（无则空串）。"""
|
|
parts: List[str] = []
|
|
for raw in sample_codes:
|
|
c = (raw or "").strip()
|
|
if not c:
|
|
continue
|
|
parts.append((code_map.get(c) or "").strip())
|
|
return ",".join(parts)
|
|
|
|
|
|
def _lab_storage_keyword_like(keyword: str) -> Optional[str]:
|
|
kw = (keyword or "").strip()
|
|
if not kw:
|
|
return None
|
|
return f"%{kw}%"
|
|
|
|
|
|
def sample_codes_matching_lab_storage_keyword(
|
|
db: Session,
|
|
tenant_no: str,
|
|
keyword: str,
|
|
*,
|
|
reception_source: Optional[str] = None,
|
|
) -> List[str]:
|
|
"""关键词模糊匹配实验室入库编码，返回关联样本编码列表。"""
|
|
like = _lab_storage_keyword_like(keyword)
|
|
if not like or not tenant_no:
|
|
return []
|
|
q = db.query(BizSampleReception.sample_code).filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.lab_storage_code.isnot(None),
|
|
BizSampleReception.lab_storage_code != "",
|
|
BizSampleReception.lab_storage_code.like(like),
|
|
BizSampleReception.sample_code.isnot(None),
|
|
BizSampleReception.sample_code != "",
|
|
)
|
|
if reception_source:
|
|
q = q.filter(BizSampleReception.reception_source == reception_source)
|
|
return list({(row[0] or "").strip() for row in q.all() if (row[0] or "").strip()})
|
|
|
|
|
|
def reception_ids_matching_lab_storage_keyword(
|
|
db: Session,
|
|
tenant_no: str,
|
|
keyword: str,
|
|
*,
|
|
reception_source: Optional[str] = None,
|
|
) -> List[str]:
|
|
"""关键词模糊匹配实验室入库编码，返回接收记录 id 列表。"""
|
|
like = _lab_storage_keyword_like(keyword)
|
|
if not like or not tenant_no:
|
|
return []
|
|
q = db.query(BizSampleReception.id).filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.lab_storage_code.isnot(None),
|
|
BizSampleReception.lab_storage_code != "",
|
|
BizSampleReception.lab_storage_code.like(like),
|
|
)
|
|
if reception_source:
|
|
q = q.filter(BizSampleReception.reception_source == reception_source)
|
|
return [row[0] for row in q.all() if row[0]]
|
|
|
|
|
|
def protocol_ids_matching_lab_storage_keyword(
|
|
db: Session,
|
|
tenant_no: str,
|
|
keyword: str,
|
|
) -> List[str]:
|
|
"""入库编码关键词 → 含匹配样本的采集协议 id。"""
|
|
codes = sample_codes_matching_lab_storage_keyword(
|
|
db, tenant_no, keyword, reception_source="cell_storage"
|
|
)
|
|
if not codes:
|
|
return []
|
|
rows = (
|
|
db.query(BizProtocolSample.collection_protocol_id)
|
|
.filter(
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
BizProtocolSample.del_flag == 0,
|
|
BizProtocolSample.sample_code.in_(codes),
|
|
)
|
|
.distinct()
|
|
.all()
|
|
)
|
|
return [r[0] for r in rows if r[0]]
|
|
|
|
|
|
def donation_ids_matching_lab_storage_keyword(
|
|
db: Session,
|
|
tenant_no: str,
|
|
keyword: str,
|
|
) -> List[str]:
|
|
"""入库编码关键词 → 含匹配样本的捐赠协议 id。"""
|
|
codes = sample_codes_matching_lab_storage_keyword(
|
|
db, tenant_no, keyword, reception_source="research_donation"
|
|
)
|
|
if not codes:
|
|
return []
|
|
rows = (
|
|
db.query(BizResearchDonationSample.research_donation_id)
|
|
.filter(
|
|
BizResearchDonationSample.tenant_no == tenant_no,
|
|
BizResearchDonationSample.del_flag == 0,
|
|
BizResearchDonationSample.sample_code.in_(codes),
|
|
)
|
|
.distinct()
|
|
.all()
|
|
)
|
|
return [r[0] for r in rows if r[0]]
|
|
|
|
|
|
def get_reception_by_sample_code(
|
|
db: Session,
|
|
sample_code: str,
|
|
source_type: Optional[str],
|
|
tenant_no: str,
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
根据样本编码查询接收记录及样本信息。
|
|
- 有记录 → 返回 {receptionId, sampleInfo}
|
|
- 无记录 → 返回 {receptionId: null, sampleInfo}
|
|
"""
|
|
code = (sample_code or "").strip()
|
|
if not code:
|
|
return {"receptionId": None, "sampleInfo": None}
|
|
|
|
source = (source_type or "cell_storage").strip()
|
|
# 先查询是否有已接收记录
|
|
q_rec = db.query(BizSampleReception).filter(
|
|
BizSampleReception.tenant_no == tenant_no,
|
|
BizSampleReception.del_flag == 0,
|
|
BizSampleReception.sample_code == code,
|
|
)
|
|
if source:
|
|
q_rec = q_rec.filter(BizSampleReception.reception_source == source)
|
|
|
|
reception = q_rec.order_by(BizSampleReception.reception_time.desc()).first()
|
|
|
|
if source == "research_donation":
|
|
# 捐赠样本查询
|
|
rows = (
|
|
_donation_sample_detail_rows_query(db)
|
|
.filter(
|
|
_donation_sample_id_col(BizResearchDonationSample.sample_code) == code,
|
|
BizResearchDonationSample.del_flag == 0,
|
|
BizResearchDonationSample.tenant_no == tenant_no,
|
|
)
|
|
.all()
|
|
)
|
|
sample_info_list = _donation_rows_to_sample_list_items(db, rows, tenant_no)
|
|
sample_info = sample_info_list[0] if sample_info_list else None
|
|
else:
|
|
# 细胞存储样本查询
|
|
rows = (
|
|
_sample_detail_rows_query(db)
|
|
.filter(
|
|
BizProtocolSample.sample_code == code,
|
|
BizProtocolSample.del_flag == 0,
|
|
BizProtocolSample.tenant_no == tenant_no,
|
|
)
|
|
.all()
|
|
)
|
|
if rows:
|
|
if reception:
|
|
sample_info_list = _rows_to_sample_list_items(
|
|
db, rows, tenant_no, reception_map={code: reception}
|
|
)
|
|
else:
|
|
sample_info_list = _rows_to_sample_list_items(db, rows, tenant_no)
|
|
sample_info = sample_info_list[0] if sample_info_list else None
|
|
else:
|
|
sample_info = None
|
|
|
|
return {
|
|
"receptionId": reception.id if reception else None,
|
|
"sampleInfo": sample_info,
|
|
"productionReception": (
|
|
prod_reception_svc.get_by_reception_id(db, reception.id, tenant_no) if reception else None
|
|
),
|
|
}
|