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

endoreg_db/management/commands/load_pdf_type_data.py

@@ -1,27 +1,29 @@
-from django.conf import settings
 import os
-from django.core.management.base import BaseCommand
 
 import yaml
-from ...utils import load_model_data_from_yaml
+from django.conf import settings
+from django.core.management.base import BaseCommand
 
+from ...data import REPORT_TYPE_DATA_DIR
 
 #### CUSTOMIZE
 from ...models import PdfType, ReportReaderFlag
-from ...data import PDF_TYPE_DATA_DIR
+from ...utils import load_model_data_from_yaml
 
-SOURCE_DIR = PDF_TYPE_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
+SOURCE_DIR = REPORT_TYPE_DATA_DIR  # e.g. settings.DATA_DIR_INTERVENTION
 
 MODEL_0 = PdfType
 
-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
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": MODEL_0,  # e.g. Intervention
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": MODEL_0,
         "foreign_keys": [
             "patient_info_line",
             "endoscope_info_line",
@@ -39,23 +41,22 @@ IMPORT_METADATA = {
     }
 }
 
+
 class Command(BaseCommand):
     help = f"""Load all .yaml files in the {SOURCE_DIR} directory"""
 
     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"]
+        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)
+
+            load_model_data_from_yaml(self, model_name, _metadata, verbose)

endoreg_db/management/commands/load_profession_data.py

@@ -1,44 +1,41 @@
 from django.conf import settings
 from django.core.management.base import BaseCommand
+
+from ...data import PROFESSION_DATA_DIR
 from ...models import Profession
 from ...utils import load_model_data_from_yaml
-from ...data import PROFESSION_DATA_DIR
 
-SOURCE_DIR = PROFESSION_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
+SOURCE_DIR = PROFESSION_DATA_DIR  # e.g. settings.DATA_DIR_INTERVENTION
 
 MODEL_0 = Profession
 
-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_qualification_data.py

@@ -1,33 +1,35 @@
 from django.core.management.base import BaseCommand
+
+from ...data import QUALIFICATION_DATA_DIR
 from ...models import Qualification, QualificationType
 from ...utils import load_model_data_from_yaml
-from ...data import QUALIFICATION_DATA_DIR
 
-SOURCE_DIR = QUALIFICATION_DATA_DIR       # qualification data directory
+SOURCE_DIR = QUALIFICATION_DATA_DIR  # qualification data directory
 
 model_0 = QualificationType
 model_1 = Qualification
 
-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__,
     model_1.__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]
     },
     model_1.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": model_1, # e.g. Intervention
