endoreg-db 0.3.5__py3-none-any.whl → 0.3.6__py3-none-any.whl

endoreg_db/models/label/label.py

@@ -1,84 +1,84 @@
-from django.db import models
-
-class LabelTypeManager(models.Manager):
-    def get_by_natural_key(self, name):
-        return self.get(name=name)
-
-class LabelType(models.Model):
-    """
-    A class representing a label type.
-
-    Attributes:
-        name (str): The name of the label type.
-        description (str): A description of the label type.
-
-    """
-    name = models.CharField(max_length=255)
-    description = models.TextField(blank=True, null=True)
-
-    objects = LabelTypeManager()
-
-    def natural_key(self):
-        return (self.name,)
-
-    def __str__(self):
-        return self.name
-
-
-class LabelManager(models.Manager):
-    def get_by_natural_key(self, name):
-        return self.get(name=name)
-
-class Label(models.Model):
-    """
-    A class representing a label.
-
-    Attributes:
-        name (str): The name of the label.
-        label_type (LabelType): The type of the label.
-        description (str): A description of the label.
-
-    """
-    name = models.CharField(max_length=255)
-    label_type = models.ForeignKey("LabelType", on_delete=models.CASCADE, related_name="labels")
-    description = models.TextField(blank=True, null=True)
-
-    objects = LabelManager()
-
-    def natural_key(self):
-        return (self.name,)
-
-    def __str__(self):
-        return self.name
-
-class LabelSetManager(models.Manager):
-    def get_by_natural_key(self, name):
-        return self.get(name=name)
-
-class LabelSet(models.Model):
-    """
-    A class representing a label set.
-
-    Attributes:
-        name (str): The name of the label set.
-        description (str): A description of the label set.
-
-    """
-    name = models.CharField(max_length=255)
-    description = models.TextField(blank=True, null=True)
-    version = models.IntegerField()
-    labels = models.ManyToManyField("Label", related_name="labels")
-    
-    objects = LabelSetManager()
-
-    def natural_key(self):
-        return (self.name,)
-
-    def __str__(self):
-        return self.name
-    
-    def get_labels_in_order(self):
-        """
-        Get all labels in this label set as list in the correct order.
-        """
-        return list(self.labels.all().order_by('id'))
+from django.db import models
+
+class LabelTypeManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
+class LabelType(models.Model):
+    """
+    A class representing a label type.
+
+    Attributes:
+        name (str): The name of the label type.
+        description (str): A description of the label type.
+
+    """
+    name = models.CharField(max_length=255)
+    description = models.TextField(blank=True, null=True)
+
+    objects = LabelTypeManager()
+
+    def natural_key(self):
+        return (self.name,)
+
+    def __str__(self):
+        return self.name
+
+
+class LabelManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
+class Label(models.Model):
+    """
+    A class representing a label.
+
+    Attributes:
+        name (str): The name of the label.
+        label_type (LabelType): The type of the label.
+        description (str): A description of the label.
+
+    """
+    name = models.CharField(max_length=255)
+    label_type = models.ForeignKey("LabelType", on_delete=models.CASCADE, related_name="labels")
+    description = models.TextField(blank=True, null=True)
+
+    objects = LabelManager()
+
+    def natural_key(self):
+        return (self.name,)
+
+    def __str__(self):
+        return self.name
+
+class LabelSetManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
+class LabelSet(models.Model):
+    """
+    A class representing a label set.
+
+    Attributes:
+        name (str): The name of the label set.
+        description (str): A description of the label set.
+
+    """
+    name = models.CharField(max_length=255)
+    description = models.TextField(blank=True, null=True)
+    version = models.IntegerField()
+    labels = models.ManyToManyField("Label", related_name="labels")
+    
+    objects = LabelSetManager()
+
+    def natural_key(self):
+        return (self.name,)
+
+    def __str__(self):
+        return self.name
+    
+    def get_labels_in_order(self):
+        """
+        Get all labels in this label set as list in the correct order.
+        """
+        return list(self.labels.all().order_by('id'))

endoreg_db/models/laboratory/lab_value.py

