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

endoreg_db/models/persons/patient/patient_medication.py

@@ -0,0 +1,44 @@
+from django.db import models
+from django.utils.translation import gettext_lazy as _
+
+class PatientMedication(models.Model):
+    patient = models.ForeignKey("Patient", on_delete= models.CASCADE)
+    medication_indication = models.ForeignKey(
+        "MedicationIndication", on_delete=models.CASCADE,
+        related_name="patient_medications", null=True
+    )
+
+    medication_schedules = models.ManyToManyField(
+        'MedicationSchedule'
+    )
+    unit = models.ForeignKey('Unit', on_delete=models.CASCADE)
+    dosage = models.JSONField()
+    active = models.BooleanField(default=True)
+
+    objects = models.Manager()
+
+    class Meta:
+        verbose_name = _('Patient Medication')
+        verbose_name_plural = _('Patient Medications')
+
+    @classmethod
+    def create_by_patient_and_indication(cls, patient, medication_indication):
+        from endoreg_db.models import MedicationIndication
+        medication_indication: MedicationIndication
+        patient_medication = cls.objects.create(patient=patient, medication_indication=medication_indication)
+        patient_medication.save()
+        patient_medication.set_schedules_from_indication()
+        
+        return patient_medication
+
+    def set_schedules_from_indication(self):
+        schedules = self.medication_indication.medication_schedules.all()
+        self.medication_schedules.set(schedules)
+        self.save()
+
+    def __str__(self):
+        indication = self.medication_indication
+        schedules = self.medication_schedules.all()
+        return f'{indication} - {schedules}'
+    
+    

endoreg_db/models/persons/patient/patient_medication_schedule.py

@@ -0,0 +1,28 @@
+from django.db import models
+
+class PatientMedicationSchedule(models.Model):
+    patient = models.ForeignKey("Patient", on_delete= models.CASCADE)
+    medication = models.ManyToManyField(
+        'PatientMedication', related_name='patient_medication_schedules'
+    )
+
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    def __str__(self):
+        return f'{self.patient} - {self.medication.all()}'
+    
+    @classmethod
+    def create_by_patient_and_indication_type(cls, patient, indication_type):
+        from endoreg_db.models import MedicationIndicationType, PatientMedication
+
+        medication_indication = MedicationIndicationType.get_random_indication_by_type(name=indication_type)
+
+        patient_medication_schedule = cls.objects.create(patient=patient)
+        patient_medication_schedule.save()
+
+        patient_medication = PatientMedication.create_by_patient_and_indication(patient, medication_indication)
+        patient_medication_schedule.medication.add(patient_medication)
+        patient_medication_schedule.save()
+
+        return patient_medication_schedule

endoreg_db/models/persons/person.py

@@ -17,10 +17,7 @@ class Person(models.Model):
     first_name = models.CharField(max_length=255)
     last_name = models.CharField(max_length=255)
     dob = models.DateField("Date of Birth", blank=True, null=True)
-    gender = models.CharField(
-        max_length=10, choices=[('Male', 'Male'), ('Female', 'Female'), ('Other', 'Other')],
-        blank=True, null=True
-    )
+    gender = models.ForeignKey("Gender", on_delete=models.SET_NULL, null=True)
     email = models.EmailField(max_length=255, blank=True, null=True)
     phone = models.CharField(max_length=255, blank=True, null=True)
     is_real_person = models.BooleanField(default=True)

endoreg_db/models/persons/portal_user_information.py

@@ -1,8 +1,6 @@
 from django.db import models
 
 # models.py in your main app
-
-from django.db import models
 from django.contrib.auth.models import User
 
 class ProfessionManager(models.Manager):

endoreg_db/models/product/__init__.py

@@ -0,0 +1,5 @@
+from .product import Product
+from .product_material import ProductMaterial
+from .product_group import ProductGroup
+from .reference_product import ReferenceProduct
+from .product_weight import ProductWeight

endoreg_db/models/product/product.py

