endoreg-db 0.4.5__py3-none-any.whl → 0.8.6.3__py3-none-any.whl

endoreg_db/migrations/0002_add_video_correction_models.py

@@ -0,0 +1,52 @@
+# Generated by Django 5.2.5 on 2025-10-09 10:12
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('endoreg_db', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='VideoMetadata',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('sensitive_frame_count', models.IntegerField(blank=True, help_text='Number of frames detected as containing sensitive information', null=True)),
+                ('sensitive_ratio', models.FloatField(blank=True, help_text='Ratio of sensitive frames to total frames (0.0-1.0)', null=True)),
+                ('sensitive_frame_ids', models.TextField(blank=True, help_text='JSON array of sensitive frame indices (0-based)', null=True)),
+                ('analyzed_at', models.DateTimeField(auto_now=True, help_text='Timestamp of last analysis')),
+                ('video', models.OneToOneField(help_text='Video file this metadata belongs to', on_delete=django.db.models.deletion.CASCADE, related_name='metadata', to='endoreg_db.videofile')),
+            ],
+            options={
+                'verbose_name': 'Video Metadata',
+                'verbose_name_plural': 'Video Metadata',
+                'db_table': 'video_metadata',
+            },
+        ),
+        migrations.CreateModel(
+            name='VideoProcessingHistory',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('operation', models.CharField(choices=[('mask_overlay', 'Mask Overlay'), ('frame_removal', 'Frame Removal'), ('analysis', 'Sensitivity Analysis'), ('reprocessing', 'Full Reprocessing')], help_text='Type of processing operation', max_length=50)),
+                ('status', models.CharField(choices=[('pending', 'Pending'), ('running', 'Running'), ('success', 'Success'), ('failure', 'Failure'), ('cancelled', 'Cancelled')], default='pending', help_text='Current status of the operation', max_length=20)),
+                ('config', models.JSONField(default=dict, help_text='Operation configuration (mask settings, frame list, etc.)')),
+                ('output_file', models.CharField(blank=True, help_text='Path to output file (relative to MEDIA_ROOT)', max_length=500)),
+                ('details', models.TextField(blank=True, help_text='Additional details or error messages')),
+                ('task_id', models.CharField(blank=True, help_text='Celery task ID for progress tracking', max_length=100)),
+                ('created_at', models.DateTimeField(auto_now_add=True, help_text='When the operation was started')),
+                ('completed_at', models.DateTimeField(blank=True, help_text='When the operation completed (success or failure)', null=True)),
+                ('video', models.ForeignKey(help_text='Video file this operation was performed on', on_delete=django.db.models.deletion.CASCADE, related_name='processing_history', to='endoreg_db.videofile')),
+            ],
+            options={
+                'verbose_name': 'Video Processing History',
+                'verbose_name_plural': 'Video Processing Histories',
+                'db_table': 'video_processing_history',
+                'ordering': ['-created_at'],
+                'indexes': [models.Index(fields=['video', '-created_at'], name='video_proce_video_i_98ffe0_idx'), models.Index(fields=['task_id'], name='video_proce_task_id_ec8b15_idx'), models.Index(fields=['status'], name='video_proce_status_6f71fd_idx')],
+            },
+        ),
+    ]

endoreg_db/migrations/0003_add_center_display_name.py

@@ -0,0 +1,30 @@
+from django.db import migrations, models
+
+
+def populate_display_name(apps, schema_editor):
+    Center = apps.get_model('endoreg_db', 'Center')
+    for center in Center.objects.all():
+        if not center.display_name:
+            center.display_name = center.name
+            center.save(update_fields=['display_name'])
+
+
+def reset_display_name(apps, schema_editor):
+    Center = apps.get_model('endoreg_db', 'Center')
+    Center.objects.update(display_name='')
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('endoreg_db', '0002_add_video_correction_models'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='center',
+            name='display_name',
+            field=models.CharField(blank=True, default='', max_length=255),
+        ),
+        migrations.RunPython(populate_display_name, reset_display_name),
+    ]

endoreg_db/models/__init__.py

