Files
janssen/IWRS/import_patients.py
2026-06-10 11:59:19 +02:00

90 lines
2.9 KiB
Python

"""
import_patients.py — import pacientských reportů z IWRS/Incoming/ do MongoDB.
Verze: 1.1 | Datum: 2026-06-10
v1.1: přesun na úroveň IWRS/, worker přejmenován na import_patients_mongo
Pořadí zpracování per typ + studie: nejstarší soubor podle mtime první
(důležité pro chronologickou správnost snapshotů).
Po úspěšném importu se soubor přesune do IWRS/Incoming/Processed/.
Při chybě zůstane soubor v Incoming/.
Volá se z IWRS/run_all_v1.1.py (ensure_indexes volá orchestrátor);
lze spustit i samostatně: python import_patients.py
"""
import os
import sys
import glob
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
if BASE_DIR not in sys.path:
sys.path.insert(0, BASE_DIR)
from common.paths import INCOMING_DIR, PROCESSED_DIR, STUDIES, move_done, sorted_by_mtime
from common.mongo_writer import ensure_indexes
import import_patients_mongo
import import_notifications_to_mongo
def import_summaries(study):
pattern = os.path.join(INCOMING_DIR, f"* {study} Subject Summary Report*.xlsx")
files = sorted_by_mtime(glob.glob(pattern))
if not files:
print(f" [{study}] summary: nic ke zpracování")
return
print(f" [{study}] summary: {len(files)} soubor(ů) (oldest first)")
for path in files:
try:
import_patients_mongo.import_subject_summary(study, path)
move_done(path)
except Exception as e:
print(f" [{study}] CHYBA summary {os.path.basename(path)}: {e}")
def import_details(study):
pattern = os.path.join(INCOMING_DIR, f"* {study} * Subject Detail.xlsx")
files = sorted_by_mtime(glob.glob(pattern))
if not files:
print(f" [{study}] detail: nic ke zpracování")
return
print(f" [{study}] detail: {len(files)} soubor(ů) (oldest first)")
for path in files:
parsed = import_patients_mongo.parse_detail_filename(path)
if not parsed:
print(f" [{study}] PŘESKAKUJI (nelze parsovat název): {os.path.basename(path)}")
continue
_, parsed_study, subject = parsed
if parsed_study != study:
continue # patří jiné studii
try:
import_patients_mongo.import_visits_single_file(study, subject, path)
move_done(path)
except Exception as e:
print(f" [{study}] CHYBA detail {os.path.basename(path)}: {e}")
def run(studies=None):
studies = studies or STUDIES
if not os.path.isdir(INCOMING_DIR):
print(f"Adresář neexistuje: {INCOMING_DIR}")
return
print("=" * 60)
print("Import Subject Summary + Visits")
print("=" * 60)
for study in studies:
import_summaries(study)
import_details(study)
print("\n" + "=" * 60)
print("Import notifikací")
print("=" * 60)
import_notifications_to_mongo.import_from_dir(INCOMING_DIR, PROCESSED_DIR, studies)
if __name__ == "__main__":
ensure_indexes()
run()