-        "foreign_keys": ["qualification_types"], # e.g. ["intervention_types"]
-        "foreign_key_models": [model_0] # e.g. [InterventionType]
-    }
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": model_1,
+        "foreign_keys": ["qualification_types"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [model_0],  # e.g. [InterventionType]
+    },
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/qualification directory
     into the Qualification / Qualification Type model"""
@@ -37,23 +39,18 @@ class Command(BaseCommand):
         Adds the --verbose flag to the command-line parser to enable detailed output.
         """
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
         """
         Loads data from YAML files into the QualificationType and Qualification models.
-        
+
         Iterates over predefined models and imports their data from YAML files using associated metadata. Supports verbose output if enabled via command-line 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_report_reader_flag_data.py

@@ -1,46 +1,44 @@
-from django.conf import settings
 import os
+
+import yaml
+from django.conf import settings
 from django.core.management.base import BaseCommand
+
+from ...data import REPORT_READER_FLAG_DATA_DIR
 from ...models import ReportReaderFlag
-import yaml
 from ...utils import load_model_data_from_yaml
-from ...data import REPORT_READER_FLAG_DATA_DIR
 
-SOURCE_DIR = REPORT_READER_FLAG_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
+SOURCE_DIR = REPORT_READER_FLAG_DATA_DIR  # e.g. settings.DATA_DIR_INTERVENTION
 
 MODEL_0 = ReportReaderFlag
 
-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_requirement_data.py

@@ -45,7 +45,7 @@ IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     RequirementOperator.__name__,
     Requirement.__name__,
     RequirementSetType.__name__,
-    ExaminationRequirementSet.__name__,
+    # ExaminationRequirementSet.__name__,
     RequirementSet.__name__,
 ]
 
@@ -62,46 +62,38 @@ def _validate_requirement_configuration(fields: dict, *, entry: dict, model):
             return True
         return any(not item for item in value)
 
-    missing = [
-        key for key in ("requirement_types", "operators") if _values_missing(key)
-    ]
+    missing = [key for key in ("requirement_types", "operators") if _values_missing(key)]
     if missing:
         missing_display = ", ".join(missing)
-        raise ValueError(
-            f"{model.__name__} '{name}' is missing required configuration for: {missing_display}."
-        )
+        raise ValueError(f"{model.__name__} '{name}' is missing required configuration for: {missing_display}.")
 
 
 IMPORT_METADATA = {
     RequirementType.__name__: {
         "dir": REQUIREMENT_TYPE_DATA_DIR,  # e.g. "interventions"
-        "model": RequirementType,  # e.g. Intervention
+        "model": RequirementType,
         "foreign_keys": [],  # e.g. ["intervention_types"]
         "foreign_key_models": [],  # e.g. [InterventionType]
     },
     RequirementOperator.__name__: {
         "dir": REQUIREMENT_OPERATOR_DATA_DIR,  # e.g. "interventions"
-        "model": RequirementOperator,  # e.g. Intervention
+        "model": RequirementOperator,
         "foreign_keys": [],  # e.g. ["intervention_types"]
         "foreign_key_models": [],  # e.g. [InterventionType]
     },
     ExaminationRequirementSet.__name__: {
         "dir": EXAMINATION_REQUIREMENT_SET_DATA_DIR,  # e.g. "interventions"
-        "model": ExaminationRequirementSet,  # e.g. Intervention
-        "foreign_keys": [
-            "examinations",
-        ],  # Through model uses foreign keys of both models
-        "foreign_key_models": [
-            Examination,
-        ],
+        "model": ExaminationRequirementSet,
+        "foreign_keys": [],  # Through model uses foreign keys of both models
+        "foreign_key_models": [],
     },
     # ExaminationRequirementSet.__name__,
     Requirement.__name__: {
         "dir": REQUIREMENT_DATA_DIR,  # e.g. "interventions"
-        "model": Requirement,  # e.g. Intervention
+        "model": Requirement,
         "foreign_keys": [
             "requirement_types",
-            "operators",
+            "operator",
             "unit",
             "examinations",
             "examination_indications",
@@ -149,13 +141,13 @@ IMPORT_METADATA = {
     },
     RequirementSetType.__name__: {
         "dir": REQUIREMENT_SET_TYPE_DATA_DIR,  # e.g. "interventions"
-        "model": RequirementSetType,  # e.g. Intervention
+        "model": RequirementSetType,
         "foreign_keys": [],  # e.g. ["intervention_types"]
         "foreign_key_models": [],  # e.g. [InterventionType]
     },
     RequirementSet.__name__: {
         "dir": REQUIREMENT_SET_DATA_DIR,  # e.g. "interventions"
-        "model": RequirementSet,  # e.g. Intervention
+        "model": RequirementSet,
         "foreign_keys": [
             "requirement_set_type",
             "requirements",  # This is a many-to-many field
@@ -163,6 +155,7 @@ IMPORT_METADATA = {
             "information_sources",
             "tags",
             "reqset_exam_links",
+            "depends_on",
         ],  # e.g. ["intervention_types"]
         "foreign_key_models": [
             RequirementSetType,
@@ -171,6 +164,7 @@ IMPORT_METADATA = {
             InformationSource,
             Tag,
             ExaminationRequirementSet,
+            RequirementSet,
         ],  # e.g. [InterventionType]
     },
 }

endoreg_db/management/commands/load_risk_data.py

@@ -1,8 +1,9 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import Risk, RiskType
-from ...utils import load_model_data_from_yaml
-from ...data import RISK_DATA_DIR, RISK_TYPE_DATA_DIR
 
+from ...data import RISK_DATA_DIR, RISK_TYPE_DATA_DIR
+from ...utils import load_model_data_from_yaml
 
 IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     RiskType.__name__,
@@ -12,13 +13,13 @@ IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
 IMPORT_METADATA = {
     RiskType.__name__: {
         "dir": RISK_TYPE_DATA_DIR,  # e.g. "interventions"
-        "model": RiskType,  # e.g. Intervention
+        "model": RiskType,
         "foreign_keys": [],  # e.g. ["intervention_types"]
         "foreign_key_models": [],  # e.g. [InterventionType]
     },
     Risk.__name__: {
         "dir": RISK_DATA_DIR,  # e.g. "interventions"
-        "model": Risk,  # e.g. Intervention
+        "model": Risk,
         "foreign_keys": ["risk_type"],  # e.g. ["intervention_types"]
         "foreign_key_models": [RiskType],  # e.g. [InterventionType]
     },
@@ -32,7 +33,7 @@ class Command(BaseCommand):
     def add_arguments(self, parser):
         """
         Adds the '--verbose' flag to the argument parser for detailed output.
-        
+
         This method extends the given parser by adding an option that, when specified,
         enables verbose output during command execution.
         """
@@ -45,7 +46,7 @@ class Command(BaseCommand):
     def handle(self, *args, **options):
         """
         Execute the command to load YAML data into configured models.