@@ -1,102 +1,102 @@
-from django.db import models
-import warnings
-
-LANG = "de"
-
-class LabValueManager(models.Manager):
-    def get_by_natural_key(self, name):
-        return self.get(name=name)
-
-class LabValue(models.Model):
-    name = models.CharField(max_length=255, unique=True)
-    name_de = models.CharField(max_length=255, blank=True, null=True)
-    name_en = models.CharField(max_length=255, blank=True, null=True)
-    abbreviation = models.CharField(max_length=10, blank=True, null=True)
-    default_unit = models.ForeignKey('Unit', on_delete=models.CASCADE, blank=True, null=True)
-    numeric_precision = models.IntegerField(default=3)
-    default_single_categorical_value_distribution = models.ForeignKey(
-        'SingleCategoricalValueDistribution',
-        on_delete=models.CASCADE,
-        blank=True, null=True,
-        related_name='default_single_categorical_value_distribution'
-    )
-    default_numerical_value_distribution = models.ForeignKey(
-        'NumericValueDistribution',
-        on_delete=models.CASCADE,
-        blank=True, null=True,
-        related_name='default_numerical_value_distribution'
-    )
-    default_multiple_categorical_value_distribution = models.ForeignKey(
-        'MultipleCategoricalValueDistribution',
-        on_delete=models.CASCADE,
-        blank=True, null=True,
-        related_name='default_multiple_categorical_value_distribution'
-    )
-    default_date_value_distribution = models.ForeignKey(
-        'DateValueDistribution',
-        on_delete=models.CASCADE,
-        blank=True, null=True,
-        related_name='default_date_value_distribution'
-    )
-    default_normal_range = models.JSONField(blank=True, null=True)
-    normal_range_age_dependent = models.BooleanField(default=False)
-    normal_range_gender_dependent = models.BooleanField(default=False)
-    normal_range_special_case = models.BooleanField(default=False)
-    objects = LabValueManager()
-
-    def natural_key(self):
-        return (self.name,)
-
-    def __str__(self):
-        return self.name
-    
-    def get_default_default_distribution(self):
-        if self.default_single_categorical_value_distribution:
-            return self.default_single_categorical_value_distribution
-        elif self.default_numerical_value_distribution:
-            return self.default_numerical_value_distribution
-        elif self.default_multiple_categorical_value_distribution:
-            return self.default_multiple_categorical_value_distribution
-        elif self.default_date_value_distribution:
-            return self.default_date_value_distribution
-        else:
-            warnings.warn("No default distribution set for lab value")
-            return None
-    
-    def get_normal_range(self, age=None, gender=None):
-        age_dependent = self.normal_range_age_dependent
-        gender_dependent = self.normal_range_gender_dependent
-        special_case = self.normal_range_special_case
-
-        min_value = None
-        max_value = None
-
-        if not age_dependent and \
-        not gender_dependent and \
-        not special_case:
-            min_value = self.default_normal_range.get('min', None)
-            max_value = self.default_normal_range.get('max', None)
-
-        if age_dependent: 
-            # get normal range for age)
-            warnings.warn("Age dependent normal range not implemented yet")
-            pass
-        
-        if gender_dependent:
-            if not gender:
-                warnings.warn("Calling get_normal_range with gender_dependent=True requires gender to be set, choosing by random")
-                # set gender to either "male" or "female"
-                from random import choice
-                choices = ["male", "female"]
-                gender = choice(choices)
-
-            default_range_dict = self.default_normal_range.get(gender, {})
-            min_value = default_range_dict.get('min', None)
-            max_value = default_range_dict.get('max', None)
-
-        if special_case:
-            # get normal range for special case
-            warnings.warn("Special case normal range not implemented yet")
-
-        return min_value, max_value
-
+from django.db import models
+import warnings
+
+LANG = "de"
+
+class LabValueManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
+class LabValue(models.Model):
+    name = models.CharField(max_length=255, unique=True)
+    name_de = models.CharField(max_length=255, blank=True, null=True)
+    name_en = models.CharField(max_length=255, blank=True, null=True)
+    abbreviation = models.CharField(max_length=10, blank=True, null=True)
+    default_unit = models.ForeignKey('Unit', on_delete=models.CASCADE, blank=True, null=True)
+    numeric_precision = models.IntegerField(default=3)
+    default_single_categorical_value_distribution = models.ForeignKey(
+        'SingleCategoricalValueDistribution',
+        on_delete=models.CASCADE,
+        blank=True, null=True,
+        related_name='default_single_categorical_value_distribution'
+    )
+    default_numerical_value_distribution = models.ForeignKey(
+        'NumericValueDistribution',
+        on_delete=models.CASCADE,
+        blank=True, null=True,
+        related_name='default_numerical_value_distribution'
+    )
+    default_multiple_categorical_value_distribution = models.ForeignKey(
+        'MultipleCategoricalValueDistribution',
+        on_delete=models.CASCADE,
+        blank=True, null=True,
+        related_name='default_multiple_categorical_value_distribution'
+    )
+    default_date_value_distribution = models.ForeignKey(
+        'DateValueDistribution',
+        on_delete=models.CASCADE,
+        blank=True, null=True,
+        related_name='default_date_value_distribution'
+    )
+    default_normal_range = models.JSONField(blank=True, null=True)
+    normal_range_age_dependent = models.BooleanField(default=False)
+    normal_range_gender_dependent = models.BooleanField(default=False)
+    normal_range_special_case = models.BooleanField(default=False)
+    objects = LabValueManager()
+
+    def natural_key(self):
+        return (self.name,)
+
+    def __str__(self):
+        return self.name
+    
+    def get_default_default_distribution(self):
+        if self.default_single_categorical_value_distribution:
+            return self.default_single_categorical_value_distribution
+        elif self.default_numerical_value_distribution:
+            return self.default_numerical_value_distribution
+        elif self.default_multiple_categorical_value_distribution:
+            return self.default_multiple_categorical_value_distribution
+        elif self.default_date_value_distribution:
+            return self.default_date_value_distribution
+        else:
+            warnings.warn("No default distribution set for lab value")
+            return None
+    
+    def get_normal_range(self, age=None, gender=None):
+        age_dependent = self.normal_range_age_dependent
+        gender_dependent = self.normal_range_gender_dependent
+        special_case = self.normal_range_special_case
+
+        min_value = None
+        max_value = None
+
+        if not age_dependent and \
+        not gender_dependent and \
+        not special_case:
+            min_value = self.default_normal_range.get('min', None)
+            max_value = self.default_normal_range.get('max', None)
+
+        if age_dependent: 
+            # get normal range for age)
+            warnings.warn("Age dependent normal range not implemented yet")
+            pass
+        
+        if gender_dependent:
+            if not gender:
+                warnings.warn("Calling get_normal_range with gender_dependent=True requires gender to be set, choosing by random")
+                # set gender to either "male" or "female"
+                from random import choice
+                choices = ["male", "female"]
+                gender = choice(choices)
+
+            default_range_dict = self.default_normal_range.get(gender, {})
+            min_value = default_range_dict.get('min', None)
+            max_value = default_range_dict.get('max', None)
+
+        if special_case:
+            # get normal range for special case
+            warnings.warn("Special case normal range not implemented yet")
+
+        return min_value, max_value
+

