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

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.

endoreg_db/management/commands/load_finding_data.py

@@ -1,128 +1,100 @@
 from django.core.management.base import BaseCommand
-from endoreg_db.models import (
-    Finding, FindingType, Examination,
-    FindingClassification, FindingClassificationType, FindingClassificationChoice,
-    Organ,
-    FindingIntervention, FindingInterventionType, LabValue, Contraindication
 
+from endoreg_db.models import (
+    Finding,
+    FindingClassification,
+    FindingClassificationChoice,
+    FindingClassificationType,
+    FindingIntervention,
+    FindingInterventionType,
+    FindingType,
 )
-from ...utils import load_model_data_from_yaml
+
 from ...data import (
-    FINDING_DATA_DIR, FINDING_TYPE_DATA_DIR,
+    FINDING_CLASSIFICATION_CHOICE_DATA_DIR,
+    FINDING_CLASSIFICATION_DATA_DIR,
+    FINDING_CLASSIFICATION_TYPE_DATA_DIR,
+    FINDING_DATA_DIR,
     FINDING_INTERVETION_DATA_DIR,
+    FINDING_TYPE_DATA_DIR,
     FINIDNG_INTERVENTION_TYPE_DATA_DIR,
-    FINDING_CLASSIFICATION_TYPE_DATA_DIR,
-    FINDING_CLASSIFICATION_DATA_DIR,
-    FINDING_CLASSIFICATION_CHOICE_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
     FindingInterventionType.__name__,
     FindingIntervention.__name__,
     FindingType.__name__,
-    Finding.__name__,
-    FindingClassificationType.__name__,
     FindingClassificationChoice.__name__,
+    FindingClassificationType.__name__,
     FindingClassification.__name__,
+    Finding.__name__,
 ]
 
 IMPORT_METADATA = {
     FindingType.__name__: {
-        "dir": FINDING_TYPE_DATA_DIR, # e.g. "interventions"
-        "model": FindingType, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": FINDING_TYPE_DATA_DIR,
+        "model": FindingType,
+        "foreign_keys": [],
+        "foreign_key_models": [],
     },
     Finding.__name__: {
-        "dir": FINDING_DATA_DIR, # e.g. "interventions"
-        "model": Finding, # e.g. Intervention
-        "foreign_keys": [
-            "finding_types",
-            "examinations",
-            "finding_interventions",
-        ], # e.g. ["intervention_types"]
-        "foreign_key_models": [
-            FindingType,
-            Examination,
-            FindingIntervention,
-            FindingIntervention,
-            FindingIntervention,
-        ] # e.g. [InterventionType]
+        "dir": FINDING_DATA_DIR,
+        "model": Finding,
+        "foreign_keys": ["finding_types", "finding_interventions", "finding_classifications", "caused_by_interventions"],
+        "foreign_key_models": [FindingType, FindingIntervention, FindingClassification, FindingIntervention],
     },
     FindingInterventionType.__name__: {
-        "dir": FINIDNG_INTERVENTION_TYPE_DATA_DIR, # e.g. "interventions"
-        "model": FindingInterventionType, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": FINIDNG_INTERVENTION_TYPE_DATA_DIR,
+        "model": FindingInterventionType,
+        "foreign_keys": [],
+        "foreign_key_models": [],
     },
     FindingIntervention.__name__: {
-        "dir": FINDING_INTERVETION_DATA_DIR, # e.g. "interventions"
-        "model": FindingIntervention, # e.g. Intervention
-        "foreign_keys": [
-            "intervention_types",
-            "required_lab_values",
-            "contraindications"
-        ], # e.g. ["intervention_types"]
-        "foreign_key_models": [
-            FindingInterventionType,
-            LabValue,
-            Contraindication
-        ] # e.g. [InterventionType]
+        "dir": FINDING_INTERVETION_DATA_DIR,
+        "model": FindingIntervention,
+        "foreign_keys": ["intervention_types"],
+        "foreign_key_models": [FindingInterventionType],
     },
     FindingClassificationType.__name__: {
-        "dir": FINDING_CLASSIFICATION_TYPE_DATA_DIR, # e.g. "interventions"
-        "model": FindingClassificationType, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": FINDING_CLASSIFICATION_TYPE_DATA_DIR,
+        "model": FindingClassificationType,
+        "foreign_keys": [],
+        "foreign_key_models": [],
     },
     FindingClassification.__name__: {
-        "dir": FINDING_CLASSIFICATION_DATA_DIR, # e.g. "interventions"
-        "model": FindingClassification, # e.g. Intervention
-        "foreign_keys": [
-            "classification_types",
-            "findings",
-            "examinations",
-            "finding_types",
-            "choices"
-        ], # e.g. ["intervention_types"]
+        "dir": FINDING_CLASSIFICATION_DATA_DIR,
+        "model": FindingClassification,
+        "foreign_keys": ["classification_types", "findings", "finding_types", "choices"],
         "foreign_key_models": [
             FindingClassificationType,
             Finding,
-            Examination,
             FindingType,
             FindingClassificationChoice,
-        ] # e.g. [InterventionType]
+        ],
     },
     FindingClassificationChoice.__name__: {
-        "dir": FINDING_CLASSIFICATION_CHOICE_DATA_DIR, # e.g. "interventions"
-        "model": FindingClassificationChoice, # e.g. Intervention
-        "foreign_keys": [
-            # "classifications",
-        ], # e.g. ["intervention_types"]
-        "foreign_key_models": [
-            # FindingClassification
-        ] # e.g. [InterventionType]
+        "dir": FINDING_CLASSIFICATION_CHOICE_DATA_DIR,
+        "model": FindingClassificationChoice,
+        "foreign_keys": [],
+        "foreign_key_models": [],
     },
 }
 