-        
+
         Retrieves the verbosity setting from the options and iterates over each model in IMPORT_MODELS.
         For each model, it obtains the corresponding metadata from IMPORT_METADATA and calls the utility
         function load_model_data_from_yaml to load data from the associated YAML files.

endoreg_db/management/commands/load_shift_data.py

@@ -1,32 +1,34 @@
 from django.core.management.base import BaseCommand
+
+from ...data import REPORT_READER_FLAG_DATA_DIR
 from ...models import Shift, ShiftType
 from ...utils import load_model_data_from_yaml
-from ...data import REPORT_READER_FLAG_DATA_DIR
 
-SOURCE_DIR = REPORT_READER_FLAG_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
+SOURCE_DIR = REPORT_READER_FLAG_DATA_DIR  # e.g. settings.DATA_DIR_INTERVENTION
 
 model_0 = ShiftType
 model_1 = Shift
 
-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]
     },
     model_1.__name__: {
-        "dir": SOURCE_DIR, # e.g. "interventions"
-        "model": model_1, # e.g. Intervention
-        "foreign_keys": ["shift_types"], # e.g. ["intervention_types"]
-        "foreign_key_models": [model_0]   # e.g. [InterventionType]
-    }
+        "dir": SOURCE_DIR,  # e.g. "interventions"
+        "model": model_1,
+        "foreign_keys": ["shift_types"],  # e.g. ["intervention_types"]
+        "foreign_key_models": [model_0],  # e.g. [InterventionType]
+    },
 }
 
+
 class Command(BaseCommand):
     help = """Load all .yaml files in the data/shift directory
     into the Shift and ShiftType model"""
@@ -36,25 +38,20 @@ class Command(BaseCommand):
         Adds the --verbose command-line option to enable detailed output during execution.
         """
         parser.add_argument(
-            '--verbose',
-            action='store_true',
-            help='Display verbose output',
+            "--verbose",
+            action="store_true",
+            help="Display verbose output",
         )
 
     def handle(self, *args, **options):
         """
         Loads YAML data files into models defined in IMPORT_MODELS using provided metadata.
-        
+
         Iterates over each model specified in IMPORT_MODELS, retrieves its import metadata,
         and calls the data loading utility to populate the database from YAML files. Supports
         optional verbose output if enabled via command-line 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_tag_data.py

@@ -1,14 +1,11 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import (
     Tag,
-
 )
 
+from ...data import TAG_DATA_DIR
 from ...utils import load_model_data_from_yaml
-from ...data import (
-    TAG_DATA_DIR
-)
-
 
 IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     Tag.__name__,
@@ -17,7 +14,7 @@ IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
 IMPORT_METADATA = {
     Tag.__name__: {
         "dir": TAG_DATA_DIR,  # e.g. "interventions"
-        "model": Tag,  # e.g. Intervention
+        "model": Tag,
         "foreign_keys": [],  # e.g. ["intervention_types"]
         "foreign_key_models": [],  # e.g. [InterventionType]
     },
@@ -32,7 +29,7 @@ class Command(BaseCommand):
     def add_arguments(self, parser):
         """
         Add command-line arguments to enable verbose output.
-        
+
         Adds an optional '--verbose' flag to the command parser. When specified,
         this flag causes the command to display detailed output during execution.
         """
@@ -45,10 +42,10 @@ class Command(BaseCommand):
     def handle(self, *args, **options):
         """
         Executes data import for requirement models from YAML files.
-        
-        Retrieves the verbosity setting from the command options and iterates over each model 
-        listed in IMPORT_MODELS. For each model, it obtains the corresponding metadata from 
-        IMPORT_METADATA and calls a helper to load the YAML data into the database. Verbose mode 
+
+        Retrieves the verbosity setting from the command options and iterates over each model
+        listed in IMPORT_MODELS. For each model, it obtains the corresponding metadata from
+        IMPORT_METADATA and calls a helper to load the YAML data into the database. Verbose mode
         enables detailed output during the process.
         """
         verbose = options["verbose"]

endoreg_db/management/commands/load_unit_data.py

@@ -1,46 +1,44 @@
-from django.conf import settings
 import os
+
+import yaml
+from django.conf import settings
 from django.core.management.base import BaseCommand
+
+from ...data import UNIT_DATA_DIR
 from ...models import Unit
-import yaml
 from ...utils import load_model_data_from_yaml
-from ...data import UNIT_DATA_DIR
 
-SOURCE_DIR = UNIT_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
+SOURCE_DIR = UNIT_DATA_DIR  # e.g. settings.DATA_DIR_INTERVENTION
 
 MODEL_0 = 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__,
 ]
 
 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)