FemCare: added cleanup function at begin of patient import

craws · Mar 7, 2025 · 93280fc · 93280fc
1 parent 6181bc3
commit 93280fc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/install/import/femcare_patients.py b/install/import/femcare_patients.py
@@ -7,6 +7,7 @@
 
 import numpy as np
 import pandas as pd
+from flask import g
 from pandas import isnull
 
 from openatlas import app
@@ -147,6 +148,15 @@ def get_diagnose_types(entries_: list[Entry]) -> dict[str, Entity]:
     case_study = Entity.get_by_id(357)
     diagnose_hierarchy = Entity.get_by_id(359)
     death_type = Entity.get_by_id(361)
+
+    # Remove former data
+    for item in case_study.get_linked_entities('P2', True):
+        item.delete()
+
+    for type_id in diagnose_hierarchy.subs:  # type: ignore
+        g.types[type_id].delete()
+
+    # Insert import
     entries = parse_csv()
     diagnose_types = get_diagnose_types(entries)
     not_imported = []

diff --git a/install/import/femcare_sisters.py b/install/import/femcare_sisters.py
@@ -105,6 +105,7 @@ def insert_rank_types(entries_: list[Entry]) -> dict[str, Any]:
 with app.test_request_context():
     app.preprocess_request()
     case_study = Entity.get_by_id(358)
+
     # Remove former data
     for item in case_study.get_linked_entities('P2', True):
         item.delete()