+
 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_information_source.py

@@ -1,51 +1,49 @@
-from django.conf import settings
 import os
+
+from django.conf import settings
 from django.core.management.base import BaseCommand
+
+from ...data import INFORMATION_DATA_DIR, INFORMATION_SOURCE_TYPE_DATA_DIR
 from ...models import InformationSource, InformationSourceType
 from ...utils import load_model_data_from_yaml
-from ...data import INFORMATION_DATA_DIR, INFORMATION_SOURCE_TYPE_DATA_DIR
 
-SOURCE_DIR = INFORMATION_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
+SOURCE_DIR = INFORMATION_DATA_DIR  # e.g. settings.DATA_DIR_INTERVENTION
 
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     InformationSourceType.__name__,
     InformationSource.__name__,
 ]
 
 IMPORT_METADATA = {
     InformationSourceType.__name__: {
-        "dir": INFORMATION_SOURCE_TYPE_DATA_DIR, # e.g. "intervention_type"
-        "model": InformationSourceType, # e.g. InterventionType
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": INFORMATION_SOURCE_TYPE_DATA_DIR,  # e.g. "intervention_type"
+        "model": InformationSourceType,
+        "foreign_keys": [],  # e.g. ["intervention_types"]
+        "foreign_key_models": [],  # e.g. [InterventionType]
     },
     InformationSource.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": InformationSource, # e.g. Intervention
-        "foreign_keys": ["information_source_types"], # e.g. ["intervention_types"]
-        "foreign_key_models": [InformationSourceType] # e.g. [InterventionType]
-    }
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": InformationSource,
+        "foreign_keys": ["information_source_types"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [InformationSourceType],  # 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_lab_value_data.py

@@ -1,50 +1,41 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import LabValue as MODEL_0
-from endoreg_db.models import PatientLabSampleType, NumericValueDistribution
-from ...utils import load_model_data_from_yaml
+from endoreg_db.models import NumericValueDistribution, PatientLabSampleType, Unit
+
 from ...data import LAB_VALUE_DATA_DIR as SOURCE_DIR
 from ...data import PATIENT_LAB_SAMPLE_TYPE_DATA_DIR
+from ...utils import load_model_data_from_yaml
 
-from endoreg_db.models import Unit
-
-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__,
-    PatientLabSampleType.__name__
+    PatientLabSampleType.__name__,
 ]
 
 IMPORT_METADATA = {
     MODEL_0.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": MODEL_0, # e.g. Intervention
-        "foreign_keys": ["default_unit","default_numerical_value_distribution"], # e.g. ["intervention_types"]
-        "foreign_key_models": [Unit, NumericValueDistribution] # e.g. [InterventionType]
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": MODEL_0,
+        "foreign_keys": ["default_unit", "default_numerical_value_distribution"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [Unit, NumericValueDistribution],  # e.g. [InterventionType]
     },
-    PatientLabSampleType.__name__: {
-        "dir": PATIENT_LAB_SAMPLE_TYPE_DATA_DIR,
-        "model": PatientLabSampleType,
-        "foreign_keys": [],
-        "foreign_key_models": []
-    }
+    PatientLabSampleType.__name__: {"dir": PATIENT_LAB_SAMPLE_TYPE_DATA_DIR, "model": PatientLabSampleType, "foreign_keys": [], "foreign_key_models": []},
 }
 
