endoreg-db 0.8.8.0__py3-none-any.whl → 0.8.8.9__py3-none-any.whl

endoreg_db/management/commands/load_contraindication_data.py

@@ -1,41 +1,39 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import Contraindication
-from ...utils import load_model_data_from_yaml
+
 from ...data import CONTRAINDICATION_DATA_DIR as SOURCE_DIR
+from ...utils import load_model_data_from_yaml
 
 MODEL_0 = Contraindication
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     MODEL_0.__name__,
 ]
 
 IMPORT_METADATA = {
     MODEL_0.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": MODEL_0, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": MODEL_0,
+        "foreign_keys": [],  # e.g. ["intervention_types"]
+        "foreign_key_models": [],  # e.g. [InterventionType]
     }
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/intervention directory
     into the Intervention and InterventionType model"""
 
     def add_arguments(self, parser):
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
-        verbose = options['verbose']
+        verbose = options["verbose"]
         for model_name in IMPORT_MODELS:
             _metadata = IMPORT_METADATA[model_name]
-            load_model_data_from_yaml(
-                self,
-                model_name,
-                _metadata,
-                verbose
-            )
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_disease_classification_choices_data.py

@@ -1,41 +1,38 @@
 from django.core.management.base import BaseCommand
-from endoreg_db.models import DiseaseClassificationChoice as MODEL_0
-from ...utils import load_model_data_from_yaml
-from ...data import DISEASE_CLASSIFICATION_CHOICE_DATA_DIR as SOURCE_DIR
 
 from endoreg_db.models import DiseaseClassification
+from endoreg_db.models import DiseaseClassificationChoice as MODEL_0
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+from ...data import DISEASE_CLASSIFICATION_CHOICE_DATA_DIR as SOURCE_DIR
+from ...utils import load_model_data_from_yaml
+
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     MODEL_0.__name__,
 ]
 
 IMPORT_METADATA = {
     MODEL_0.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": MODEL_0, # e.g. Intervention
-        "foreign_keys": ["disease_classification"], # e.g. ["intervention_types"]
-        "foreign_key_models": [DiseaseClassification] # e.g. [InterventionType]
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": MODEL_0,
+        "foreign_keys": ["disease_classification"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [DiseaseClassification],  # e.g. [InterventionType]
     }
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/intervention directory
     into the Intervention and InterventionType model"""
 
     def add_arguments(self, parser):
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
-        verbose = options['verbose']
+        verbose = options["verbose"]
         for model_name in IMPORT_MODELS:
             _metadata = IMPORT_METADATA[model_name]
-            load_model_data_from_yaml(
-                self,
-                model_name,
-                _metadata,
-                verbose
-            )
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_disease_classification_data.py

@@ -1,41 +1,38 @@
 from django.core.management.base import BaseCommand
-from endoreg_db.models import DiseaseClassification as MODEL_0
-from ...utils import load_model_data_from_yaml
-from ...data import DISEASE_CLASSIFICATION_DATA_DIR as SOURCE_DIR
 
 from endoreg_db.models import Disease