@@ -0,0 +1,97 @@
+from django.db import models
+
+class ProductManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+    
+def sum_weights(product_materials):
+    # sum up the weights
+    weight = 0
+    reference_unit = None
+    for product_material in product_materials:
+        if not reference_unit:
+            reference_unit = product_material.unit
+        else:
+            assert reference_unit == product_material.unit, "ProductMaterial units do not match"
+        weight += product_material.quantity
+
+    return weight, reference_unit
+
+def sum_emissions(product_materials):
+    # sum up the emissions
+    emission = 0
+    reference_unit = None
+    for product_material in product_materials:
+        if not reference_unit:
+            reference_unit = product_material.unit
+        else:
+            assert reference_unit == product_material.unit, "ProductMaterial units do not match"
+        emission, emission_unit = product_material.get_emission()
+        assert reference_unit == emission_unit, "ProductMaterial units do not match"
+        emission += emission
+
+    return emission, reference_unit
+
+class Product(models.Model):
+    objects = ProductManager()
+
+    name = models.CharField(max_length=255)
+    name_de = models.CharField(max_length=255, null=True)
+    name_en = models.CharField(max_length=255, null=True)
+
+    transport_route = models.ForeignKey("TransportRoute", on_delete=models.SET_NULL, null=True)
+    product_group = models.ForeignKey("ProductGroup", on_delete=models.SET_NULL, null=True)
+
+    def natural_key(self):
+        return (self.name,)
+    
+    def __str__(self):
+        return self.name
+    
+    def get_product_weight(self):
+        # check if there is a product material weight
+        from .product_material import ProductMaterial
+        product_materials = ProductMaterial.objects.filter(product=self, component="product")
+        if product_materials:
+            return self.get_product_material_weight()
+        
+        # check if there is a product weight
+        #TODO
+
+    def get_package_weight(self):
+        # check if there is a package material weight
+        from .product_material import ProductMaterial
+        product_materials = ProductMaterial.objects.filter(product=self, component="package")
+        if product_materials:
+            return self.get_package_material_weight()
+        
+        # check if there is a package weight
+        #TODO
+
+    def get_product_material_weight(self):
+        # get all materials with component == "product"
+        from .product_material import ProductMaterial
+        product_materials = ProductMaterial.objects.filter(product=self, component="product")
+
+        return sum_weights(product_materials)
+    
+    def get_package_material_weight(self):
+        # get all materials with component == "package"
+        from .product_material import ProductMaterial
+        product_materials = ProductMaterial.objects.filter(product=self, component="package")
+
+        return sum_weights(product_materials)
+
+    def get_product_material_emission(self):
+        # get all materials with component == "product"
+        from .product_material import ProductMaterial
+        product_materials = ProductMaterial.objects.filter(product=self, component="product")
+
+        return sum_emissions(product_materials)
+
+    def get_package_material_emission(self):
+        # get all materials with component == "package"
+        from .product_material import ProductMaterial
+        product_materials = ProductMaterial.objects.filter(product=self, component="package")
+
+        return sum_emissions(product_materials)

endoreg_db/models/product/product_group.py

@@ -0,0 +1,19 @@
+from django.db import models
+
+class ProductGroupManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+    
+class ProductGroup(models.Model):
+    objects = ProductGroupManager()
+
+    name = models.CharField(max_length=255)
+    name_de = models.CharField(max_length=255, null=True)
+    name_en = models.CharField(max_length=255, null=True)
+    
+    def natural_key(self):
+        return (self.name,)
+    
+    def __str__(self):
+        return self.name
+    

endoreg_db/models/product/product_material.py

@@ -0,0 +1,24 @@
+from django.db import models
+
+class ProductMaterial(models.Model):
+    component = models.CharField(max_length=255)
+    material = models.ForeignKey("Material", on_delete=models.CASCADE)
+    product = models.ForeignKey("Product", on_delete=models.CASCADE, related_name="product_materials")
+    unit = models.ForeignKey("Unit", on_delete=models.CASCADE)
+    quantity = models.FloatField()
+
+    def get_emission(self):
+        from ..emission import EmissionFactor
+        emission_factor:EmissionFactor = self.material.emission_factor
+        if emission_factor is None:
+            raise Exception("No emission factor for material " + self.material.name + " found.")
+        
+        # make sure product_material.unit is the same as emission_factor.unit
+        if self.unit != emission_factor.unit:
+            raise Exception("Unit mismatch: " + self.unit.name + " != " + emission_factor.unit.name)
+        
+        emmision_value = emission_factor.value * self.quantity
+        emission_unit = emission_factor.unit
+        return emmision_value, emission_unit
+        
+

endoreg_db/models/product/product_weight.py

