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

endoreg_db/management/commands/init_default_ai_model.py

@@ -75,7 +75,7 @@ class Command(BaseCommand):
         self.stdout.write("Creating default model metadata...")
         
         # Use a dummy weights file path for now - in production this should point to actual model weights
-        dummy_weights_path = Path(__file__).parent.parent.parent / "assets" / "dummy_model.ckpt"
+        dummy_weights_path = Path(__file__).parent.parent.parent / "assets" / "dummy_model.safetensors"
         
         # Create the dummy weights file if it doesn't exist
         dummy_weights_path.parent.mkdir(parents=True, exist_ok=True)

endoreg_db/management/commands/list_routes.py

@@ -0,0 +1,18 @@
+from django.core.management.base import BaseCommand
+from django.urls import get_resolver, URLPattern, URLResolver
+
+def iter_patterns(prefix, patterns):
+    for p in patterns:
+        if isinstance(p, URLPattern):
+            yield p
+        elif isinstance(p, URLResolver):
+            yield from iter_patterns(prefix, p.url_patterns)
+
+class Command(BaseCommand):
+    help = "List all URL names (useful to fill policy.py)"
+
+    def handle(self, *args, **options):
+        resolver = get_resolver()
+        for p in iter_patterns("", resolver.url_patterns):
+            if p.name:
+                self.stdout.write(p.name)

endoreg_db/management/commands/load_ai_model_data.py