+from endoreg_db.models import DiseaseClassification as MODEL_0
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+from ...data import DISEASE_CLASSIFICATION_DATA_DIR as SOURCE_DIR
+from ...utils import load_model_data_from_yaml
+
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     MODEL_0.__name__,
 ]
 
 IMPORT_METADATA = {
     MODEL_0.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": MODEL_0, # e.g. Intervention
-        "foreign_keys": ["disease"], # e.g. ["intervention_types"]
-        "foreign_key_models": [Disease] # e.g. [InterventionType]
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": MODEL_0,
+        "foreign_keys": ["disease"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [Disease],  # e.g. [InterventionType]
     }
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/intervention directory
     into the Intervention and InterventionType model"""
 
     def add_arguments(self, parser):
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
-        verbose = options['verbose']
+        verbose = options["verbose"]
         for model_name in IMPORT_MODELS:
             _metadata = IMPORT_METADATA[model_name]
-            load_model_data_from_yaml(
-                self,
-                model_name,
-                _metadata,
-                verbose
-            )
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_disease_data.py

@@ -1,18 +1,19 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import (
-    Disease, 
+    Disease,
     DiseaseClassification,
     DiseaseClassificationChoice,
 )
-from ...utils import load_model_data_from_yaml
+
 from ...data import (
-    DISEASE_DATA_DIR, 
-    DISEASE_CLASSIFICATION_DATA_DIR,
     DISEASE_CLASSIFICATION_CHOICE_DATA_DIR,
+    DISEASE_CLASSIFICATION_DATA_DIR,
+    DISEASE_DATA_DIR,
 )
+from ...utils import load_model_data_from_yaml
 
-
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     Disease.__name__,
     DiseaseClassification.__name__,
     DiseaseClassificationChoice.__name__,
@@ -20,43 +21,39 @@ IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
 
 IMPORT_METADATA = {
     Disease.__name__: {
-        "dir": DISEASE_DATA_DIR, # e.g. "interventions"
-        "model": Disease, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": DISEASE_DATA_DIR,  # e.g. "interventions"
+        "model": Disease,
+        "foreign_keys": [],  # e.g. ["intervention_types"]
+        "foreign_key_models": [],  # e.g. [InterventionType]
     },
     DiseaseClassification.__name__: {
-        "dir": DISEASE_CLASSIFICATION_DATA_DIR, # e.g. "interventions"
-        "model": DiseaseClassification, # e.g. Intervention
-        "foreign_keys": ["disease"], # e.g. ["intervention_types"]
-        "foreign_key_models": [Disease] # e.g. [InterventionType]
+        "dir": DISEASE_CLASSIFICATION_DATA_DIR,  # e.g. "interventions"
+        "model": DiseaseClassification,
+        "foreign_keys": ["disease"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [Disease],  # e.g. [InterventionType]
     },
     DiseaseClassificationChoice.__name__: {
-        "dir": DISEASE_CLASSIFICATION_CHOICE_DATA_DIR, # e.g. "interventions"
-        "model": DiseaseClassificationChoice, # e.g. Intervention
-        "foreign_keys": ["disease_classification"], # e.g. ["intervention_types"]
-        "foreign_key_models": [DiseaseClassification] # e.g. [InterventionType]
+        "dir": DISEASE_CLASSIFICATION_CHOICE_DATA_DIR,  # e.g. "interventions"
+        "model": DiseaseClassificationChoice,
+        "foreign_keys": ["disease_classification"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [DiseaseClassification],  # e.g. [InterventionType]
     },
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/intervention directory
     into the Intervention and InterventionType model"""
 
     def add_arguments(self, parser):
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
-        verbose = options['verbose']
+        verbose = options["verbose"]
         for model_name in IMPORT_MODELS:
             _metadata = IMPORT_METADATA[model_name]
-            load_model_data_from_yaml(
-                self,
-                model_name,
-                _metadata,
-                verbose
-            )
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_endoscope_data.py

@@ -1,24 +1,21 @@
-from django.conf import settings
 import os
+
+from django.conf import settings
 from django.core.management.base import BaseCommand
-from ...models import (
-    EndoscopeType,
-    Endoscope,
-    EndoscopyProcessor,
-    Center
-)
-from ...utils import load_model_data_from_yaml
+
 from ...data import (
+    ENDOSCOPE_DATA_DIR,
     ENDOSCOPE_TYPE_DATA_DIR,
     ENDOSCOPY_PROCESSOR_DATA_DIR,
-    ENDOSCOPE_DATA_DIR,
 )
+from ...models import Center, Endoscope, EndoscopeType, EndoscopyProcessor
+from ...utils import load_model_data_from_yaml
 
-SOURCE_DIR = ENDOSCOPE_TYPE_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
+SOURCE_DIR = ENDOSCOPE_TYPE_DATA_DIR  # e.g. settings.DATA_DIR_INTERVENTION
 
 MODEL_0 = EndoscopeType
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     EndoscopeType.__name__,
     EndoscopyProcessor.__name__,
     Endoscope.__name__,
@@ -26,43 +23,39 @@ IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
 
 IMPORT_METADATA = {
     EndoscopeType.__name__: {
-        "dir": ENDOSCOPE_TYPE_DATA_DIR, # e.g. "interventions"
-        "model": EndoscopeType, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": ENDOSCOPE_TYPE_DATA_DIR,  # e.g. "interventions"
+        "model": EndoscopeType,
+        "foreign_keys": [],  # e.g. ["intervention_types"]
+        "foreign_key_models": [],  # e.g. [InterventionType]
     },
     EndoscopyProcessor.__name__: {
         "dir": ENDOSCOPY_PROCESSOR_DATA_DIR,
         "model": EndoscopyProcessor,
         "foreign_keys": ["centers"],
-        "foreign_key_models": [Center]
+        "foreign_key_models": [Center],
     },
     Endoscope.__name__: {
         "dir": ENDOSCOPE_DATA_DIR,
         "model": Endoscope,
         "foreign_keys": ["center", "endoscope_type"],
-        "foreign_key_models": [Center, EndoscopeType]
+        "foreign_key_models": [Center, EndoscopeType],
     },
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/intervention directory
     into the Intervention and InterventionType model"""
 
     def add_arguments(self, parser):
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
-        verbose = options['verbose']
+        verbose = options["verbose"]
         for model_name in IMPORT_MODELS:
             _metadata = IMPORT_METADATA[model_name]
-            load_model_data_from_yaml(
-                self,
-                model_name,
-                _metadata,
-                verbose
-            )
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_event_data.py

@@ -1,41 +1,39 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import Event
-from ...utils import load_model_data_from_yaml
+
 from ...data import EVENT_DATA_DIR as SOURCE_DIR
+from ...utils import load_model_data_from_yaml
 
 MODEL_0 = Event
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     MODEL_0.__name__,
 ]
 
 IMPORT_METADATA = {
     MODEL_0.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": MODEL_0, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": MODEL_0,
+        "foreign_keys": [],  # e.g. ["intervention_types"]
+        "foreign_key_models": [],  # e.g. [InterventionType]
     }
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/intervention directory
     into the Intervention and InterventionType model"""
 
     def add_arguments(self, parser):
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
-        verbose = options['verbose']
+        verbose = options["verbose"]
         for model_name in IMPORT_MODELS:
             _metadata = IMPORT_METADATA[model_name]
-            load_model_data_from_yaml(
-                self,
-                model_name,
-                _metadata,
-                verbose
-            )
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_examination_data.py

@@ -1,75 +1,62 @@
+import os
+
 from django.conf import settings
 from django.core.management.base import BaseCommand
-from ...models import (
-    Examination,
-    ExaminationType,  # Replace 'your_app_name' with the actual app name
-    ExaminationTime,
-    ExaminationTimeType,
-)
-import os
-from ...utils import load_model_data_from_yaml
+
 from ...data import EXAMINATION_DATA_DIR
+from ...models import Examination, ExaminationIndication, ExaminationTime, ExaminationTimeType, ExaminationType, Finding
+from ...utils import load_model_data_from_yaml
 
-SOURCE_DIR = EXAMINATION_DATA_DIR 
+SOURCE_DIR = EXAMINATION_DATA_DIR
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     "ExaminationType",
-    "Examination",
     "ExaminationTimeType",
-    "ExaminationTime"
+    "ExaminationTime",
+    "Examination",
 ]
 
 IMPORT_METADATA = {
-    # "": { # same as model name in "import models", e.g. "Intervention"
-    #     "subdir": os.path.join(SOURCE_DIR,""), # e.g. "interventions"
-    #     "model": None, # e.g. Intervention
-    #     "foreign_keys": [], # e.g. ["intervention_types"]
-    #     "foreign_key_models": [] # e.g. [InterventionType]
-    # },
     "ExaminationType": {
-        "dir": os.path.join(SOURCE_DIR,"type"), # e.g. "interventions"
-        "model": ExaminationType, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": os.path.join(SOURCE_DIR, "type"),
+        "model": ExaminationType,
+        "foreign_keys": [],
+        "foreign_key_models": [],
     },
     "Examination": {
-        "dir": os.path.join(SOURCE_DIR,"examinations"), # e.g. "interventions"
-        "model": Examination, # e.g. Intervention
-        "foreign_keys": ["examination_types"], # e.g. ["intervention_types"]
-        "foreign_key_models": [ExaminationType] # e.g. [InterventionType]
+        "dir": os.path.join(SOURCE_DIR, "examinations"),
+        "model": Examination,
+        "foreign_keys": ["findings", "examination_types", "examination_times", "indications"],
+        "foreign_key_models": [Finding, ExaminationType, ExaminationTime, ExaminationIndication],
     },
     "ExaminationTimeType": {
-        "dir": os.path.join(SOURCE_DIR,"time-type"), # e.g. "interventions"
-        "model": ExaminationTimeType, # e.g. Intervention
-        "foreign_keys": ["examinations"], # e.g. ["intervention_types"]
-        "foreign_key_models": [Examination] # e.g. [InterventionType]
+        "dir": os.path.join(SOURCE_DIR, "time-type"),
+        "model": ExaminationTimeType,
+        "foreign_keys": ["examinations"],
+        "foreign_key_models": [Examination],
     },
     "ExaminationTime": {
-        "dir": os.path.join(SOURCE_DIR,"time"), # e.g. "interventions"
-        "model": ExaminationTime, # e.g. Intervention
-        "foreign_keys": ["time_types"], # e.g. ["intervention_types"]
-        "foreign_key_models": [ExaminationTimeType] # e.g. [InterventionType]
+        "dir": os.path.join(SOURCE_DIR, "time"),
+        "model": ExaminationTime,
+        "foreign_keys": ["time_types"],
+        "foreign_key_models": [ExaminationTimeType],
     },
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/intervention directory
     into the Intervention and InterventionType model"""
 
     def add_arguments(self, parser):
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
-        verbose = options['verbose']
+        verbose = options["verbose"]
         for model_name in IMPORT_MODELS:
             _metadata = IMPORT_METADATA[model_name]
-            load_model_data_from_yaml(
-                self,
-                model_name,
-                _metadata,
-                verbose
-            )
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_examination_indication_data.py

@@ -1,24 +1,23 @@
-from django.conf import settings
 from django.core.management.base import BaseCommand
+
+from ...data import (
+    EXAMINATION_INDICATION_CLASSIFICATION_CHOICE_DATA_DIR,
+    EXAMINATION_INDICATION_CLASSIFICATION_DATA_DIR,
+    EXAMINATION_INDICATION_DATA_DIR,
+)
 from ...models import (
-    ExaminationIndicationClassification,
     ExaminationIndication,
+    ExaminationIndicationClassification,
     ExaminationIndicationClassificationChoice,
-    Examination,
     FindingIntervention,
+    InformationSource,
 )
 from ...utils import load_model_data_from_yaml
-from ...data import (
-    EXAMINATION_INDICATION_CLASSIFICATION_CHOICE_DATA_DIR,
-    EXAMINATION_INDICATION_CLASSIFICATION_DATA_DIR,
-    EXAMINATION_INDICATION_DATA_DIR,
-)
-
 
 IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
-    ExaminationIndication.__name__,
-    ExaminationIndicationClassification.__name__,
     ExaminationIndicationClassificationChoice.__name__,
+    ExaminationIndicationClassification.__name__,
+    ExaminationIndication.__name__,
 ]
 
 IMPORT_METADATA = {
@@ -26,31 +25,31 @@ IMPORT_METADATA = {
         "dir": EXAMINATION_INDICATION_DATA_DIR,
         "model": ExaminationIndication,
         "foreign_keys": [
-            "examinations",
             "expected_interventions",
+            "classifications",
+            "information_sources",
         ],
         "foreign_key_models": [
-            Examination,
             FindingIntervention,
+            ExaminationIndicationClassification,
+            InformationSource,
         ],
     },
     ExaminationIndicationClassification.__name__: {
         "dir": EXAMINATION_INDICATION_CLASSIFICATION_DATA_DIR,
         "model": ExaminationIndicationClassification,
         "foreign_keys": [
-            "examinations",
-            "indications",  # This is a many-to-many field
+            "choices",  # This is a many-to-many field
         ],
         "foreign_key_models": [
-            Examination,
-            ExaminationIndication,
+            ExaminationIndicationClassificationChoice,
         ],
     },
     ExaminationIndicationClassificationChoice.__name__: {
         "dir": EXAMINATION_INDICATION_CLASSIFICATION_CHOICE_DATA_DIR,
         "model": ExaminationIndicationClassificationChoice,
-        "foreign_keys": ["classification"],
-        "foreign_key_models": [ExaminationIndicationClassification],
+        "foreign_keys": [],
+        "foreign_key_models": [],
     },
 }
 
@@ -62,7 +61,7 @@ class Command(BaseCommand):
     def add_arguments(self, parser):
         """
         Add the --verbose flag to the command-line argument parser.
-        
+
         This method augments the parser with a '--verbose' option to enable detailed output
         during command execution.
         """
@@ -75,7 +74,7 @@ class Command(BaseCommand):
     def handle(self, *args, **options):
         """
         Executes YAML data import for configured models.
-        
+
         Retrieves the verbosity flag from the options and iterates through each model
         in IMPORT_MODELS. For every model, it obtains the associated metadata from
         IMPORT_METADATA and invokes load_model_data_from_yaml to load data from its YAML files.