@@ -0,0 +1,26 @@
+from django.db import models
+import pandas as pd
+
+class ProductWeightManager(models.Manager):
+    def get_by_natural_key(self, product, product_group, weight):
+        return self.get(product=product, product_group=product_group, weight=weight)
+    
+class ProductWeight(models.Model):
+    objects = ProductWeightManager()
+
+    name = models.CharField(max_length=255, null = True)
+    product = models.ForeignKey("Product", on_delete=models.CASCADE)
+    measured = models.FloatField(null=True)
+    verified = models.FloatField(null=True)
+    manufacturer = models.FloatField(null=True)
+    unit = models.ForeignKey("Unit", on_delete=models.SET_NULL, null=True)
+
+    def get_weight(self):
+        if not pd.isnull(self.verified):
+            return self.verified
+        elif not pd.isnull(self.measured):
+            return self.measured
+        elif not pd.isnull(self.manufacturer):
+            return self.manufacturer
+        else:
+            return None

endoreg_db/models/product/reference_product.py

@@ -0,0 +1,99 @@
+from django.db import models
+from typing import List
+
+class ReferenceProductManager(models.Manager):
+    def get_by_natural_key(self, product_name:str, product_group_name:str):
+        return self.get(product__name=product_name, product_group__name=product_group_name)
+
+class ReferenceProduct(models.Model):
+    name = models.CharField(max_length=255)
+    product = models.ForeignKey("Product", on_delete=models.CASCADE)
+    product_group = models.OneToOneField("ProductGroup", on_delete=models.CASCADE, related_name="reference_product")
+    emission_factor_total = models.ForeignKey("EmissionFactor", on_delete=models.SET_NULL, null=True, blank = True)
+    emission_factor_package = models.ForeignKey("EmissionFactor", on_delete=models.SET_NULL, null=True, related_name="reference_product_package")
+    emission_factor_product = models.ForeignKey("EmissionFactor", on_delete=models.SET_NULL, null=True, related_name="reference_product_product")
+    
+    objects = ReferenceProductManager()
+
+    def __str__(self):
+        return self.product.name + " (" + self.product_group.name + ")"
+    
+    def set_emission_factors(self):
+        from .product import Product
+        from .product_material import ProductMaterial
+        from ..emission import EmissionFactor
+
+        product:Product = self.product
+        materials = product.product_materials.all()
+        emission_factor_name = f"{self.product_group.name}_{product.name}_total_emission_factor"
+        emission_factor_package_name = f"{self.product_group.name}_{product.name}_package_emission_factor"
+        emission_factor_product_name = f"{self.product_group.name}_{product.name}_product_emission_factor"
+
+        product_emissions = 0
+        package_emissions = 0
+
+        product_weight, product_weight_unit = product.get_product_material_weight()
+        package_weight, package_weight_unit = product.get_package_material_weight()
+        product_emission, product_emission_unit = product.get_product_material_emission()
+        package_emission, package_emission_unit = product.get_package_material_emission()
+
+        total_weight = product_weight + package_weight
+        total_emission = product_emission + package_emission
+
+        reference_unit = product_weight_unit
+        assert reference_unit == package_weight_unit, "Package weight units do not match"
+        assert reference_unit == product_emission_unit, "Product emission units do not match"
+        assert reference_unit == package_emission_unit, "Package emission units do not match"
+
+        product_emission_factor_value = product_emission / product_weight
+        package_emission_factor_value = package_emission / package_weight
+        total_emission_factor_value = total_emission / total_weight
+
+        emission_factor, created = EmissionFactor.objects.get_or_create(
+            name=emission_factor_name,
+            defaults={
+                "name": emission_factor_name,
+                "value": total_emission_factor_value,
+                "unit": reference_unit
+            }
+        )
+        self.emission_factor_total = emission_factor
+
+        emission_factor_package, created = EmissionFactor.objects.get_or_create(
+            name=emission_factor_package_name,
+            defaults={
+                "name": emission_factor_package_name,
+                "value": package_emission_factor_value,
+                "unit": reference_unit
+            }
+        )
+        self.emission_factor_package = emission_factor_package
+
+        emission_factor_product, created = EmissionFactor.objects.get_or_create(
+            name=emission_factor_product_name,
+            defaults={
+                "name": emission_factor_product_name,
+                "value": product_emission_factor_value,
+                "unit": reference_unit
+            }
+        )
+        self.emission_factor_product = emission_factor_product
+
+        self.save()
+
+    def get_emission_factor(self, component:str):
+        # check if emission_factor_total exists:
+        if self.emission_factor_total is None:
+            self.set_emission_factors()
+
+        if component == "total":
+            return self.emission_factor_total
+        elif component == "package":
+            return self.emission_factor_package
+        elif component == "product":
+            return self.emission_factor_product
+        else:
+            raise Exception("Unknown component: " + component)
+        
+
+

