fix(lab-sample): 新旧实验室样本台账补录双向联动
新模块读取旧台账补录并映射字段;旧台账保存时同步到新补充表与生产传染四项,新模块保存时回写旧补录。 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
1 parent
2aecb69cc4
commit
c0aba421da
3 files changed
+339
-2
No files matched your search
@@ -179,6 +179,17 @@ def upsert_supplement(
|
||||
del_flag=0,
|
||||
)
|
||||
db.add(row)
|
||||
# 同步到新模块补充表/生产接收,保证实验室样本处理数据联动
|
||||
from app.services.lab_sample_processing import sync_legacy_supplement_to_processing
|
||||
|
||||
sync_legacy_supplement_to_processing(
|
||||
db,
|
||||
tenant_no=tenant_no,
|
||||
lab_storage_code=code,
|
||||
sample_type_name=type_name,
|
||||
payload=clean_payload,
|
||||
operator_id=user_id,
|
||||
)
|
||||
db.commit()
|
||||
db.refresh(row)
|
||||
return get_supplement(db, tenant_no=tenant_no, lab_storage_code=code) or {}
|
||||
|
||||
@@ -206,6 +206,240 @@ def _pick_extra(
|
||||
return {}
|
||||
|
||||
|
||||
# 旧台账补录字段名 → 新模块字段名(同名键保持不变)
|
||||
_LEGACY_SUPPLEMENT_TO_NEW: Dict[str, str] = {
|
||||
"receiveCheckText": "receiveCheck",
|
||||
"infectiousFourTestLabel": "infectiousFourCard",
|
||||
"preOpNucleatedCount": "preOpTnc",
|
||||
"preEnrichNucleatedCount": "preEnrichTnc",
|
||||
"postEnrichNucleatedCount": "postEnrichTnc",
|
||||
"cd34Count1e4": "cd34Count",
|
||||
"transportPackingList": "hasPackingList",
|
||||
"cordBloodReceiveConfirm": "hasReceiveConfirm",
|
||||
"remarkProcess": "processRemark",
|
||||
"motherHbsag": "hbsag",
|
||||
"motherAntiHcv": "antiHcv",
|
||||
"motherAntiTp": "antiTp",
|
||||
"motherAntiHiv": "antiHiv",
|
||||
"motherCmvIgm": "cmvIgm",
|
||||
}
|
||||
|
||||
_NEW_TO_LEGACY_SUPPLEMENT: Dict[str, str] = {
|
||||
v: k for k, v in _LEGACY_SUPPLEMENT_TO_NEW.items()
|
||||
}
|
||||
# MSC 等仍使用 mother* 键名,双向都保留
|
||||
for _same in (
|
||||
"labMaterialCode",
|
||||
"storageStatus",
|
||||
"discardReason",
|
||||
"hasMaternalBlood",
|
||||
"remeasureMaternalBlood",
|
||||
"infectiousFourExternal",
|
||||
"endotoxinTest",
|
||||
"bacteriaTest",
|
||||
"fungusTest",
|
||||
"mycoplasmaTest",
|
||||
"motherHbsag",
|
||||
"motherHbvDna",
|
||||
"motherAntiHcv",
|
||||
"motherHcvRna",
|
||||
"motherAntiTp",
|
||||
"motherCmvIgm",
|
||||
"motherAntiHiv",
|
||||
"motherHiv1Rna",
|
||||
"storageDateP2",
|
||||
"storageTotalTubes",
|
||||
"availableCellCount1e7",
|
||||
):
|
||||
_LEGACY_SUPPLEMENT_TO_NEW.setdefault(_same, _same)
|
||||
_NEW_TO_LEGACY_SUPPLEMENT.setdefault(_same, _same)
|
||||
|
||||
|
||||
def _is_blank(val: Any) -> bool:
|
||||
return val is None or (isinstance(val, str) and not val.strip())
|
||||
|
||||
|
||||
def _map_legacy_supplement(payload: Optional[Dict[str, Any]]) -> Dict[str, Any]:
|
||||
out: Dict[str, Any] = {}
|
||||
for key, value in (payload or {}).items():
|
||||
if _is_blank(value):
|
||||
continue
|
||||
new_key = _LEGACY_SUPPLEMENT_TO_NEW.get(str(key), str(key))
|
||||
out[new_key] = value
|
||||
# MSC 场景保留 mother* 原键,避免被 UBMC 别名覆盖后丢失
|
||||
if str(key).startswith("mother") and new_key != str(key):
|
||||
out[str(key)] = value
|
||||
return out
|
||||
|
||||
|
||||
def _map_new_to_legacy_supplement(payload: Optional[Dict[str, Any]]) -> Dict[str, Any]:
|
||||
out: Dict[str, Any] = {}
|
||||
for key, value in (payload or {}).items():
|
||||
if _is_blank(value):
|
||||
continue
|
||||
legacy_key = _NEW_TO_LEGACY_SUPPLEMENT.get(str(key), str(key))
|
||||
out[legacy_key] = value
|
||||
return out
|
||||
|
||||
|
||||
def _merge_extra_with_legacy(
|
||||
extra: Optional[Dict[str, Any]], legacy: Optional[Dict[str, Any]]
|
||||
) -> Dict[str, Any]:
|
||||
"""新补充表优先;旧台账补录填空(含字段名映射)。"""
|
||||
mapped = _map_legacy_supplement(legacy)
|
||||
merged = dict(mapped)
|
||||
for key, value in (extra or {}).items():
|
||||
if not _is_blank(value):
|
||||
merged[key] = value
|
||||
return merged
|
||||
|
||||
|
||||
def _load_legacy_supplements(
|
||||
db: Session, tenant_no: str, codes: List[str]
|
||||
) -> Dict[str, Dict[str, Any]]:
|
||||
from app.models.models import BizLabSampleProcessSupplement
|
||||
from app.services import lab_sample_process as legacy_ledger
|
||||
|
||||
uniq = [c for c in {(x or "").strip() for x in codes} if c]
|
||||
if not uniq:
|
||||
return {}
|
||||
rows = (
|
||||
db.query(BizLabSampleProcessSupplement)
|
||||
.filter(
|
||||
BizLabSampleProcessSupplement.tenant_no == tenant_no,
|
||||
BizLabSampleProcessSupplement.del_flag == 0,
|
||||
BizLabSampleProcessSupplement.lab_storage_code.in_(uniq),
|
||||
)
|
||||
.all()
|
||||
)
|
||||
out: Dict[str, Dict[str, Any]] = {}
|
||||
for row in rows:
|
||||
out[row.lab_storage_code] = legacy_ledger.parse_payload(row.payload)
|
||||
return out
|
||||
|
||||
|
||||
def sync_legacy_supplement_to_processing(
|
||||
db: Session,
|
||||
*,
|
||||
tenant_no: str,
|
||||
lab_storage_code: str,
|
||||
sample_type_name: Optional[str] = None,
|
||||
payload: Optional[Dict[str, Any]] = None,
|
||||
operator_id: str = "legacy-sync",
|
||||
) -> None:
|
||||
"""旧台账补录保存后,同步到新模块补充表,并把传染四项写回生产接收。"""
|
||||
code = (lab_storage_code or "").strip()
|
||||
if not code:
|
||||
return
|
||||
mapped = _map_legacy_supplement(payload)
|
||||
if not mapped:
|
||||
return
|
||||
dataset = resolve_dataset_by_type_name(sample_type_name or "", code)
|
||||
if not dataset:
|
||||
if code.upper().startswith("UBMC"):
|
||||
dataset = DATASET_UBMC
|
||||
elif code.upper().startswith(("BC", "UC")):
|
||||
dataset = DATASET_MSC
|
||||
else:
|
||||
dataset = DATASET_HSC
|
||||
|
||||
upsert_extra_payload(
|
||||
db,
|
||||
tenant_no=tenant_no,
|
||||
dataset=dataset,
|
||||
sample_code=code,
|
||||
payload=mapped,
|
||||
operator_id=operator_id,
|
||||
replace=False,
|
||||
)
|
||||
|
||||
infectious = mapped.get("infectiousFourCard")
|
||||
if infectious not in (None, ""):
|
||||
reception = (
|
||||
db.query(BizSampleReception)
|
||||
.filter(
|
||||
BizSampleReception.tenant_no == tenant_no,
|
||||
BizSampleReception.del_flag == 0,
|
||||
or_(
|
||||
BizSampleReception.lab_storage_code == code,
|
||||
BizSampleReception.sample_code == code,
|
||||
),
|
||||
)
|
||||
.order_by(BizSampleReception.reception_time.desc())
|
||||
.first()
|
||||
)
|
||||
if reception:
|
||||
production = (
|
||||
db.query(BizSampleProductionReception)
|
||||
.filter(
|
||||
BizSampleProductionReception.tenant_no == tenant_no,
|
||||
BizSampleProductionReception.reception_id == reception.id,
|
||||
BizSampleProductionReception.del_flag == 0,
|
||||
)
|
||||
.order_by(BizSampleProductionReception.process_time.desc())
|
||||
.first()
|
||||
)
|
||||
if production is not None:
|
||||
production.infectious_four_test_result = _infectious_code(infectious)
|
||||
production.update_by = operator_id
|
||||
production.update_time = datetime.now()
|
||||
|
||||
|
||||
def sync_processing_to_legacy_supplement(
|
||||
db: Session,
|
||||
*,
|
||||
tenant_no: str,
|
||||
lab_storage_code: str,
|
||||
sample_type_name: Optional[str] = None,
|
||||
payload: Optional[Dict[str, Any]] = None,
|
||||
operator_id: str = "processing-sync",
|
||||
) -> None:
|
||||
"""新模块保存后,回写旧台账补录,保持两边可见字段一致。"""
|
||||
from app.models.models import BizLabSampleProcessSupplement
|
||||
from app.services import lab_sample_process as legacy_ledger
|
||||
|
||||
code = (lab_storage_code or "").strip()
|
||||
mapped = _map_new_to_legacy_supplement(payload)
|
||||
if not code or not mapped:
|
||||
return
|
||||
now = datetime.now()
|
||||
row = (
|
||||
db.query(BizLabSampleProcessSupplement)
|
||||
.filter(
|
||||
BizLabSampleProcessSupplement.tenant_no == tenant_no,
|
||||
BizLabSampleProcessSupplement.lab_storage_code == code,
|
||||
BizLabSampleProcessSupplement.del_flag == 0,
|
||||
)
|
||||
.first()
|
||||
)
|
||||
current = legacy_ledger.parse_payload(row.payload) if row else {}
|
||||
current.update(mapped)
|
||||
src = "research_donation" if code.upper().startswith("UBMC") else "cell_storage"
|
||||
if row is None:
|
||||
db.add(
|
||||
BizLabSampleProcessSupplement(
|
||||
id=_new_id(),
|
||||
lab_storage_code=code,
|
||||
reception_source=src,
|
||||
sample_type_name=(sample_type_name or None),
|
||||
payload=legacy_ledger.dumps_payload(current),
|
||||
tenant_no=tenant_no,
|
||||
create_by=operator_id,
|
||||
create_dept_id="",
|
||||
create_time=now,
|
||||
update_by=operator_id,
|
||||
update_time=now,
|
||||
del_flag=0,
|
||||
)
|
||||
)
|
||||
else:
|
||||
if sample_type_name:
|
||||
row.sample_type_name = sample_type_name
|
||||
row.payload = legacy_ledger.dumps_payload(current)
|
||||
row.update_by = operator_id
|
||||
row.update_time = now
|
||||
|
||||
|
||||
def _match_date_range(row: Dict[str, Any], date_from: Optional[str], date_to: Optional[str]) -> bool:
|
||||
if not date_from and not date_to:
|
||||
return True
|
||||
@@ -384,13 +618,24 @@ def list_rows(
|
||||
)
|
||||
check_flags_by_reception = _check_flag_map(db, tenant_no, reception_ids_for_flags)
|
||||
|
||||
legacy_codes: List[str] = []
|
||||
for sample, *_ in filtered:
|
||||
legacy_codes.append(sample.sample_code or "")
|
||||
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)
|
||||
legacy_supplements = _load_legacy_supplements(db, tenant_no, legacy_codes)
|
||||
|
||||
built_map: Dict[str, Dict[str, Any]] = {}
|
||||
for sample, stype, party, coll, hospital in filtered:
|
||||
code = sample.sample_code
|
||||
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 = _pick_extra(extras, lab_code, code)
|
||||
extra = _merge_extra_with_legacy(
|
||||
_pick_extra(extras, lab_code, code),
|
||||
legacy_supplements.get(lab_code) or legacy_supplements.get(code or ""),
|
||||
)
|
||||
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(
|
||||
@@ -466,12 +711,26 @@ 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
|
||||
],
|
||||
)
|
||||
legacy_supplements.update(ubmc_legacy)
|
||||
|
||||
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 = _pick_extra(extras, code, reception.sample_code)
|
||||
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()),
|
||||
)
|
||||
mother = mother_by_donation.get(str(reception.research_donation_id or ""))
|
||||
fake_sample = SimpleNamespace(
|
||||
id=reception.donation_sample_id or reception.id,
|
||||
@@ -596,6 +855,12 @@ def _build_row(
|
||||
# 仅源系统没有的列走补充表;业务源表字段不被补充表覆盖
|
||||
val = extra.get(f.key)
|
||||
|
||||
# 源表为空时,允许用新/旧补充表填空(旧台账联动)
|
||||
if f.source != SOURCE_EXTRA and _is_blank(val):
|
||||
fallback = extra.get(f.key)
|
||||
if not _is_blank(fallback):
|
||||
val = fallback
|
||||
|
||||
if isinstance(val, datetime):
|
||||
val = _fmt_dt(val)
|
||||
elif isinstance(val, Decimal):
|
||||
@@ -894,6 +1159,19 @@ def update_row(
|
||||
operator_id=operator_id,
|
||||
replace=False,
|
||||
)
|
||||
legacy_code = (
|
||||
(reception.lab_storage_code if reception else None)
|
||||
or (sample.sample_code if sample else None)
|
||||
or code
|
||||
)
|
||||
sync_processing_to_legacy_supplement(
|
||||
db,
|
||||
tenant_no=tenant_no,
|
||||
lab_storage_code=str(legacy_code),
|
||||
sample_type_name=None,
|
||||
payload=extra_payload,
|
||||
operator_id=operator_id,
|
||||
)
|
||||
db.commit()
|
||||
|
||||
rows, _ = list_rows(db, tenant_no=tenant_no, dataset=dataset, keyword=code, page=1, page_size=20)
|
||||
|
||||
@@ -133,6 +133,54 @@ class LabSampleProcessingFieldsTest(unittest.TestCase):
|
||||
self.assertEqual(_infectious_code("正常"), "zc")
|
||||
self.assertEqual(_infectious_code("不正常"), "bzc")
|
||||
|
||||
def test_legacy_supplement_fills_empty_source_fields(self):
|
||||
from app.services.lab_sample_processing import _build_row, _merge_extra_with_legacy
|
||||
from types import SimpleNamespace
|
||||
|
||||
extra = _merge_extra_with_legacy(
|
||||
{},
|
||||
{"infectiousFourTestLabel": "正常", "labMaterialCode": "202611006"},
|
||||
)
|
||||
sample = SimpleNamespace(id="S1", sample_code="UBMC202611006", collection_protocol_id=None)
|
||||
reception = SimpleNamespace(
|
||||
id="R1",
|
||||
reception_time=None,
|
||||
receive_temperature=None,
|
||||
handler_name=None,
|
||||
remark=None,
|
||||
lab_storage_code="UBMC202611006",
|
||||
)
|
||||
production = SimpleNamespace(
|
||||
id="PR1",
|
||||
original_blood_weight_g=None,
|
||||
hes_volume_ml=None,
|
||||
blood_type=None,
|
||||
infectious_four_test_result=None,
|
||||
enriched_blood_weight_g=None,
|
||||
cryopreservation_volume_ml=None,
|
||||
handler_name=None,
|
||||
remark=None,
|
||||
original_blood_volume_ml=None,
|
||||
nucleated_cell_count_1e8=None,
|
||||
inoculation_amount=None,
|
||||
cryopreservation_time=None,
|
||||
storage_total_tubes=None,
|
||||
available_cell_count_1e7=None,
|
||||
)
|
||||
row = _build_row(
|
||||
dataset=DATASET_UBMC,
|
||||
sample=sample,
|
||||
type_name="脐血单个核细胞",
|
||||
party=None,
|
||||
coll=None,
|
||||
hospital=None,
|
||||
reception=reception,
|
||||
production=production,
|
||||
extra=extra,
|
||||
)
|
||||
self.assertEqual(row["infectiousFourCard"], "正常")
|
||||
self.assertEqual(row["labMaterialCode"], "202611006")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in new issue
Block a user