+
 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_medication_data.py

@@ -1,25 +1,26 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import (
+    Disease,
+    DiseaseClassificationChoice,
+    Event,
+    InformationSource,
     Medication,
     MedicationIndication,
     MedicationIndicationType,
     MedicationIntakeTime,
     MedicationSchedule,
     Unit,
-    Disease,
-    Event,
-    InformationSource,
-    DiseaseClassificationChoice,
 )
-from ...utils import load_model_data_from_yaml
+
 from ...data import (
     MEDICATION_DATA_DIR,
+    MEDICATION_INDICATION_DATA_DIR,
     MEDICATION_INDICATION_TYPE_DATA_DIR,
     MEDICATION_INTAKE_TIME_DATA_DIR,
     MEDICATION_SCHEDULE_DATA_DIR,
-    MEDICATION_INDICATION_DATA_DIR,
 )
-
+from ...utils import load_model_data_from_yaml
 
 IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     Medication.__name__,
@@ -32,25 +33,25 @@ IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
 IMPORT_METADATA = {
     Medication.__name__: {
         "dir": MEDICATION_DATA_DIR,  # e.g. "interventions"
-        "model": Medication,  # e.g. Intervention
+        "model": Medication,
         "foreign_keys": ["default_unit"],  # e.g. ["intervention_types"]
         "foreign_key_models": [Unit],  # e.g. [InterventionType]
     },
     MedicationIndicationType.__name__: {
         "dir": MEDICATION_INDICATION_TYPE_DATA_DIR,  # e.g. "interventions"
-        "model": MedicationIndicationType,  # e.g. Intervention
+        "model": MedicationIndicationType,
         "foreign_keys": [],  # e.g. ["intervention_types"]
         "foreign_key_models": [],  # e.g. [InterventionType]
     },
     MedicationIntakeTime.__name__: {
         "dir": MEDICATION_INTAKE_TIME_DATA_DIR,  # e.g. "interventions"
-        "model": MedicationIntakeTime,  # e.g. Intervention
+        "model": MedicationIntakeTime,
         "foreign_keys": [],  # e.g. ["intervention_types"]
         "foreign_key_models": [],  # e.g. [InterventionType]
     },
     MedicationSchedule.__name__: {
         "dir": MEDICATION_SCHEDULE_DATA_DIR,  # e.g. "interventions"
-        "model": MedicationSchedule,  # e.g. Intervention
+        "model": MedicationSchedule,
         "foreign_keys": [
             "medication",
             "intake_times",
@@ -64,7 +65,7 @@ IMPORT_METADATA = {
     },
     MedicationIndication.__name__: {
         "dir": MEDICATION_INDICATION_DATA_DIR,  # e.g. "interventions"
-        "model": MedicationIndication,  # e.g. Intervention
+        "model": MedicationIndication,
         "foreign_keys": [
             "indication_type",
             "medication_schedules",

endoreg_db/management/commands/load_organ_data.py

@@ -1,43 +1,39 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import Organ
+
+from ...data import ORGAN_DATA_DIR
 from ...utils import load_model_data_from_yaml
-from ...data import (
-    ORGAN_DATA_DIR
-)
 
 MODEL_0 = Organ
 
-IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
-    MODEL_0.__name__,#
+IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
+    MODEL_0.__name__,  #
 ]
 
 IMPORT_METADATA = {
     MODEL_0.__name__: {
-        "dir": ORGAN_DATA_DIR, # e.g. "interventions"
-        "model": MODEL_0, # e.g. Intervention
-        "foreign_keys": [], # e.g. ["intervention_types"]
-        "foreign_key_models": [] # e.g. [InterventionType]
+        "dir": ORGAN_DATA_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)