endoreg_db/models/quiz/__init__.py

@@ -0,0 +1,2 @@
+from .quiz_answer import QuizAnswer, QuizAnswerType
+from .quiz_question import QuizQuestion, QuizQuestionType

endoreg_db/models/quiz/quiz_answer.py

@@ -0,0 +1,41 @@
+from django.db import models
+
+class QuizAnswerTypeManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
+class QuizAnswerType(models.Model):
+    objects = QuizAnswerTypeManager()
+
+    name = models.CharField(max_length=255)
+    description = models.TextField(blank=True, null=True)
+
+    def natural_key(self):
+        return (self.name,)
+
+    def __str__(self):
+        return self.name
+
+class QuizAnswerManager(models.Manager):
+    def get_by_natural_key(self, text):
+        return self.get(text=text)
+    
+class QuizAnswer(models.Model):
+    objects = QuizAnswerManager()
+
+    text_value = models.CharField(max_length=255, null=True, blank=True)
+    number_value = models.FloatField(null=True, blank=True)
+    answer_type = models.ForeignKey("QuizAnswerType", on_delete=models.CASCADE)
+    question = models.ForeignKey("QuizQuestion", on_delete=models.CASCADE)
+
+    last_updated = models.DateTimeField(auto_now=True)
+
+    #TODO add user as foreign key
+
+    def natural_key(self):
+        return (self.text,)
+    
+    def __str__(self):
+        return self.text
+    
+

endoreg_db/models/quiz/quiz_question.py

@@ -0,0 +1,54 @@
+from django.db import models
+
+class QuizQuestionTypeManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+    
+class QuizQuestionType(models.Model):
+    """
+    A class representing a quiz question type.
+
+    Attributes:
+        name (str): The name of the quiz question type.
+        description (str): A description of the quiz question type.
+
+    """
+    name = models.CharField(max_length=255)
+    description = models.TextField(blank=True, null=True)
+
+    objects = QuizQuestionTypeManager()
+
+    def natural_key(self):
+        return (self.name,)
+
+    def __str__(self):
+        return self.name
+    
+class QuizQuestionManager(models.Manager):
+    def get_by_natural_key(self, text):
+        return self.get(text=text)
+    
+class QuizQuestion(models.Model):
+    """
+    A class representing a quiz question.
+
+    Attributes:
+        text (str): The text of the quiz question.
+        question_type (QuizQuestionType): The type of the quiz question.
+        description (str): A description of the quiz question.
+
+    """
+    text = models.TextField()
+    description = models.TextField(blank=True, null=True)
+    question_type = models.ForeignKey("QuizQuestionType", on_delete=models.CASCADE)
+
+    objects = QuizQuestionManager()
+
+    def natural_key(self):
+        return (self.text,)
+    
+    def __str__(self):
+        return self.text
+    
+    def get_question_type(self):
+        return self.question_type.name

endoreg_db/models/rules/__init__.py

@@ -0,0 +1,5 @@
+from .rule_type import RuleType
+from .rule import Rule
+from .ruleset import Ruleset
+from .rule_attribute_dtype import RuleAttributeDType
+from .rule_applicator import RuleApplicator

endoreg_db/models/rules/rule.py

@@ -0,0 +1,24 @@
+from django.db import models
+
+class RuleManager(models.Manager):
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
+class Rule(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)
+    description = models.TextField(blank=True, null=True)
+    attribute_key = models.CharField(max_length=255)
+    rule_type = models.ForeignKey("RuleType", on_delete=models.CASCADE)
+    attribute_dtype = models.ForeignKey("RuleAttributeDType", on_delete=models.CASCADE)
+
+    objects = RuleManager()
+
+    class Meta:
+        verbose_name = 'Rule'
+        verbose_name_plural = 'Rules'
+
+    def natural_key(self):
+        return (self.name,)
+