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

endoreg_db/views/stats/stats_views.py

@@ -4,20 +4,21 @@ Statistics views for the endoreg_db API.
 Provides dashboard statistics for the frontend including examination stats,
 video segment stats, sensitive meta stats, and general overview stats.
 """
+
 from rest_framework.views import APIView
 from rest_framework.response import Response
 from rest_framework import status
-from django.db.models import Count, Q
+from django.db.models import Count
 from django.utils import timezone
 from datetime import timedelta
 
 from ...models import (
-    VideoFile, 
-    LabelVideoSegment, 
-    SensitiveMeta, 
+    VideoFile,
+    LabelVideoSegment,
+    SensitiveMeta,
     Examination,
     PatientExamination,
-    Label
+    Label,
 )
 from ...utils.permissions import EnvironmentAwarePermission
 
@@ -25,11 +26,12 @@ from ...utils.permissions import EnvironmentAwarePermission
 class ExaminationStatsView(APIView):
     """
     Statistics about examinations and patient examinations.
-    
+
     GET /api/examinations/stats/
     """
+
     permission_classes = [EnvironmentAwarePermission]
-    
+
     def get(self, request):
         """
         Returns statistics about examinations including total counts,
@@ -39,40 +41,51 @@ class ExaminationStatsView(APIView):
             # Total examination counts
             total_examinations = Examination.objects.count()
             total_patient_examinations = PatientExamination.objects.count()
-            
+
             # Recent activity (last 30 days)
             thirty_days_ago = timezone.now() - timedelta(days=30)
-            recent_examinations = Examination.objects.filter(
-                created_at__gte=thirty_days_ago
-            ).count() if hasattr(Examination, 'created_at') else 0
-            
+            recent_examinations = (
+                Examination.objects.filter(created_at__gte=thirty_days_ago).count()
+                if hasattr(Examination, "created_at")
+                else 0
+            )
+
             # Status distribution for patient examinations
-            status_distribution = PatientExamination.objects.values('status').annotate(
-                count=Count('id')
-            ).order_by('status') if hasattr(PatientExamination, 'status') else []
-            
-            return Response({
-                'total_examinations': total_examinations,
-                'total_patient_examinations': total_patient_examinations,
-                'recent_examinations': recent_examinations,
-                'status_distribution': list(status_distribution)
-            }, status=status.HTTP_200_OK)
-            
+            status_distribution = (
+                PatientExamination.objects.values("status")
+                .annotate(count=Count("id"))
+                .order_by("status")
+                if hasattr(PatientExamination, "status")
+                else []
+            )
+
+            return Response(
+                {
+                    "total_examinations": total_examinations,
+                    "total_patient_examinations": total_patient_examinations,
+                    "recent_examinations": recent_examinations,
+                    "status_distribution": list(status_distribution),
+                },
+                status=status.HTTP_200_OK,
+            )
+
         except Exception as e:
-            return Response({
-                'error': f'Failed to retrieve examination stats: {str(e)}'
-            }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            return Response(
+                {"error": f"Failed to retrieve examination stats: {str(e)}"},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
 
 
 class VideoSegmentStatsView(APIView):
     """
     Statistics about video segments and label distribution.
-    
+
     GET /api/video-segment/stats/
     GET /api/video-segments/stats/
     """
+
     permission_classes = [EnvironmentAwarePermission]
-    
+
     def get(self, request):
         """
         Returns statistics about video segments including total counts,
@@ -81,49 +94,58 @@ class VideoSegmentStatsView(APIView):
         try:
             # Total segment counts
             total_segments = LabelVideoSegment.objects.count()
-            total_videos_with_segments = VideoFile.objects.filter(
-                labelvideosegment__isnull=False
-            ).distinct().count()
-            
+            total_videos_with_segments = (
+                VideoFile.objects.filter(labelvideosegment__isnull=False)
+                .distinct()
+                .count()
+            )
+
             # Label distribution
-            label_distribution = LabelVideoSegment.objects.values(
-                'label__name'
-            ).annotate(
-                count=Count('id')
-            ).order_by('-count')
-            
+            label_distribution = (
+                LabelVideoSegment.objects.values("label__name")
+                .annotate(count=Count("id"))
+                .order_by("-count")
+            )
+
             # Videos without segments
             videos_without_segments = VideoFile.objects.filter(
                 labelvideosegment__isnull=True
             ).count()
-            
+
             # Average segments per video
             avg_segments_per_video = 0
             if total_videos_with_segments > 0:
-                avg_segments_per_video = round(total_segments / total_videos_with_segments, 2)
-            
-            return Response({
-                'total_segments': total_segments,
-                'total_videos_with_segments': total_videos_with_segments,
-                'videos_without_segments': videos_without_segments,
-                'avg_segments_per_video': avg_segments_per_video,
-                'label_distribution': list(label_distribution)
-            }, status=status.HTTP_200_OK)
-            
+                avg_segments_per_video = round(
+                    total_segments / total_videos_with_segments, 2
+                )
+
+            return Response(
+                {
+                    "total_segments": total_segments,
+                    "total_videos_with_segments": total_videos_with_segments,
+                    "videos_without_segments": videos_without_segments,
+                    "avg_segments_per_video": avg_segments_per_video,
+                    "label_distribution": list(label_distribution),
+                },
+                status=status.HTTP_200_OK,
+            )
+
         except Exception as e:
-            return Response({
-                'error': f'Failed to retrieve video segment stats: {str(e)}'
-            }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            return Response(
+                {"error": f"Failed to retrieve video segment stats: {str(e)}"},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
 
 
 class SensitiveMetaStatsView(APIView):
     """
     Statistics about sensitive metadata entries and verification status.
-    
+
     GET /api/video/sensitivemeta/stats/
     """
+
     permission_classes = [EnvironmentAwarePermission]
-    
+
     def get(self, request):
         """
         Returns statistics about sensitive metadata including verification status,
@@ -132,50 +154,57 @@ class SensitiveMetaStatsView(APIView):
         try:
             # Total sensitive meta entries
             total_sensitive_meta = SensitiveMeta.objects.count()
-            
+
             # Verification status distribution
-            verified_count = SensitiveMeta.objects.filter(
-                verified=True
-            ).count() if hasattr(SensitiveMeta, 'verified') else 0
-            
+            verified_count = (
+                SensitiveMeta.objects.filter(verified=True).count()
+                if hasattr(SensitiveMeta, "verified")
+                else 0
+            )
+
             unverified_count = total_sensitive_meta - verified_count
-            
+
             # Videos with sensitive data
-            videos_with_sensitive_data = VideoFile.objects.filter(
-                sensitivemeta__isnull=False
-            ).distinct().count()
-            
+            videos_with_sensitive_data = (
+                VideoFile.objects.filter(sensitivemeta__isnull=False).distinct().count()
+            )
+
             # Type distribution (if available)
             type_distribution = []
-            if hasattr(SensitiveMeta, 'meta_type'):
-                type_distribution = SensitiveMeta.objects.values(
-                    'meta_type'
-                ).annotate(
-                    count=Count('id')
-                ).order_by('-count')
-            
-            return Response({
-                'total_sensitive_meta': total_sensitive_meta,
-                'verified_count': verified_count,
-                'unverified_count': unverified_count,
-                'videos_with_sensitive_data': videos_with_sensitive_data,
-                'type_distribution': list(type_distribution)
-            }, status=status.HTTP_200_OK)
-            
+            if hasattr(SensitiveMeta, "meta_type"):
+                type_distribution = (
+                    SensitiveMeta.objects.values("meta_type")
+                    .annotate(count=Count("id"))
+                    .order_by("-count")
+                )
+
+            return Response(
+                {
+                    "total_sensitive_meta": total_sensitive_meta,
+                    "verified_count": verified_count,
+                    "unverified_count": unverified_count,
+                    "videos_with_sensitive_data": videos_with_sensitive_data,
+                    "type_distribution": list(type_distribution),
+                },
+                status=status.HTTP_200_OK,
+            )
+
         except Exception as e:
-            return Response({
-                'error': f'Failed to retrieve sensitive meta stats: {str(e)}'
-            }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            return Response(
+                {"error": f"Failed to retrieve sensitive meta stats: {str(e)}"},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
 
 
 class GeneralStatsView(APIView):
     """
     General dashboard statistics providing an overview of the system.
-    
+
     GET /api/stats/
     """
+
     permission_classes = [EnvironmentAwarePermission]
-    
+
     def get(self, request):
         """
         Returns general overview statistics for the dashboard including
@@ -188,42 +217,50 @@ class GeneralStatsView(APIView):
             total_segments = LabelVideoSegment.objects.count()
             total_examinations = Examination.objects.count()
             total_sensitive_meta = SensitiveMeta.objects.count()
-            
+
             # Processing status
-            videos_with_segments = VideoFile.objects.filter(
-                labelvideosegment__isnull=False
-            ).distinct().count()
-            
-            videos_with_sensitive_data = VideoFile.objects.filter(
-                sensitivemeta__isnull=False
-            ).distinct().count()
-            
+            videos_with_segments = (
+                VideoFile.objects.filter(labelvideosegment__isnull=False)
+                .distinct()
+                .count()
+            )
+
+            videos_with_sensitive_data = (
+                VideoFile.objects.filter(sensitivemeta__isnull=False).distinct().count()
+            )
+
             # Calculate percentages
             segment_coverage = 0
             sensitive_data_coverage = 0
             if total_videos > 0:
                 segment_coverage = round((videos_with_segments / total_videos) * 100, 1)
-                sensitive_data_coverage = round((videos_with_sensitive_data / total_videos) * 100, 1)
-            
-            return Response({
-                'totals': {
-                    'videos': total_videos,
-                    'labels': total_labels,
-                    'segments': total_segments,
-                    'examinations': total_examinations,
-                    'sensitive_meta': total_sensitive_meta
-                },
-                'coverage': {
-                    'segment_coverage_percent': segment_coverage,
-                    'sensitive_data_coverage_percent': sensitive_data_coverage
+                sensitive_data_coverage = round(
+                    (videos_with_sensitive_data / total_videos) * 100, 1
+                )
+
+            return Response(
+                {
+                    "totals": {
+                        "videos": total_videos,
+                        "labels": total_labels,
+                        "segments": total_segments,
+                        "examinations": total_examinations,
+                        "sensitive_meta": total_sensitive_meta,
+                    },
+                    "coverage": {
+                        "segment_coverage_percent": segment_coverage,
+                        "sensitive_data_coverage_percent": sensitive_data_coverage,
+                    },
+                    "processing_status": {
+                        "videos_with_segments": videos_with_segments,
+                        "videos_with_sensitive_data": videos_with_sensitive_data,
+                    },
                 },
-                'processing_status': {
-                    'videos_with_segments': videos_with_segments,
-                    'videos_with_sensitive_data': videos_with_sensitive_data
-                }
-            }, status=status.HTTP_200_OK)
-            
+                status=status.HTTP_200_OK,
+            )
+
         except Exception as e:
-            return Response({
-                'error': f'Failed to retrieve general stats: {str(e)}'
-            }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            return Response(
+                {"error": f"Failed to retrieve general stats: {str(e)}"},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )

endoreg_db/views/video/__init__.py

@@ -7,9 +7,7 @@ from .correction import (
 
 from ..media.video_media import VideoMediaView
 
-from .reimport import (
-    VideoReimportView
-)
+from .reimport import VideoReimportView
 from .video_stream import VideoStreamView
 from .video_examination_viewset import VideoExaminationViewSet
 from .video_meta_stats import VideoMetadataStatsView
@@ -24,35 +22,28 @@ from .segments_crud import (
     video_segments_validate_bulk,
     video_segments_validation_status,
 )
-from ..ai.label import label_list
+from .ai.label import label_list
 
 __all__ = [
     # Video Correction (Phase 1.1) - Implemented
-    'VideoProcessingHistoryView',
-    'VideoApplyMaskView',
-    'VideoRemoveFramesView',
-    'VideoMetadataStatsView',
-    
-    'VideoMediaView',
-    
-    'VideoCorrectionView',
-
+    "VideoProcessingHistoryView",
+    "VideoApplyMaskView",
+    "VideoRemoveFramesView",
+    "VideoMetadataStatsView",
+    "VideoMediaView",
+    "VideoCorrectionView",
     # Reimport views
-    'VideoReimportView',
-
+    "VideoReimportView",
     # Segmentation views
-    'VideoStreamView',
-
+    "VideoStreamView",
     # Video Examination ViewSet
     "VideoExaminationViewSet",
-    
-    'video_segments_collection',
-    'video_segments_by_video',
-    'video_segment_detail',
-    'video_segments_stats',
-    'video_segment_validate',
-    'video_segments_validate_bulk',
-    'video_segments_validation_status',
-    
-    'label_list',
+    "video_segments_collection",
+    "video_segments_by_video",
+    "video_segment_detail",
+    "video_segments_stats",
+    "video_segment_validate",
+    "video_segments_validate_bulk",
+    "video_segments_validation_status",
+    "label_list",
 ]

endoreg_db/views/{ai → video/ai}/__init__.py

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

endoreg_db/views/{ai → video/ai}/label.py

@@ -5,7 +5,7 @@ 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.serializers.label_video_segment.label import LabelSerializer
 from endoreg_db.utils.permissions import EnvironmentAwarePermission
 # from rest_framework.permissions import IsAuthenticated
 # from endoreg_db.authz.permissions import PolicyPermission
@@ -38,6 +38,7 @@ def label_list(request) -> Response:
             status=status.HTTP_500_INTERNAL_SERVER_ERROR,
         )
 
+
 @api_view(["POST"])
 @permission_classes([EnvironmentAwarePermission])
 def add_label(request) -> Response:
@@ -53,7 +54,9 @@ def add_label(request) -> Response:
 
         return Response(
             {
-                "success": "label added to database" if created else "label already existed",
+                "success": "label added to database"
+                if created
+                else "label already existed",
                 "id": label.id,
                 "name": label.name,
             },
@@ -80,12 +83,11 @@ def delete_label(request) -> Response:
         if isinstance(Label.get_or_create_from_name(name), Label):
             return Response(
                 {"error": "Field not deleted"},
-                status=status.HTTP_500_INTERNAL_SERVER_ERROR     
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
             )
         else:
             return Response(
-                {"success": f"label {name} deleted"},
-                status=status.HTTP_200_OK
+                {"success": f"label {name} deleted"}, status=status.HTTP_200_OK
             )
     except Exception as e:
         logger.error(f"Error creating label: {e}")
@@ -111,16 +113,16 @@ def update_label(request) -> Response:
     new_name = request.data.get("name")
 
     if not name_old:
-      return Response(
-          {"error": "Field 'name_old' is required"},
-          status=status.HTTP_400_BAD_REQUEST,
-      )
+        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,
-      )
+        return Response(
+            {"error": "Field 'name' is required"},
+            status=status.HTTP_400_BAD_REQUEST,
+        )
 
     try:
         try:
@@ -142,9 +144,11 @@ def update_label(request) -> Response:
         label.save()
 
         return Response(
-            {"success": f"Label '{name_old}' renamed to '{new_name}'",
-             "id": label.id,
-             "name": label.name},
+            {
+                "success": f"Label '{name_old}' renamed to '{new_name}'",
+                "id": label.id,
+                "name": label.name,
+            },
             status=status.HTTP_200_OK,
         )
     except Exception as e:

endoreg_db/views/video/correction.py

@@ -31,7 +31,7 @@ from endoreg_db.models import (
 )
 from endoreg_db.serializers import VideoMetaSerializer, VideoProcessingHistorySerializer
 from endoreg_db.serializers.video.video_file_detail import VideoDetailSerializer
-from endoreg_db.utils.paths import ANONYM_VIDEO_DIR, IMPORT_VIDEO_DIR
+from endoreg_db.utils.paths import ANONYM_VIDEO_DIR
 from endoreg_db.utils.permissions import EnvironmentAwarePermission
 
 logger = logging.getLogger(__name__)
@@ -123,7 +123,7 @@ def update_segments_after_frame_removal(video: VideoFile, removed_frames: list)
         # Delete segment if all frames were removed
         if new_start >= new_end:
             logger.info(
-                f"Deleting segment {segment.id} (original: {original_start}-{original_end}) "
+                f"Deleting segment {segment.pk} (original: {original_start}-{original_end}) "
                 f"- all {frames_within} frames removed"
             )
             segment.delete()
@@ -131,7 +131,7 @@ def update_segments_after_frame_removal(video: VideoFile, removed_frames: list)
         elif new_start != original_start or new_end != original_end:
             # Update segment boundaries
             logger.info(
-                f"Updating segment {segment.id}: "
+                f"Updating segment {segment.pk}: "
                 f"{original_start}-{original_end} → {new_start}-{new_end} "
                 f"(before: {frames_before}, within: {frames_within})"
             )
@@ -144,7 +144,7 @@ def update_segments_after_frame_removal(video: VideoFile, removed_frames: list)
             segments_unchanged += 1
 
     logger.info(
-        f"Segment update complete for video {video.id}: "
+        f"Segment update complete for video {video.pk}: "
         f"{segments_updated} updated, {segments_deleted} deleted, {segments_unchanged} unchanged"
     )
 
@@ -336,7 +336,6 @@ class VideoApplyMaskView(APIView):
 
             if success:
                 # Update video record with anonymized file
-                from django.core.files import File
 
                 processed_file_path = output_path
                 update_processed_file(video, processed_file_path)
@@ -472,7 +471,7 @@ class VideoRemoveFramesView(APIView):
             output_path = (
                 Path(settings.MEDIA_ROOT)
                 / "anonym_videos"
-                / f"{video.uuid}_cleaned.mp4"
+                / f"{video.video_hash}_cleaned.mp4"
             )
             output_path.parent.mkdir(parents=True, exist_ok=True)