@@ -9,6 +9,7 @@ from ...data import (
 )
 from ...models import (
     AiModel,
+    LabelSet,  # Add LabelSet import
     ModelMeta,  # Add ModelMeta back to imports
     ModelType,
     VideoSegmentationLabel,
@@ -35,7 +36,7 @@ IMPORT_METADATA = {
         "dir": AI_MODEL_META_DATA_DIR,  # e.g. "ai_model_meta"
         "model": ModelMeta,  # e.g. ModelMeta
         "foreign_keys": ["labelset", "model"],  # Foreign key relationships
-        "foreign_key_models": [VideoSegmentationLabelSet, AiModel],  # Actual model classes
+        "foreign_key_models": [LabelSet, AiModel],  # Actual model classes
     },
     VideoSegmentationLabel.__name__: {
         "dir": VIDEO_SEGMENTATION_LABEL_DATA_DIR,  # e.g. "interventions"

endoreg_db/management/commands/load_center_data.py

@@ -14,22 +14,22 @@ IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
 
 IMPORT_METADATA = {
     FirstName.__name__: {
-        "dir": NAMES_FIRST_DATA_DIR,  # e.g. "interventions"
-        "model": FirstName,  # e.g. Intervention
-        "foreign_keys": [],  # e.g. ["intervention_types"]
-        "foreign_key_models": [],  # e.g. [InterventionType]
+        "dir": NAMES_FIRST_DATA_DIR,  # e.g. "first names"
+        "model": FirstName,  # e.g. first name
+        "foreign_keys": [],
+        "foreign_key_models": [],
     },
     LastName.__name__: {
-        "dir": NAMES_LAST_DATA_DIR,  # e.g. "interventions"
-        "model": LastName,  # e.g. Intervention
-        "foreign_keys": [],  # e.g. ["intervention_types"]
-        "foreign_key_models": [],  # e.g. [InterventionType]
+        "dir": NAMES_LAST_DATA_DIR,  # e.g. "last names"
+        "model": LastName,  # e.g. last name
+        "foreign_keys": [],  # e.g. ["last name_types"]
+        "foreign_key_models": [],  # e.g. [last nameType]
     },
     Center.__name__: {
-        "dir": SOURCE_DIR,  # e.g. "interventions"
-        "model": Center,  # e.g. Intervention
-        "foreign_keys": ["first_names", "last_names"],  # e.g. ["intervention_types"]
-        "foreign_key_models": [FirstName, LastName],  # e.g. [InterventionType]
+        "dir": SOURCE_DIR,  # e.g. "centers"
+        "model": Center,  # e.g. center
+        "foreign_keys": ["first_names", "last_names"],
+        "foreign_key_models": [FirstName, LastName],
     },
 }
 

endoreg_db/management/commands/load_requirement_data.py

@@ -1,43 +1,44 @@
 from django.core.management.base import BaseCommand
+
 from endoreg_db.models import (
-    RequirementType,
-    RequirementOperator,
-    Requirement,
-    RequirementSetType,
-    RequirementSet,
-    Examination,
-    ExaminationIndication,
     Disease,
     DiseaseClassificationChoice,
     Event,
-    LabValue,
+    Examination,
+    ExaminationIndication,
+    ExaminationRequirementSet,  # Added to avoid circular import issues
     Finding,
     FindingClassification,
     FindingClassificationChoice,
     FindingIntervention,
     InformationSource,
-    Unit,
-    Risk,
-    RiskType,
+    LabValue,
+    Medication,  # Added Medication model
     MedicationIndication,
     MedicationIndicationType,
-    MedicationSchedule,
-    Medication,  # Added Medication model
     MedicationIntakeTime,
+    MedicationSchedule,
+    Requirement,
+    RequirementOperator,
+    RequirementSet,
+    RequirementSetType,
+    RequirementType,
+    Risk,
+    RiskType,
     Tag,
-    ExaminationRequirementSet,  # Added to avoid circular import issues
+    Unit,
 )
 from endoreg_db.models.other.gender import Gender
-from ...utils import load_model_data_from_yaml
+
 from ...data import (
-    REQUIREMENT_TYPE_DATA_DIR,
-    REQUIREMENT_OPERATOR_DATA_DIR,
+    EXAMINATION_REQUIREMENT_SET_DATA_DIR,
     REQUIREMENT_DATA_DIR,
-    REQUIREMENT_SET_TYPE_DATA_DIR,
+    REQUIREMENT_OPERATOR_DATA_DIR,
     REQUIREMENT_SET_DATA_DIR,
-    EXAMINATION_REQUIREMENT_SET_DATA_DIR
+    REQUIREMENT_SET_TYPE_DATA_DIR,
+    REQUIREMENT_TYPE_DATA_DIR,
 )
-
+from ...utils import load_model_data_from_yaml
 
 IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     RequirementType.__name__,
@@ -48,6 +49,29 @@ IMPORT_MODELS = [  # string as model key, serves as key in IMPORT_METADATA
     RequirementSet.__name__,
 ]
 
+
+def _validate_requirement_configuration(fields: dict, *, entry: dict, model):
+    """Ensures requirement fixtures declare both requirement_types and operators."""
+    name = fields.get("name") or entry.get("pk") or "<unnamed>"
+
+    def _values_missing(key: str) -> bool:
+        value = fields.get(key)
+        if not isinstance(value, list):
+            return True
+        if not value:
+            return True
+        return any(not item for item in value)
+
+    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}."
+        )
+
+
 IMPORT_METADATA = {
     RequirementType.__name__: {
         "dir": REQUIREMENT_TYPE_DATA_DIR,  # e.g. "interventions"
@@ -64,8 +88,12 @@ IMPORT_METADATA = {
     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,],
+        "foreign_keys": [
+            "examinations",
+        ],  # Through model uses foreign keys of both models
+        "foreign_key_models": [
+            Examination,
+        ],
     },
     # ExaminationRequirementSet.__name__,
     Requirement.__name__: {
@@ -92,7 +120,7 @@ IMPORT_METADATA = {
             "medication_schedules",
             "medications",  # Added medications
             "medication_intake_times",
-            "genders"
+            "genders",
         ],
         "foreign_key_models": [
             RequirementType,
@@ -115,8 +143,9 @@ IMPORT_METADATA = {
             MedicationSchedule,
             Medication,  # Added Medication model
             MedicationIntakeTime,
-            Gender
+            Gender,
         ],
+        # "validators": [_validate_requirement_configuration],
     },
     RequirementSetType.__name__: {
         "dir": REQUIREMENT_SET_TYPE_DATA_DIR,  # e.g. "interventions"
@@ -133,7 +162,7 @@ IMPORT_METADATA = {
             "links_to_sets",
             "information_sources",
             "tags",
-            "reqset_exam_links"
+            "reqset_exam_links",
         ],  # e.g. ["intervention_types"]
         "foreign_key_models": [
             RequirementSetType,
@@ -141,7 +170,7 @@ IMPORT_METADATA = {
             RequirementSet,
             InformationSource,
             Tag,
-            ExaminationRequirementSet
+            ExaminationRequirementSet,
         ],  # e.g. [InterventionType]
     },
 }
@@ -155,7 +184,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.
         """
@@ -168,10 +197,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_requirement_set_tags.py

@@ -0,0 +1,95 @@
+"""
+Management command to load default requirement set tags for role-based filtering.
+
+⚠️ DEPRECATED: This command is deprecated. Please use 'load_tag_data' instead.
+
+The new method loads tags from YAML configuration files, making it easier to
+add, modify, or remove tags without changing Python code.
+
+New Usage:
+    uv run python manage.py load_tag_data --verbose
+
+Tags are now managed in: endoreg_db/data/tag/requirement_set_tags.yaml
+
+This command is kept for backward compatibility but may be removed in future versions.
+"""
+
+import logging
+
+from django.core.management.base import BaseCommand
+
+from endoreg_db.models import Tag
+
+logger = logging.getLogger(__name__)
+
+
+class Command(BaseCommand):
+    """
+    ⚠️ DEPRECATED: Load default requirement set tags for role-based filtering.
+
+    Please use 'load_tag_data' instead:
+        uv run python manage.py load_tag_data --verbose
+
+    This command creates or verifies the existence of default tags used
+    for filtering requirement sets by user role/expertise level.
+    """
+
+    help = "Load default requirement set tags for role-based filtering"
+
+    def handle(self, *args, **options):
+        """
+        Create or verify default tags for requirement set filtering.
+
+        Tags created:
+            - Gastroenterologist: For specialist-level requirements
+            - Student: For educational/basic requirements
+            - Professor: For academic/research requirements
+            - Terminology Expert: For terminology standardization
+            - Default Prototype: For baseline/reference requirements
+        """
+        default_tags = [
+            ("Gastroenterologist", "Specialist-level endoscopy requirements"),
+            ("Student", "Educational and basic learning requirements"),
+            ("Professor", "Academic and research-focused requirements"),
+            ("Terminology Expert", "Terminology standardization requirements"),
+            ("Default Prototype", "Baseline reference requirements"),
+        ]
+
+        created_count = 0
+        updated_count = 0
+
+        for tag_name, description in default_tags:
+            tag, created = Tag.objects.get_or_create(
+                name=tag_name,
+                defaults={"description": description}
+                if hasattr(Tag, "description")
+                else {},
+            )
+
+            if created:
+                created_count += 1
+                self.stdout.write(self.style.SUCCESS(f"✓ Created tag: {tag_name}"))
+                logger.info(f"Created requirement set tag: {tag_name}")
+            else:
+                updated_count += 1
+                self.stdout.write(f"  Tag already exists: {tag_name}")
+
+        # Summary
+        total_tags = len(default_tags)
+        self.stdout.write(
+            self.style.SUCCESS(
+                f"\n{'=' * 60}\n"
+                f"Tag Loading Complete\n"
+                f"{'=' * 60}\n"
+                f"  New tags created:     {created_count}\n"
+                f"  Existing tags found:  {updated_count}\n"
+                f"  Total tags verified:  {total_tags}\n"
+                f"{'=' * 60}"
+            )
+        )
+
+        logger.info(
+            f"Requirement set tag loading complete: "
+            f"{created_count} created, {updated_count} existing, "
+            f"{total_tags} total"
+        )

endoreg_db/management/commands/setup_endoreg_db.py

@@ -119,13 +119,17 @@ class Command(BaseCommand):
                     # Try to create model metadata using configurable approach
                     model_path = self._find_model_weights_file()
                     if model_path:
-                        call_command(
-                            "create_multilabel_model_meta",
-                            model_name=default_model_name,
-                            model_meta_version=1,
-                            image_classification_labelset_name=primary_labelset,
-                            model_path=str(model_path),
-                        )
+                        call_command_kwargs = {
+                            "model_name": default_model_name,
+                            "model_meta_version": 1,
+                            "image_classification_labelset_name": primary_labelset,
+                            "model_path": str(model_path),
+                        }
+                        # Add bump_version flag if force_recreate is enabled
+                        if force_recreate:
+                            call_command_kwargs["bump_version"] = True
+
+                        call_command("create_multilabel_model_meta", **call_command_kwargs)
                         self.stdout.write(self.style.SUCCESS("✅ AI model metadata created successfully"))
                     else:
                         self.stdout.write(self.style.WARNING("⚠️  Model weights file not found. AI features may not work properly."))
@@ -154,9 +158,9 @@ class Command(BaseCommand):
 
         self.stdout.write(self.style.SUCCESS("\n🎉 EndoReg DB embedded app setup completed successfully!"))
         self.stdout.write("\nNext steps:")
-        self.stdout.write("1. Run migrations: python manage.py migrate")
-        self.stdout.write("2. Create superuser: python manage.py createsuperuser")
-        self.stdout.write("3. Start development server: python manage.py runserver")
+        # self.stdout.write("1. Run migrations: python manage.py migrate")
+        self.stdout.write("1. Create superuser: python manage.py createsuperuser")
+        self.stdout.write("2. Start development server: python manage.py runserver")
 
     def _find_model_weights_file(self):
         """Find the model weights file using configurable search patterns and directories."""