endoreg_db/models/legacy_data/__init__.py

@@ -1,3 +1,3 @@
-from ..data_file.frame import LegacyFrame
-from .image import LegacyImage
-from ..data_file.video import LegacyVideo
+from ..data_file.frame import LegacyFrame
+from .image import LegacyImage
+from ..data_file.video import LegacyVideo

endoreg_db/models/legacy_data/image.py

@@ -1,34 +1,34 @@
-from django.db import models
-from ..label.label import Label
-from ..annotation.image_classification import ImageClassificationAnnotation
-
-class LegacyImage(models.Model):
-    # Add any other fields you need to store frame-related information
-    image = models.ImageField(upload_to="legacy_images")  # Or some other field type, depending on how you're storing the frame
-    suffix = models.CharField(max_length=255)
-    # ImageClassificationAnnotation has a foreign key to this model (related name: image_classification_annotations)
-
-    def get_classification_annotations(self):
-        """
-        Get all image classification annotations for this image.
-        """
-        return ImageClassificationAnnotation.objects.filter(legacy_image=self)
-    
-    def get_classification_annotations_by_label(self, label:Label):
-        """
-        Get all image classification annotations for this image with the given label.
-        """
-        return ImageClassificationAnnotation.objects.filter(legacy_image=self, label=label)
-    
-    def get_classification_annotations_by_value(self, value:bool):
-        """
-        Get all image classification annotations for this image with the given value.
-        """
-        return ImageClassificationAnnotation.objects.filter(legacy_image=self, value=value)
-    
-    def get_classification_annotations_by_label_and_value(self, label:Label, value:bool):
-        """
-        Get all image classification annotations for this image with the given label and value.
-        """
-        return ImageClassificationAnnotation.objects.filter(legacy_image=self, label=label, value=value)
-    
+from django.db import models
+from ..label.label import Label
+from ..annotation.image_classification import ImageClassificationAnnotation
+
+class LegacyImage(models.Model):
+    # Add any other fields you need to store frame-related information
+    image = models.ImageField(upload_to="legacy_images")  # Or some other field type, depending on how you're storing the frame
+    suffix = models.CharField(max_length=255)
+    # ImageClassificationAnnotation has a foreign key to this model (related name: image_classification_annotations)
+
+    def get_classification_annotations(self):
+        """
+        Get all image classification annotations for this image.
+        """
+        return ImageClassificationAnnotation.objects.filter(legacy_image=self)
+    
+    def get_classification_annotations_by_label(self, label:Label):
+        """
+        Get all image classification annotations for this image with the given label.
+        """
+        return ImageClassificationAnnotation.objects.filter(legacy_image=self, label=label)
+    
+    def get_classification_annotations_by_value(self, value:bool):
+        """
+        Get all image classification annotations for this image with the given value.
+        """
+        return ImageClassificationAnnotation.objects.filter(legacy_image=self, value=value)
+    
+    def get_classification_annotations_by_label_and_value(self, label:Label, value:bool):
+        """
+        Get all image classification annotations for this image with the given label and value.
+        """
+        return ImageClassificationAnnotation.objects.filter(legacy_image=self, label=label, value=value)
+