fix(lab-sample): 恢复 Excel 真菌等补充字段并同步质检

纠正入库编码被误软删问题,恢复可关联样本的补充字段;列表接入质检填空,并补充间充质科研捐献接收行。

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
zlwuandCursor committed 2026-11-12 17:32:09 +08:00
1 parent c0aba421da
commit e7b3876db6
4 files changed
+370 -17

No files matched your search

+308 -17
View File
@@ -385,6 +385,196 @@ def sync_legacy_supplement_to_processing(
production.update_time = datetime.now()
def _extra_field_keys(dataset: str) -> set:
return {f.key for f in fields_for(dataset) if f.source == SOURCE_EXTRA}
def _live_sample_codes(db: Session, tenant_no: str) -> set:
codes = set()
for (code,) in (
db.query(BizProtocolSample.sample_code)
.filter(
BizProtocolSample.tenant_no == tenant_no,
BizProtocolSample.del_flag == 0,
BizProtocolSample.sample_code.isnot(None),
BizProtocolSample.sample_code != "",
)
.all()
):
if code:
codes.add(str(code).strip())
for lab_code, sample_code in (
db.query(BizSampleReception.lab_storage_code, BizSampleReception.sample_code)
.filter(
BizSampleReception.tenant_no == tenant_no,
BizSampleReception.del_flag == 0,
)
.all()
):
if lab_code:
codes.add(str(lab_code).strip())
if sample_code:
codes.add(str(sample_code).strip())
return codes
def restore_retired_excel_extras(
db: Session,
*,
tenant_no: str,
operator_id: str = "restore-excel-extra",
) -> Dict[str, Any]:
"""恢复被误删的 Excel 补充字段(真菌/内毒素等),并同步旧台账。
当初按协议样本号匹配软删,但 Excel 多用实验室入库编码,导致真实样本补录被删。
仅恢复仍能关联到当前有效协议/接收样本编码的补充字段。
"""
from app.models.models import BizLabSampleProcessingExtra
live_codes = _live_sample_codes(db, tenant_no)
retired = (
db.query(BizLabSampleProcessingExtra)
.filter(
BizLabSampleProcessingExtra.tenant_no == tenant_no,
BizLabSampleProcessingExtra.del_flag == 1,
BizLabSampleProcessingExtra.update_by == "retire-excel-demo",
)
.all()
)
restored = 0
skipped = 0
field_hits = 0
for row in retired:
code = (row.sample_code or "").strip()
if not code or code not in live_codes:
skipped += 1
continue
payload = _parse_extra(row.payload_json)
allowed = _extra_field_keys(row.dataset)
extra_only = {
k: v
for k, v in payload.items()
if k in allowed and not _is_blank(v)
}
if not extra_only:
skipped += 1
continue
upsert_extra_payload(
db,
tenant_no=tenant_no,
dataset=row.dataset,
sample_code=code,
payload=extra_only,
operator_id=operator_id,
replace=False,
)
sync_processing_to_legacy_supplement(
db,
tenant_no=tenant_no,
lab_storage_code=code,
sample_type_name=None,
payload=extra_only,
operator_id=operator_id,
)
restored += 1
field_hits += len(extra_only)
db.commit()
return {
"liveCodes": len(live_codes),
"retiredCandidates": len(retired),
"restoredRows": restored,
"skippedRows": skipped,
"restoredFields": field_hits,
}
def _load_qc_tasks_map(
db: Session, tenant_no: str, codes: List[str]
) -> Dict[str, List[Dict[str, Any]]]:
from app.models.models import BizQcTask, BizQcWorkOrder, BizQcWorkOrderInspectionBatch
from app.models.qc_param import BizQcTaskResult
from app.services.sample_storage_cert_qc_prefill import _stamp, batch_kind_from_name
uniq = [c for c in {(x or "").strip() for x in codes} if c]
if not uniq:
return {}
rows = (
db.query(
BizQcWorkOrder.sample_code,
BizQcTask.id,
BizQcTask.item_name,
BizQcTask.completed_time,
BizQcTask.update_time,
BizQcWorkOrderInspectionBatch.batch_name,
BizQcTaskResult.param_name,
BizQcTaskResult.param_value,
)
.join(BizQcTask, BizQcTask.work_order_id == BizQcWorkOrder.id)
.outerjoin(
BizQcWorkOrderInspectionBatch,
BizQcWorkOrderInspectionBatch.id == BizQcTask.batch_id,
)
.outerjoin(
BizQcTaskResult,
(BizQcTaskResult.task_id == BizQcTask.id) & (BizQcTaskResult.del_flag == 0),
)
.filter(
BizQcWorkOrder.tenant_no == tenant_no,
BizQcWorkOrder.sample_code.in_(uniq),
BizQcWorkOrder.del_flag == 0,
BizQcTask.tenant_no == tenant_no,
BizQcTask.del_flag == 0,
BizQcTask.task_status == "ywc",
)
.all()
)
by_code: Dict[str, Dict[str, Dict[str, Any]]] = {}
for row in rows:
code = (row.sample_code or "").strip()
if not code:
continue
grouped = by_code.setdefault(code, {})
task = grouped.setdefault(
row.id,
{
"item_name": row.item_name or "",
"batch_kind": batch_kind_from_name(row.batch_name),
"stamp": _stamp(row.completed_time, row.update_time),
"params": [],
"values": [],
},
)
if (row.param_name or "").strip() or (row.param_value or "").strip():
task["params"].append((row.param_name or "", row.param_value or ""))
if (row.param_value or "").strip():
task["values"].append(row.param_value or "")
return {code: list(tasks.values()) for code, tasks in by_code.items()}
def _qc_fields_for_dataset(dataset: str, tasks: List[Dict[str, Any]]) -> Dict[str, Any]:
from app.services import lab_sample_process as legacy_ledger
if not tasks:
return {}
out: Dict[str, Any] = {}
out.update(legacy_ledger.extract_process_tests(tasks))
mother = legacy_ledger.extract_mother_blood(tasks)
if dataset == DATASET_UBMC:
remap = {
"motherHbsag": "hbsag",
"motherAntiHcv": "antiHcv",
"motherAntiTp": "antiTp",
"motherAntiHiv": "antiHiv",
"motherCmvIgm": "cmvIgm",
}
for key, value in mother.items():
out[remap.get(key, key)] = value
else:
out.update(mother)
allowed = _extra_field_keys(dataset)
return {k: v for k, v in out.items() if k in allowed and not _is_blank(v)}
def sync_processing_to_legacy_supplement(
db: Session,
*,
@@ -624,7 +814,28 @@ def list_rows(
reception = receptions_by_sample_id.get(sample.id) or receptions_by_code.get(sample.sample_code or "")
if reception and reception.lab_storage_code:
legacy_codes.append(reception.lab_storage_code)
if reception and reception.sample_code:
legacy_codes.append(reception.sample_code)
legacy_supplements = _load_legacy_supplements(db, tenant_no, legacy_codes)
qc_tasks_map = _load_qc_tasks_map(db, tenant_no, legacy_codes)
def _compose_extra(*codes: Optional[str]) -> Dict[str, Any]:
picked = _pick_extra(extras, *codes)
legacy = {}
for c in codes:
key = (c or "").strip()
if key and key in legacy_supplements:
legacy = legacy_supplements[key]
break
merged = _merge_extra_with_legacy(picked, legacy)
for c in codes:
key = (c or "").strip()
if not key:
continue
for qk, qv in _qc_fields_for_dataset(dataset, qc_tasks_map.get(key) or []).items():
if qk not in merged or _is_blank(merged.get(qk)):
merged[qk] = qv
return merged
built_map: Dict[str, Dict[str, Any]] = {}
for sample, stype, party, coll, hospital in filtered:
@@ -632,10 +843,7 @@ def list_rows(
reception = receptions_by_sample_id.get(sample.id) or receptions_by_code.get(code)
production = productions.get(reception.id) if reception else None
lab_code = (reception.lab_storage_code if reception else None) or ""
extra = _merge_extra_with_legacy(
_pick_extra(extras, lab_code, code),
legacy_supplements.get(lab_code) or legacy_supplements.get(code or ""),
)
extra = _compose_extra(lab_code, code, reception.sample_code if reception else None)
protocol = protocols.get(sample.collection_protocol_id) if sample.collection_protocol_id else None
check_flags = check_flags_by_reception.get(reception.id, {}) if reception else {}
row = _build_row(
@@ -711,26 +919,20 @@ def list_rows(
)
check_flags_by_reception.update(ubmc_flags)
ubmc_legacy = _load_legacy_supplements(
db,
tenant_no,
[
(r.lab_storage_code or r.sample_code or "")
for r in ubmc_recs
],
)
ubmc_legacy_codes = [
(r.lab_storage_code or r.sample_code or "")
for r in ubmc_recs
]
ubmc_legacy = _load_legacy_supplements(db, tenant_no, ubmc_legacy_codes)
legacy_supplements.update(ubmc_legacy)
qc_tasks_map.update(_load_qc_tasks_map(db, tenant_no, ubmc_legacy_codes))
for reception in ubmc_recs:
code = (reception.lab_storage_code or reception.sample_code or "").strip()
if not code or code in built_map:
continue
production = productions.get(reception.id)
extra = _merge_extra_with_legacy(
_pick_extra(extras, code, reception.sample_code),
legacy_supplements.get(code)
or legacy_supplements.get((reception.sample_code or "").strip()),
)
extra = _compose_extra(code, reception.sample_code)
mother = mother_by_donation.get(str(reception.research_donation_id or ""))
fake_sample = SimpleNamespace(
id=reception.donation_sample_id or reception.id,
@@ -752,6 +954,95 @@ def list_rows(
check_flags=check_flags_by_reception.get(reception.id, {}),
)
# 间充质:科研捐献接收的 UC*/BC* 入库编码也需展示(无协议样本时)
if dataset == DATASET_MSC:
msc_recs = (
db.query(BizSampleReception)
.filter(
BizSampleReception.tenant_no == tenant_no,
BizSampleReception.del_flag == 0,
or_(
BizSampleReception.lab_storage_code.like("UC%"),
BizSampleReception.lab_storage_code.like("BC%"),
),
)
.order_by(BizSampleReception.reception_time.desc())
.all()
)
mother_by_donation: Dict[str, str] = {}
donation_ids = sorted(
{r.research_donation_id for r in msc_recs if r.research_donation_id}
)
if donation_ids:
rows = db.execute(
text(
"SELECT research_donation_id, mother_name FROM biz_research_donation_party "
"WHERE del_flag=0 AND tenant_no=:tenant_no AND research_donation_id IN ("
+ ",".join(f":d{i}" for i in range(len(donation_ids)))
+ ")"
),
{
"tenant_no": tenant_no,
**{f"d{i}": did for i, did in enumerate(donation_ids)},
},
).fetchall()
for did, mother in rows:
if did and mother and did not in mother_by_donation:
mother_by_donation[str(did)] = mother
missing_rids = [r.id for r in msc_recs if r.id and r.id not in productions]
if missing_rids:
for p in (
db.query(BizSampleProductionReception)
.filter(
BizSampleProductionReception.tenant_no == tenant_no,
BizSampleProductionReception.del_flag == 0,
BizSampleProductionReception.reception_id.in_(missing_rids),
)
.order_by(BizSampleProductionReception.process_time.desc())
.all()
):
productions.setdefault(p.reception_id, p)
msc_flags = _check_flag_map(
db,
tenant_no,
[r.id for r in msc_recs if r.id and r.id not in check_flags_by_reception],
)
check_flags_by_reception.update(msc_flags)
msc_legacy_codes = [
(r.lab_storage_code or r.sample_code or "") for r in msc_recs
]
legacy_supplements.update(_load_legacy_supplements(db, tenant_no, msc_legacy_codes))
qc_tasks_map.update(_load_qc_tasks_map(db, tenant_no, msc_legacy_codes))
for reception in msc_recs:
code = (reception.lab_storage_code or reception.sample_code or "").strip()
if not code or code in built_map:
continue
production = productions.get(reception.id)
extra = _compose_extra(code, reception.sample_code)
mother = mother_by_donation.get(str(reception.research_donation_id or ""))
fake_sample = SimpleNamespace(
id=reception.donation_sample_id or reception.id,
sample_code=reception.sample_code or code,
collection_protocol_id=None,
)
fake_party = SimpleNamespace(mother_name=mother) if mother else None
built_map[code] = _build_row(
dataset=dataset,
sample=fake_sample,
type_name="脐带间充质干细胞",
party=fake_party,
coll=None,
hospital=None,
reception=reception,
production=production,
extra=extra,
protocol=None,
check_flags=check_flags_by_reception.get(reception.id, {}),
)
kw = (keyword or "").strip()
built = []
for row in built_map.values():