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

endoreg_db/views/__init__.py

@@ -1,136 +1,86 @@
+from ..utils.translation import build_multilingual_response
 from .anonymization import (
     AnonymizationOverviewView,
     AnonymizationValidateView,
-    anonymization_status,
     anonymization_current,
+    anonymization_status,
     start_anonymization,
 )
-
-
-from ..utils.translation import build_multilingual_response
-
-
-from .auth import (
-    KeycloakVideoView,
-    keycloak_login,
-    keycloak_callback,
-    public_home,
-)
-
+from .auth import KeycloakVideoView, keycloak_callback, keycloak_login, public_home
 from .examination import (
     ExaminationManifestCache,
     ExaminationViewSet,
-
     get_classification_choices_for_examination,
-    get_morphology_classification_choices_for_examination,
-    get_location_classification_choices_for_examination,
-
     get_classifications_for_examination,
-    get_location_classifications_for_examination,
-    get_morphology_classifications_for_examination,
-
     get_findings_for_examination,
     get_instruments_for_examination,
     get_interventions_for_examination,
+    get_location_classification_choices_for_examination,
+    get_location_classifications_for_examination,
+    get_morphology_classification_choices_for_examination,
+    get_morphology_classifications_for_examination,
 )
-
 from .finding import (
-    get_interventions_for_finding,
+    FindingViewSet,
     get_classifications_for_finding,
-    FindingViewSet
+    get_interventions_for_finding,
 )
-
 from .finding_classification import (
     FindingClassificationViewSet,
     get_classification_choices,
-    get_morphology_choices, # DEPRECATED
-    get_location_choices, # DEPRECATED
+    get_location_choices,  # DEPRECATED
+    get_morphology_choices,  # DEPRECATED
 )
