endoreg-db 0.1.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

endoreg_db/queries/get/patient_examination.py

@@ -0,0 +1,20 @@
+from endoreg_db.models import PatientExamination
+from typing import Optional
+
+def get_patient_examinations() -> PatientExamination:
+    """
+    Returns all PatientExamination objects from the database.
+    """
+    return PatientExamination.objects.all()
+
+def get_patient_examinations_without_report_file() -> PatientExamination:
+    """
+    Returns all PatientExamination objects from the database without a report_file.
+    """
+    return PatientExamination.objects.filter(report_file__isnull=True)
+
+def get_patient_examinations_without_video() -> PatientExamination:
+    """
+    Returns all PatientExamination objects from the database without a video.
+    """
+    return PatientExamination.objects.filter(video__isnull=True)

endoreg_db/queries/get/report_file.py

@@ -0,0 +1,33 @@
+from endoreg_db.models import ReportFile
+from typing import Optional
+
+def get_report_files() -> ReportFile:
+    """
+    Returns all ReportFile objects from the database.
+    """
+    return ReportFile.objects.all()
+
+def get_report_file_by_id(id) -> Optional[ReportFile]:
+    """Retrieve a ReportFile object by its id.
+
+    Args:
+        id (int): The id of the report_file to retrieve.
+
+    Returns:
+        Optional[ReportFile]: The ReportFile object with the given id, or None if it does not exist.
+    """
+    return ReportFile.objects.get(id=id)
+
+def get_report_files_without_examination() -> ReportFile:
+    """
+    Returns all ReportFile objects from the database without an examination.
+    """
+    return ReportFile.objects.filter(patient_examination__isnull=True)
+
+def get_report_files_without_patient() -> ReportFile:
+    """
+    Returns all ReportFile objects from the database without a patient.
+    """
+    return ReportFile.objects.filter(patient__isnull=True)
+
+

endoreg_db/queries/get/video.py

@@ -0,0 +1,31 @@
+from endoreg_db.models import Video, PatientExamination
+from typing import Optional
+
+def get_videos() -> Video:
+    """
+    Returns all Video objects from the database.
+    """
+    return Video.objects.all()
+
+def get_video_by_id(id) -> Optional[Video]:
+    """Retrieve a Video object by its id.
+
+    Args:
+        id (int): The id of the video to retrieve.
+
+    Returns:
+        Optional[Video]: The Video object with the given id, or None if it does not exist.
+    """
+    return Video.objects.get(id=id)
+
+def get_video_without_examination() -> Video:
+    """
+    Returns all Video objects from the database without an examination.
+    """
+    return Video.objects.filter(patient_examination__isnull=True)
+
+def get_video_without_patient() -> Video:
+    """
+    Returns all Video objects from the database without a patient.
+    """
+    return Video.objects.filter(patient__isnull=True)

endoreg_db/serializers/__init__.py

@@ -0,0 +1,10 @@
+from .ai_model import *
+from .annotation import *
+from .center import *
+from .examination import *
+from .frame import *
+from .hardware import *
+from .label import *
+from .patient import *
+from .report_file import *
+from .video import *

endoreg_db/serializers/ai_model.py

@@ -0,0 +1,19 @@
+from endoreg_db.models import ModelMeta, ModelType, ActiveModel
+from rest_framework import serializers
+
+class ModelMetaSerializer(serializers.ModelSerializer):
+    weights = serializers.FileField(use_url=True)  # use_url=True will provide the file's URI
+
+    class Meta:
+        model = ModelMeta
+        fields = '__all__'
+
+class ModelTypeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ModelType
+        fields = '__all__'
+
+class ActiveModelSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ActiveModel
+        fields = '__all__'

endoreg_db/serializers/annotation.py

@@ -0,0 +1,17 @@
+from rest_framework import serializers
+from endoreg_db.models import LegacyBinaryClassificationAnnotationTask, BinaryClassificationAnnotationTask, ImageClassificationAnnotation
+
+class LegacyBinaryClassificationAnnotationTaskSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LegacyBinaryClassificationAnnotationTask
+        fields = '__all__'
+
+class BinaryClassificationAnnotationTaskSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = BinaryClassificationAnnotationTask
+        fields = '__all__'
+
+class ImageClassificationAnnotationSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ImageClassificationAnnotation
+        fields = '__all__'