@@ -1,73 +1,359 @@
-from .unit import Unit
-from .information_source import InformationSource
-from .center import *
-from .report_reader import ReportReaderConfig, ReportReaderFlag
-from .emission import *
-from .persons import *
-from .network import *
-from .logging import *
-
-from .case_template import *
-# from .other.distribution import (
-#     SingleCategoricalValueDistribution,
-#     NumericValueDistribution,
-#     MultipleCategoricalValueDistribution,
-#     DateValueDistribution,
-# )
-from .rules import (
-    Rule,
-    RuleType,
-    Ruleset,
-    RuleApplicator,
-    Rule
+####### Administration ########
+from .administration import (
+    AiModel,
+    ActiveModel,
+    ModelType,
+    Case,
+    CaseTemplate,
+    CaseTemplateRule,
+    CaseTemplateRuleType,
+    CaseTemplateRuleValue,
+    CaseTemplateRuleValueType,
+    CaseTemplateType,
+    Center,
+    CenterProduct,
+    CenterResource,
+    CenterWaste,
+    CenterShift,
+    Person,
+    Patient,
+    Examiner,
+    PortalUserInfo,
+    FirstName,
+    LastName,
+    Profession,
+    Product,
+    ProductMaterial,
+    ProductGroup,
+    ReferenceProduct,
+    ProductWeight,
+    Qualification,
+    QualificationType,
+    Shift,
+    ShiftType,
+    ScheduledDays,
+    Employee,
+    EmployeeType,
+    EmployeeQualification,
 )
 
-from .disease import Disease, DiseaseClassification, DiseaseClassificationChoice
-from .event import Event
-from .laboratory import LabValue
-from .medication import *
 
-from .examination import (
+####### Label ########
+from .label import (
+    Label,
+    LabelSet,
+    LabelType,
+    VideoSegmentationLabel,
+    VideoSegmentationLabelSet,
+    LabelVideoSegment,
+    ImageClassificationAnnotation,
+    VideoSegmentationAnnotation,
+)
+
+####### Media ########
+from .media import (
+    VideoFile,
+    Frame,
+    RawPdfFile,
+    DocumentType,
+    AnonymExaminationReport,
+    AnonymHistologyReport,
+    ReportReaderConfig,
+    ReportReaderFlag,
+    VideoMetadata,
+    VideoProcessingHistory,
+)
+
+######## Medical ########
+from .medical import (
+    Disease,
+    DiseaseClassification,
+    DiseaseClassificationChoice,
+    Event,
+    EventClassification,
+    EventClassificationChoice,
+    Contraindication,
     Examination,
+    ExaminationRequirementSet,
     ExaminationType,
+    ExaminationIndication,
+    ExaminationIndicationClassificationChoice,
+    ExaminationIndicationClassification,
     ExaminationTime,
     ExaminationTimeType,
+    Finding,
+    PatientFindingClassification,
+    FindingClassificationType,
+    FindingClassification,
+    FindingClassificationChoice,
+    FindingType,
+    FindingIntervention,
+    FindingInterventionType,
+    PatientDisease,
+    PatientEvent,
+    PatientExaminationIndication,
+    PatientExamination,
+    PatientFinding,
+    PatientFindingIntervention,
+    PatientLabSample,
+    PatientLabSampleType,
+    PatientLabValue,
+    PatientMedication,
+    PatientMedicationSchedule,
+    Organ,
+    Risk,
+    RiskType,
+    Medication,
+    MedicationSchedule,
+    MedicationIntakeTime,
+    MedicationIndicationType,
+    MedicationIndication,
+    Endoscope,
+    EndoscopeType,
+    EndoscopyProcessor,
+    LabValue,
 )
-from .data_file import *
 
-from .patient_examination import PatientExamination
+####### Metadata ########
+from .metadata import (
+    SensitiveMeta,
+    PdfMeta,
+    PdfType,
+    VideoMeta,
+    FFMpegMeta,
+    VideoImportMeta,
+    ModelMeta,
+    VideoPredictionMeta,
+)
 
-from .label import Label, LabelType, LabelSet
+from .other import (
+    Material,
+    Resource,
+    TransportRoute,
+    Waste,
+    BaseValueDistribution,
+    NumericValueDistribution,
+    SingleCategoricalValueDistribution,
+    MultipleCategoricalValueDistribution,
+    DateValueDistribution,
+    Gender,
+    InformationSource,
+    InformationSourceType,
+    Unit,
+    EmissionFactor,
+    Tag
+)
 
-from .annotation import *
+from .requirement import (
+    Requirement,
+    RequirementType,
+    RequirementOperator,
+    RequirementSet,
+    RequirementSetType,
+)
+from .rule import (
+    RuleType,
+    Rule,
+    Ruleset,
+    RuleAttributeDType,
+    RuleApplicator,
+)
 
-from .legacy_data import (
-    LegacyImage,
+from .state import (
+    SensitiveMetaState,
+    VideoState,
+    LabelVideoSegmentState,
+    AnonymizationStatus,
+    RawPdfState,
 )
 
-from .other import *
+__all__ = [
 
-from .product import *
+    ####### Administration ########
+        # AI
+    "AiModel",
+    "ActiveModel",
+    "ModelType",
 
-from .ai_model import (
-    ModelMeta,
-    ModelType,
-    ActiveModel,
-)
+    # Case
+    "Case",
+    "CaseTemplate",
+    "CaseTemplateRule",
+    "CaseTemplateRuleType",
+    "CaseTemplateRuleValue",
+    "CaseTemplateRuleValueType",
+    "CaseTemplateType",
 
-from .quiz import *
+    # Center
+    "Center",
+    "CenterProduct",
+    "CenterResource",
+    "CenterWaste",
+    "CenterShift",
 
-from .prediction import (
-    ImageClassificationPrediction,
-    LegacyVideoPredictionMeta,
-    VideoPredictionMeta,
-)
+    # Person
+    "Person",
+    "Patient",
+    "Examiner",
+    "PortalUserInfo",
+    "FirstName",
+    "LastName",
+    "Profession",
+    "Employee",
+    "EmployeeType",
+    "EmployeeQualification",
 
-from .hardware import (
-    EndoscopyProcessor,
-    Endoscope,
-    EndoscopeType,
-)
+    # Product
+    'Product',
+    'ProductMaterial',
+    'ProductGroup',
+    'ReferenceProduct',
+    'ProductWeight',
+
+    # Qualification
+    "Qualification",
+    "QualificationType",
+
+    # Shift
+    "Shift",
+    "ShiftType",
+    "ScheduledDays",
+    
+    ####### Label ########
+    "Label",
+    "LabelSet",
+    "LabelType",
+    "VideoSegmentationLabel",
+    "VideoSegmentationLabelSet",
+    "LabelVideoSegment",
+    "ImageClassificationAnnotation",
+    "VideoSegmentationAnnotation",
+
+
+    ####### Media ########
+    "VideoFile",
+    "Frame",
+    "RawPdfFile",
+    "DocumentType",
+    "AnonymExaminationReport",
+    "AnonymHistologyReport",
+    'ReportReaderConfig',
+    'ReportReaderFlag',
+    'VideoMetadata',
+    'VideoProcessingHistory',
+
+
+    ######## Medical ########
+    # Disease
+    "Disease",
+    "DiseaseClassification",
+    "DiseaseClassificationChoice",
+    # Event
+    "Event",
+    "EventClassification",
+    "EventClassificationChoice",
+    # Contraindication
+    "Contraindication",
+    # Examination
+    "Examination",
+    "ExaminationType",
+    "ExaminationIndication",
+    "ExaminationIndicationClassificationChoice",
+    "ExaminationIndicationClassification",
+    "ExaminationTime",
+    "ExaminationTimeType",
+    # Finding
+    "Finding",
+    "FindingClassificationType",
+    "FindingClassification",
+    "FindingClassificationChoice",
+    "FindingType",
+    "FindingIntervention",
+    "FindingInterventionType",
+    # Patient
+    ## Disease
+    "PatientDisease",
+    ## Event
+    "PatientEvent",
+    ## Examination
+    "PatientExaminationIndication",
+    "PatientExamination",
+    ## Finding
+    "PatientFinding",
+    "PatientFindingIntervention",
+    "PatientFindingClassification",
+    ## Laboratory
+    "PatientLabSample",
+    "PatientLabSampleType",
+    "PatientLabValue",
+    ## Medication
+    "PatientMedication",
+    "PatientMedicationSchedule",
+    # Organ
+    "Organ",
+    # Risk
+    "Risk",
+    "RiskType",
+    # Medication
+    "Medication",
+    "MedicationSchedule",
+    "MedicationIntakeTime",
+    "MedicationIndicationType",
+    "MedicationIndication",
+
+    # Hardware
+    "Endoscope",
+    "EndoscopeType",
+    "EndoscopyProcessor",
+
+    # Laboratory
+    "LabValue",
+
+    ####### Metadata ########
+    "SensitiveMeta",
+    "PdfMeta",
+    "PdfType",
+    "VideoMeta",
+    "FFMpegMeta",
+    "VideoImportMeta",
+    "ModelMeta",
+    "VideoPredictionMeta",
+
+    ####### Other #######
+    'Material',
+    'Resource',
+    'TransportRoute',
+    'Waste',
+    'BaseValueDistribution',
+    'NumericValueDistribution',
+    'SingleCategoricalValueDistribution',
+    'MultipleCategoricalValueDistribution',
+    'DateValueDistribution',
+    "Gender",
+    "InformationSource",
+    "InformationSourceType",
+    "Unit",
+    "EmissionFactor",
+    "Tag",
+
+    ###### Requirement ######
+    "Requirement",
+    "RequirementType",
+    "RequirementOperator",
+    "RequirementSet",
+    "RequirementSetType",
+
+    ######## Rule #######
+    "RuleType",
+    "Rule",
+    "Ruleset",
+    "RuleAttributeDType",
+    "RuleApplicator",
 
-from .questionnaires import TtoQuestionnaire
-from .annotation import AnonymousImageAnnotation, AnonymizedImageLabel, DroppedName
+    ####### State ########
+    "SensitiveMetaState",
+    "VideoState",
+    "LabelVideoSegmentState",
+    "AnonymizationStatus",
+    "RawPdfState",
+]

endoreg_db/models/administration/__init__.py

@@ -0,0 +1,116 @@
+# AI
+from .ai import (
+    AiModel,
+    ActiveModel,
+    ModelType,
+)
+
+# Case
+from .case import (
+    Case,
+    CaseTemplate,
+    CaseTemplateRule,
+    CaseTemplateRuleType,
+    CaseTemplateRuleValue,
+    CaseTemplateRuleValueType,
+    CaseTemplateType,
+)
+
+# Center
+from .center import (
+    Center,
+    CenterProduct,
+    CenterResource,
+    CenterWaste,
+    CenterShift,
+)
+
+
+#TODO Review module
+# from .permissions import () 
+
+# Person
+from .person import (
+    Person,
+    Examiner, 
+    Patient, 
+    PortalUserInfo,     
+    FirstName,
+    LastName,
+    Profession,
+    Employee,
+    EmployeeType,
+    EmployeeQualification,
+)
+
+# Product
+from .product import (
+    Product,
+    ProductMaterial,
+    ProductGroup,
+    ReferenceProduct,
+    ProductWeight,
+)
+
+from .qualification import (
+    Qualification,
+    QualificationType,
+)
+
+from .shift import (
+    Shift,
+    ShiftType,
+    ScheduledDays,
+)
+
+__all__ = [
+    # AI
+    "AiModel",
+    "ActiveModel",
+    "ModelType",
+
+    # Case
+    "Case",
+    "CaseTemplate",
+    "CaseTemplateRule",
+    "CaseTemplateRuleType",
+    "CaseTemplateRuleValue",
+    "CaseTemplateRuleValueType",
+    "CaseTemplateType",
+
+    # Center
+    "Center",
+    "CenterProduct",
+    "CenterResource",
+    "CenterWaste",
+    "CenterShift",
+
+    # Person
+    "Person",
+    "Patient",
+    "Examiner",
+    "PortalUserInfo",
+    "FirstName",
+    "LastName",
+    "Profession",
+    "Employee",
+    "EmployeeType",
+    "EmployeeQualification",
+
+    # Product
+    'Product',
+    'ProductMaterial',
+    'ProductGroup',
+    'ReferenceProduct',
+    'ProductWeight',
+
+    # Qualification
+    "Qualification",
+    "QualificationType",
+
+    # Shift
+    "Shift",
+    "ShiftType",
+    "ScheduledDays",
+]
+

endoreg_db/models/administration/ai/__init__.py

@@ -0,0 +1,9 @@
+from .model_type import ModelType
+from .active_model import ActiveModel
+from .ai_model import AiModel
+
+__all__ = [
+    "ModelType",
+    "ActiveModel",
+    "AiModel",
+]

endoreg_db/models/administration/ai/active_model.py

@@ -0,0 +1,35 @@
+from django.db import models
+from typing import TYPE_CHECKING, Optional
+
+if TYPE_CHECKING:
+    from endoreg_db.models.metadata import ModelMeta
+
+
+class ActiveModelManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+    
+class ActiveModel(models.Model):
+    """
+    ActiveModel represents an active instance of a model within the application.
+    Attributes:
+        name (str): A unique identifier for the active model.
+        model_meta (ModelMeta, optional): A reference to the metadata of the model. This field acts as a ForeignKey
+                                            to the ModelMeta model and can be null.
+    Notes:
+        - The model_meta attribute is configured with on_delete=models.SET_NULL, meaning that if the related ModelMeta
+          record is deleted, model_meta will be set to null.
+        - Type hints for model_meta are provided under a TYPE_CHECKING conditional import to avoid circular dependencies,
+          importing ModelMeta only when type checking is performed.
+    Managers:
+        objects (ActiveModelManager): Custom manager providing specialized query capabilities for ActiveModel instances.
+    """
+
+    name = models.CharField(max_length=255, unique=True)
+    
+    model_meta:Optional[models.ForeignKey["ModelMeta"]] = models.ForeignKey(
+        'ModelMeta', on_delete=models.SET_NULL, 
+        blank=True, null=True
+    )
+
+    objects = ActiveModelManager()