-
-from .label_video_segment import (
-    create_video_segment_annotation,
-    video_segments_by_label_id_view,
-    video_segments_by_label_name_view,
-    video_segment_detail_view,
-    video_segments_view,
-    update_label_video_segment,
-    get_lvs_by_name_and_video_id
+from .media import (
+    get_sensitive_metadata_pk,
+    label_list,
+    pdf_sensitive_metadata,
+    pdf_sensitive_metadata_list,
+    pdf_sensitive_metadata_verify,
+    sensitive_metadata_list,
+    video_sensitive_metadata,
+    video_sensitive_metadata_verify,
 )
-
-from .meta import (
-    AvailableFilesListView,
-    SensitiveMetaDetailView,
-    SensitiveMetaListView,
-    SensitiveMetaVerificationView,
-    ReportFileMetadataView,
-)
-
+from .meta import SensitiveMetaListView, SensitiveMetaVerificationView
 from .misc import (
+    MODELTRANSLATION_SETTINGS,
     CenterViewSet,
-    csrf_token_view,
-    GenderViewSet,
     ExaminationStatsView,
-    VideoSegmentStatsView,
-    SensitiveMetaStatsView,
-    GeneralStatsView,
-    SecureFileUrlView,
-    SecureFileServingView,
-    validate_secure_url,
     ExaminationTranslationOptions,
-    FindingTranslationOptions,
-    FindingClassificationTranslationOptions,
     FindingClassificationChoiceTranslationOptions,
+    FindingClassificationTranslationOptions,
+    FindingTranslationOptions,
+    GenderViewSet,
+    GeneralStatsView,
     InterventionTranslationOptions,
+    SensitiveMetaStatsView,
     TranslatedFieldMixin,
-    TranslationMigrationHelper,
     TranslatedFixtureLoader,
-    MODELTRANSLATION_SETTINGS,
+    TranslationMigrationHelper,
     UploadFileView,
     UploadStatusView,
+    VideoSegmentStatsView,
+    csrf_token_view,
 )
-
 from .patient import PatientViewSet
-
 from .patient_examination import (
     ExaminationCreateView,
     PatientExaminationDetailView,
     PatientExaminationListView,
     PatientExaminationViewSet,
 )
-
-from .patient_finding import (
-    PatientFindingViewSet,
-    OptimizedPatientFindingViewSet
-)
-
-from .patient_finding_classification import (
-    create_patient_finding_classification,
-)
-
-from .pdf import (
-    PdfReimportView,
-    PdfStreamView,
-)
-from .requirement import (
-    evaluate_requirements,
-    LookupViewSet,
-)
-
-from .video import (
-    # Video Correction (Phase 1.1) - Implemented
-    VideoProcessingHistoryView,
+from .patient_finding import OptimizedPatientFindingViewSet, PatientFindingViewSet
+from .patient_finding_classification import create_patient_finding_classification
+from .report import ReportReimportView, ReportStreamView
+from .requirement import LookupViewSet, evaluate_requirements
+from .video import (  # Video Correction (Phase 1.1) - Implemented; Existing views
     VideoApplyMaskView,
-    VideoRemoveFramesView,
-    
-    # Existing views
+    VideoCorrectionView,
+    VideoExaminationViewSet,
+    VideoProcessingHistoryView,
     VideoReimportView,
-    VideoViewSet,
+    VideoRemoveFramesView,
     VideoStreamView,
-    VideoLabelView,
-    UpdateLabelSegmentsView,
-    rerun_segmentation,
-    VideoExaminationViewSet,
-    VideoCorrectionView,
 )
 
 __all__ = [
@@ -140,126 +90,88 @@ __all__ = [
     "start_anonymization",
     "AnonymizationOverviewView",
     "AnonymizationValidateView",
-
     # Auth views
     "KeycloakVideoView",
     "keycloak_login",
     "keycloak_callback",
     "public_home",
-
     # Examination views
     "ExaminationManifestCache",
-    'ExaminationViewSet',
-
-    'get_classification_choices_for_examination',
-    'get_morphology_classification_choices_for_examination',
-    'get_location_classification_choices_for_examination',
-
-    'get_classifications_for_examination',
-    'get_location_classifications_for_examination',
-    'get_morphology_classifications_for_examination',
-
-    'get_findings_for_examination',
-    'get_instruments_for_examination',
-    'get_interventions_for_examination',
-
+    "ExaminationViewSet",
+    "get_classification_choices_for_examination",
+    "get_morphology_classification_choices_for_examination",
+    "get_location_classification_choices_for_examination",
+    "get_classifications_for_examination",
+    "get_location_classifications_for_examination",
+    "get_morphology_classifications_for_examination",
+    "get_findings_for_examination",
+    "get_instruments_for_examination",
+    "get_interventions_for_examination",
     # Finding Views
     "FindingViewSet",
     "get_interventions_for_finding",
     "get_classifications_for_finding",
-
     # Finding Classification Views
     "FindingClassificationViewSet",
     "get_classification_choices",
-    "get_morphology_choices", # DEPRECATED
-    "get_location_choices", # DEPRECATED
-
-    # Label Video Segment Views
-    'create_video_segment_annotation',
-    'video_segments_by_label_id_view',
-    'video_segments_by_label_name_view',
-    'video_segment_detail_view',
-    'video_segments_view',
-    'update_label_video_segment',
-    "get_lvs_by_name_and_video_id",
-
+    "get_morphology_choices",  # DEPRECATED
+    "get_location_choices",  # DEPRECATED
     # Meta Views
-    "AvailableFilesListView",
-    "SensitiveMetaDetailView",
     "SensitiveMetaListView",
     "SensitiveMetaVerificationView",
-    "ReportFileMetadataView",
-
     # Misc
     "CenterViewSet",
-    'csrf_token_view',
+    "csrf_token_view",
     "GenderViewSet",
-    'ExaminationStatsView',
-    'VideoSegmentStatsView',
-    'SensitiveMetaStatsView',
+    "ExaminationStatsView",
+    "VideoSegmentStatsView",
+    "SensitiveMetaStatsView",
     "GeneralStatsView",
-    "SecureFileUrlView",
-    "SecureFileServingView",
-    "validate_secure_url",
-    'ExaminationTranslationOptions',
-    'FindingTranslationOptions',
-    'FindingClassificationTranslationOptions',
-    'FindingClassificationChoiceTranslationOptions',
-    'InterventionTranslationOptions',
-    'TranslatedFieldMixin',
-    'TranslationMigrationHelper',
-    'TranslatedFixtureLoader',
+    "ExaminationTranslationOptions",
+    "FindingTranslationOptions",
+    "FindingClassificationTranslationOptions",
+    "FindingClassificationChoiceTranslationOptions",
+    "InterventionTranslationOptions",
+    "TranslatedFieldMixin",
+    "TranslationMigrationHelper",
+    "TranslatedFixtureLoader",
     "build_multilingual_response",
-    'MODELTRANSLATION_SETTINGS',
-    'UploadFileView',
-    'UploadStatusView',
-
+    "MODELTRANSLATION_SETTINGS",
+    "UploadFileView",
+    "UploadStatusView",
     # Patient Views
     "PatientViewSet",
-
     # Patient Examination Views
     "ExaminationCreateView",
     "PatientExaminationDetailView",
     "PatientExaminationListView",
     "PatientExaminationViewSet",
-
     # Patient Finding Views
     "PatientFindingViewSet",
     "OptimizedPatientFindingViewSet",
-
     # Patient Finding Classification Views
     "create_patient_finding_classification",
-
-    # PDF
-    "PdfMediaView",
-    "PdfReimportView",
-    "PdfStreamView",
-
-    # Report
-    "ReportListView",
-    "ReportWithSecureUrlView",
-    "start_examination",
-    
+    # report
+    "ReportReimportView",
+    "ReportStreamView",
     # Requirement Views
-    
     "evaluate_requirements",
     "LookupViewSet",
-
     # Video Views (Phase 1.1 - Implemented)
-    'VideoMetadataView',
-    'VideoProcessingHistoryView',
-    'VideoApplyMaskView',
-    'VideoRemoveFramesView',
-
-    'VideoCorrectionView',
-    
-    # Video Views (Existing)
-    'VideoReimportView',
-    'VideoViewSet',
-    'VideoStreamView',
-    'VideoLabelView',
-    'UpdateLabelSegmentsView',
-    'rerun_segmentation',
-    'video_timeline_view',
+    "VideoProcessingHistoryView",
+    "VideoApplyMaskView",
+    "VideoRemoveFramesView",
+    "VideoCorrectionView",
+    "VideoReimportView",
+    "VideoStreamView",
     "VideoExaminationViewSet",
-]
+    "ReportReimportView",
+    "label_list",
+    "get_sensitive_metadata_pk",
+    "video_sensitive_metadata",
+    "video_sensitive_metadata_verify",
+    "pdf_sensitive_metadata",
+    "pdf_sensitive_metadata_verify",
+    "sensitive_metadata_list",
+    "pdf_sensitive_metadata_list",
+]

endoreg_db/views/ai/__init__.py

@@ -0,0 +1,8 @@
+from .label import (label_list, add_label, delete_label, update_label)
+
+__all__ = [
+    "label_list",
+    "add_label",
+    "delete_label",
+    "update_label",
+]   

endoreg_db/views/ai/label.py

@@ -0,0 +1,155 @@
+# endoreg_db/views/media/label_media.py
+import logging
+from rest_framework.decorators import api_view, permission_classes
+from rest_framework.response import Response
+from rest_framework import status
+
+from endoreg_db.models import Label
+from endoreg_db.serializers.label.label import LabelSerializer
+from endoreg_db.utils.permissions import EnvironmentAwarePermission
+# from rest_framework.permissions import IsAuthenticated
+# from endoreg_db.authz.permissions import PolicyPermission
+
+logger = logging.getLogger(__name__)
+
+
+@api_view(["GET"])
+@permission_classes([EnvironmentAwarePermission])
+# or: @permission_classes([IsAuthenticated, PolicyPermission])
+def label_list(request) -> Response:
+    """
+    List all annotation labels used for video segments.
+
+    GET /api/media/labels/
+    Response:
+    [
+      { "id": 1, "name": "polyp" },
+      ...
+    ]
+    """
+    try:
+        labels = Label.objects.all().order_by("name")
+        serializer = LabelSerializer(labels, many=True)
+        return Response(serializer.data, status=status.HTTP_200_OK)
+    except Exception as e:
+        logger.error(f"Error fetching labels: {e}")
+        return Response(
+            {"error": "Failed to fetch labels"},
+            status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        )
+
+@api_view(["POST"])
+@permission_classes([EnvironmentAwarePermission])
+def add_label(request) -> Response:
+    try:
+        name = request.data.get("name")
+        if not name:
+            return Response(
+                {"error": "Field 'name' is required"},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+
+        label, created = Label.get_or_create_from_name(name)
+
+        return Response(
+            {
+                "success": "label added to database" if created else "label already existed",
+                "id": label.id,
+                "name": label.name,
+            },
+            status=status.HTTP_201_CREATED if created else status.HTTP_200_OK,
+        )
+    except Exception as e:
+        logger.error(f"Error creating label: {e}")
+        return Response(
+            {"error": "Failed to create label"},
+            status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        )
+
+
+@api_view(["DELETE"])
+def delete_label(request) -> Response:
+    try:
+        name = request.data.get("name")
+        if not name:
+            return Response(
+                {"error": "Field 'name' is required"},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+        Label.delete(name)
+        if isinstance(Label.get_or_create_from_name(name), Label):
+            return Response(
+                {"error": "Field not deleted"},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR     
+            )
+        else:
+            return Response(
+                {"success": f"label {name} deleted"},
+                status=status.HTTP_200_OK
+            )
+    except Exception as e:
+        logger.error(f"Error creating label: {e}")
+        return Response(
+            {"error": "Failed to create label"},
+            status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        )
+
+
+@api_view(["PATCH", "POST"])
+@permission_classes([EnvironmentAwarePermission])
+def update_label(request) -> Response:
+    """
+    Update/rename a label.
+
+    Body:
+    {
+      "name_old": "polyp_old",
+      "name": "polyp"
+    }
+    """
+    name_old = request.data.get("name_old")
+    new_name = request.data.get("name")
+
+    if not name_old:
+      return Response(
+          {"error": "Field 'name_old' is required"},
+          status=status.HTTP_400_BAD_REQUEST,
+      )
+
+    if not new_name:
+      return Response(
+          {"error": "Field 'name' is required"},
+          status=status.HTTP_400_BAD_REQUEST,
+      )
+
+    try:
+        try:
+            label = Label.objects.get(name=name_old)
+        except Label.DoesNotExist:
+            return Response(
+                {"error": f"Label '{name_old}' not found"},
+                status=status.HTTP_404_NOT_FOUND,
+            )
+
+        # Optional: handle duplicate target names
+        if Label.objects.filter(name=new_name).exclude(pk=label.pk).exists():
+            return Response(
+                {"error": f"Label '{new_name}' already exists"},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+
+        label.name = new_name
+        label.save()
+
+        return Response(
+            {"success": f"Label '{name_old}' renamed to '{new_name}'",
+             "id": label.id,
+             "name": label.name},
+            status=status.HTTP_200_OK,
+        )
+    except Exception as e:
+        logger.error(f"Error updating label '{name_old}' → '{new_name}': {e}")
+        return Response(
+            {"error": "Failed to update label"},
+            status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        )

endoreg_db/views/anonymization/media_management.py

@@ -136,7 +136,7 @@ class MediaManagementView(APIView):
         return stats
 
     def _get_pdf_stats(self) -> Dict[str, int]:
-        """Get PDF file statistics"""
+        """Get report file statistics"""
         pdfs = RawPdfFile.objects.all()
 
         stats = {
@@ -150,7 +150,7 @@ class MediaManagementView(APIView):
         }
 
         for pdf in pdfs:
-            # PDF status logic based on anonymized_text presence and validation
+            # report status logic based on anonymized_text presence and validation
             has_anonymized = bool(pdf.anonymized_text and pdf.anonymized_text.strip())
             is_validated = (
                 getattr(pdf.sensitive_meta, "is_verified", False)
@@ -196,10 +196,11 @@ class MediaManagementView(APIView):
 
         if media_type == "video":
             video_file_obj = (
-                VideoFile.get_video_by_id(self, video_id=file_id) if file_id else None
+                VideoFile.get_video_by_pk(pk=file_id) if file_id else None
             )
+
         elif media_type == "pdf":
-            pdf_file_obj = RawPdfFile.get_pdf_by_id(file_id) if file_id else None
+            pdf_file_obj = RawPdfFile.get_pdf_by_pk(pk=file_id) if file_id else None
 
         with transaction.atomic():
             if video_file_obj:
@@ -414,7 +415,7 @@ def force_remove_media(request, file_id: int):
 
             return Response(
                 {
-                    "detail": f"PDF file '{filename}' (ID: {file_id}) removed successfully",
+                    "detail": f"report file '{filename}' (ID: {file_id}) removed successfully",
                     "file_type": "pdf",
                     "file_id": file_id,
                 }
@@ -462,7 +463,7 @@ def reset_processing_status(request, file_id: int):
         except VideoFile.DoesNotExist:
             pass
 
-        # PDF files don't have state, but we can clear anonymized_text
+        # report files don't have state, but we can clear anonymized_text
         try:
             pdf = RawPdfFile.objects.get(id=file_id)
             pdf.anonymized_text = ""
@@ -470,7 +471,7 @@ def reset_processing_status(request, file_id: int):
 
             return Response(
                 {
-                    "detail": "PDF file processing reset",
+                    "detail": "report file processing reset",
                     "file_type": "pdf",
                     "file_id": file_id,
                     "new_status": "not_started",