endoreg_db/serializers/center.py

@@ -0,0 +1,11 @@
+from rest_framework import serializers
+from endoreg_db.models import (
+    Center,
+)
+
+class CenterSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Center
+        fields = '__all__'
+
+

endoreg_db/serializers/examination.py

@@ -0,0 +1,33 @@
+from rest_framework import serializers
+from endoreg_db.models import (
+    Examination,
+    ExaminationTimeType,
+    ExaminationTime,
+    ExaminationType,
+    PatientExamination
+)
+
+class ExaminationTimeTypeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ExaminationTimeType
+        fields = '__all__'
+
+class ExaminationTimeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ExaminationTime
+        fields = '__all__'
+
+class ExaminationTypeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ExaminationType
+        fields = '__all__'
+
+class ExaminationSerializer(serializers.ModelSerializer): 
+    class Meta:
+        model = Examination
+        fields = '__all__'
+
+class PatientExaminationSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = PatientExamination
+        fields = '__all__'

endoreg_db/serializers/frame.py

@@ -0,0 +1,13 @@
+from rest_framework import serializers
+from endoreg_db.models import Frame, LegacyFrame
+
+class FrameSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Frame
+        fields = '__all__'
+
+class LegacyFrameSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LegacyFrame
+        fields = '__all__'
+

endoreg_db/serializers/hardware.py

@@ -0,0 +1,21 @@
+from rest_framework import serializers
+from endoreg_db.models import (
+    Endoscope,
+    EndoscopeType,
+    EndoscopyProcessor,
+)
+
+class EndoscopeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Endoscope
+        fields = '__all__'
+
+class EndoscopyProcessorSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = EndoscopyProcessor
+        fields = '__all__'
+
+class EndoscopeTypeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = EndoscopeType
+        fields = '__all__'

endoreg_db/serializers/label.py

@@ -0,0 +1,22 @@
+from rest_framework import serializers
+from endoreg_db.models import (
+    Label,
+    LabelSet,
+    LabelType,
+)
+
+class LabelTypeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LabelType
+        fields = '__all__'
+
+class LabelSetSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LabelSet
+        fields = '__all__'
+
+class LabelSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Label
+        fields = '__all__'
+        

endoreg_db/serializers/patient.py

@@ -0,0 +1,10 @@
+from rest_framework import serializers
+from endoreg_db.models import (
+    Patient,
+)
+
+class PatientSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Patient
+        fields = '__all__'
+

endoreg_db/serializers/prediction.py

@@ -0,0 +1,15 @@
+from rest_framework import serializers
+from endoreg_db.models import (
+    ImageClassificationPrediction,
+    LegacyVideoPredictionMeta
+)
+
+class ImageClassificationPredictionSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ImageClassificationPrediction
+        fields = '__all__'
+
+class LegacyVideoPredictionMetaSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LegacyVideoPredictionMeta
+        fields = '__all__'

endoreg_db/serializers/report_file.py

@@ -0,0 +1,7 @@
+from rest_framework import serializers
+from endoreg_db.models import ReportFile
+
+class ReportFileSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ReportFile
+        fields = '__all__'

endoreg_db/serializers/video.py

@@ -0,0 +1,27 @@
+from rest_framework import serializers
+from endoreg_db.models import Video, LegacyVideo, VideoImportMeta, LegacyLabelVideoSegment, LabelVideoSegment
+
+class VideoSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Video
+        fields = '__all__'
+
+class LegacyVideoSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LegacyVideo
+        fields = '__all__'
+
+class VideoImportMetaSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = VideoImportMeta
+        fields = '__all__'
+
+class LabelVideoSegmentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LabelVideoSegment
+        fields = '__all__'
+
+class LegacyLabelVideoSegmentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LegacyLabelVideoSegment
+        fields = '__all__'