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

endoreg_db/models/medical/patient/patient_finding.py

@@ -16,32 +16,62 @@ if TYPE_CHECKING:
 
 
 class PatientFinding(models.Model):
-    patient_examination = models.ForeignKey("PatientExamination", on_delete=models.CASCADE, related_name="patient_findings")
-    finding = models.ForeignKey("Finding", on_delete=models.CASCADE, related_name="finding_patient_findings")
+    patient_examination = models.ForeignKey(
+        "PatientExamination", on_delete=models.CASCADE, related_name="patient_findings"
+    )
+    finding = models.ForeignKey(
+        "Finding", on_delete=models.CASCADE, related_name="finding_patient_findings"
+    )
 
     # Audit-Felder für medizinische Nachverfolgung
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
-    created_by = models.ForeignKey("auth.User", on_delete=models.PROTECT, related_name="created_findings", null=True, blank=True)
-    updated_by = models.ForeignKey("auth.User", on_delete=models.PROTECT, related_name="updated_findings", null=True, blank=True)
+    created_by = models.ForeignKey(
+        "auth.User",
+        on_delete=models.PROTECT,
+        related_name="created_findings",
+        null=True,
+        blank=True,
+    )
+    updated_by = models.ForeignKey(
+        "auth.User",
+        on_delete=models.PROTECT,
+        related_name="updated_findings",
+        null=True,
+        blank=True,
+    )
 
     # Soft Delete für historische Daten
-    is_active = models.BooleanField(default=True, help_text="Deaktiviert statt gelöscht für Audit-Trail")
+    is_active = models.BooleanField(
+        default=True, help_text="Deaktiviert statt gelöscht für Audit-Trail"
+    )
     deactivated_at = models.DateTimeField(null=True, blank=True)
-    deactivated_by = models.ForeignKey("auth.User", on_delete=models.PROTECT, related_name="deactivated_findings", null=True, blank=True)
+    deactivated_by = models.ForeignKey(
+        "auth.User",
+        on_delete=models.PROTECT,
+        related_name="deactivated_findings",
+        null=True,
+        blank=True,
+    )
 
     if TYPE_CHECKING:
         patient_examination: models.ForeignKey["PatientExamination"]
         finding: models.ForeignKey["Finding"]
 
         @property
-        def video_segments(self) -> models.manager.RelatedManager["LabelVideoSegment"]: ...
+        def video_segments(
+            self,
+        ) -> models.manager.RelatedManager["LabelVideoSegment"]: ...
 
         @property
-        def interventions(self) -> models.manager.RelatedManager["PatientFindingIntervention"]: ...
+        def interventions(
+            self,
+        ) -> models.manager.RelatedManager["PatientFindingIntervention"]: ...
 
         @property
-        def classifications(self) -> models.manager.RelatedManager["PatientFindingClassification"]: ...
+        def classifications(
+            self,
+        ) -> models.manager.RelatedManager["PatientFindingClassification"]: ...
 
     class Meta:
         verbose_name = "Patient Finding"
@@ -51,13 +81,19 @@ class PatientFinding(models.Model):
         # Wichtige Constraints für Datenintegrität
         constraints = [
             models.UniqueConstraint(
-                fields=["patient_examination", "finding"], condition=models.Q(is_active=True), name="unique_active_finding_per_examination"
+                fields=["patient_examination", "finding"],
+                condition=models.Q(is_active=True),
+                name="unique_active_finding_per_examination",
             ),
             models.CheckConstraint(
                 condition=models.Q(  # called .check in future?
                     deactivated_at__isnull=True, deactivated_by__isnull=True
                 )
-                | models.Q(deactivated_at__isnull=False, deactivated_by__isnull=False, is_active=False),
+                | models.Q(
+                    deactivated_at__isnull=False,
+                    deactivated_by__isnull=False,
+                    is_active=False,
+                ),
                 name="deactivation_fields_consistency",
             ),
         ]
@@ -89,10 +125,14 @@ class PatientFinding(models.Model):
 
         # Prüfe ob Finding für diese Examination erlaubt ist
         if self.finding and self.patient_examination:
-            available_findings = self.patient_examination.examination_safe.get_available_findings()
+            available_findings = (
+                self.patient_examination.examination_safe.get_available_findings()
+            )
             if self.finding not in available_findings:
                 raise ValidationError(
-                    {"finding": f'Finding "{self.finding.name}" ist nicht für Examination "{self.patient_examination.examination_safe.name}" erlaubt.'}
+                    {
+                        "finding": f'Finding "{self.finding.name}" ist nicht für Examination "{self.patient_examination.examination_safe.name}" erlaubt.'
+                    }
                 )
 
         # Prüfe Required Findings Logic
@@ -118,10 +158,15 @@ class PatientFinding(models.Model):
         required_findings = getattr(examination, "required_findings", None)
         if required_findings and required_findings.exists():
             # Prüfe ob alle Required Findings vorhanden sind
-            existing_findings = self.patient_examination.patient_findings.filter(is_active=True).values_list("finding", flat=True)
+            existing_findings = self.patient_examination.patient_findings.filter(
+                is_active=True
+            ).values_list("finding", flat=True)
 
             missing_required = required_findings.exclude(id__in=existing_findings)
-            if missing_required.exists() and self.finding not in required_findings.all():
+            if (
+                missing_required.exists()
+                and self.finding not in required_findings.all()
+            ):
                 missing_names = ", ".join([f.name for f in missing_required])
                 raise ValidationError(f"Erforderliche Findings fehlen: {missing_names}")
 
@@ -151,7 +196,14 @@ class PatientFinding(models.Model):
                 self.deactivated_at = None
                 self.deactivated_by = None
                 self.updated_by = user
-                self.save(update_fields=["is_active", "deactivated_at", "deactivated_by", "updated_by"])
+                self.save(
+                    update_fields=[
+                        "is_active",
+                        "deactivated_at",
+                        "deactivated_by",
+                        "updated_by",
+                    ]
+                )
             except ValidationError as e:
                 raise ValidationError(f"Reaktivierung nicht möglich: {e}")
 
@@ -189,20 +241,30 @@ class PatientFinding(models.Model):
 
         try:
             classification = FindingClassification.objects.get(id=classification_id)
-            classification_choice = FindingClassificationChoice.objects.filter(id=classification_choice_id).first()
+            classification_choice = FindingClassificationChoice.objects.filter(
+                id=classification_choice_id
+            ).first()
 
             if not classification.choices.filter(id=classification_choice_id).exists():
-                raise ValidationError(f"Classification Choice {classification_choice_id} gehört nicht zu Classification {classification_id}")
+                raise ValidationError(
+                    f"Classification Choice {classification_choice_id} gehört nicht zu Classification {classification_id}"
+                )
 
-            existing = self.classifications.filter(classification=classification, classification_choice=classification_choice, is_active=True).first()
+            existing = self.classifications.filter(
+                classification=classification,
+                classification_choice=classification_choice,
+                is_active=True,
+            ).first()
 
             if existing:
                 return existing
 
-            patient_finding_classification = PatientFindingClassification.objects.create(
-                finding=self,
-                classification_id=classification_id,
-                classification_choice_id=classification_choice_id,
+            patient_finding_classification = (
+                PatientFindingClassification.objects.create(
+                    finding=self,
+                    classification_id=classification_id,
+                    classification_choice_id=classification_choice_id,
+                )
             )
 
             return patient_finding_classification
@@ -233,7 +295,11 @@ class PatientFinding(models.Model):
             intervention = FindingIntervention.objects.get(id=intervention_id)
 
             patient_finding_intervention = PatientFindingIntervention.objects.create(
-                patient_finding=self, intervention=intervention, state=state, date=date or timezone.now(), created_by=user
+                patient_finding=self,
+                intervention=intervention,
+                state=state,
+                date=date or timezone.now(),
+                created_by=user,
             )
 
             return patient_finding_intervention
@@ -273,7 +339,9 @@ class PatientFinding(models.Model):
         Returns:
                 QuerySet: Classifications related to this finding filtered by the "location" classification type.
         """
-        classifications = self.classifications.filter(classification__classification_types__name__iexact="location")
+        classifications = self.classifications.filter(
+            classification__classification_types__name__iexact="location"
+        )
         return classifications
 
     @property
@@ -284,7 +352,9 @@ class PatientFinding(models.Model):
         Returns:
                 QuerySet: Classifications related to this finding filtered by the "morphology" classification type.
         """
-        classifications = self.classifications.filter(classification__classification_types__name__iexact="morphology")
+        classifications = self.classifications.filter(
+            classification__classification_types__name__iexact="morphology"
+        )
         return classifications
 
     @property
@@ -318,7 +388,9 @@ class PatientFinding(models.Model):
             if pf_classification.classification:
                 finding_classifications_list.append(pf_classification.classification)
             if pf_classification.classification_choice:
-                finding_classification_choices_list.append(pf_classification.classification_choice)
+                finding_classification_choices_list.append(
+                    pf_classification.classification_choice
+                )
 
         # Get all active finding interventions
         finding_interventions_list = []
@@ -327,8 +399,12 @@ class PatientFinding(models.Model):
                 finding_interventions_list.append(pf_intervention.intervention)
 
         # Include patient examination and patient for context
-        patient_examinations_list = [self.patient_examination] if self.patient_examination else []
-        patient_findings_list = cast("List[PatientFinding]", [self])  # Include self for direct patient finding evaluations
+        patient_examinations_list = (
+            [self.patient_examination] if self.patient_examination else []
+        )
+        patient_findings_list = cast(
+            "List[PatientFinding]", [self]
+        )  # Include self for direct patient finding evaluations
 
         return RequirementLinks(
             findings=findings_list,

endoreg_db/models/medical/patient/patient_finding_classification.py

@@ -18,11 +18,23 @@ class PatientFindingClassification(models.Model):
     Links a PatientFinding to a specific classification and choice, with optional subcategory values.
     """
 
-    finding = models.ForeignKey("PatientFinding", on_delete=models.CASCADE, related_name="classifications")
-    classification = models.ForeignKey("FindingClassification", on_delete=models.CASCADE, related_name="patient_finding_classifications")
-    classification_choice = models.ForeignKey("FindingClassificationChoice", on_delete=models.CASCADE, related_name="patient_finding_classifications")
+    finding = models.ForeignKey(
+        "PatientFinding", on_delete=models.CASCADE, related_name="classifications"
+    )
+    classification = models.ForeignKey(
+        "FindingClassification",
+        on_delete=models.CASCADE,
+        related_name="patient_finding_classifications",
+    )
+    classification_choice = models.ForeignKey(
+        "FindingClassificationChoice",
+        on_delete=models.CASCADE,
+        related_name="patient_finding_classifications",
+    )
 
-    is_active = models.BooleanField(default=True, help_text="Indicates if the classification is currently active.")
+    is_active = models.BooleanField(
+        default=True, help_text="Indicates if the classification is currently active."
+    )
     subcategories = models.JSONField(blank=True, null=True)
     numerical_descriptors = models.JSONField(blank=True, null=True)
 
@@ -55,7 +67,9 @@ class PatientFindingClassification(models.Model):
             self.subcategories = self.classification_choice.subcategories
 
         if not self.numerical_descriptors:
-            self.numerical_descriptors = self.classification_choice.numerical_descriptors
+            self.numerical_descriptors = (
+                self.classification_choice.numerical_descriptors
+            )
 
         super().save(*args, **kwargs)
 
@@ -80,7 +94,9 @@ class PatientFindingClassification(models.Model):
             self.save()
         return self.numerical_descriptors
 
-    def set_subcategory(self, subcategory_name: str, subcategory_value: Dict[str, dict]):
+    def set_subcategory(
+        self, subcategory_name: str, subcategory_value: Dict[str, dict]
+    ):
         """
         Update the value of a specified subcategory and save the classification.
 
@@ -92,7 +108,9 @@ class PatientFindingClassification(models.Model):
             dict: The updated subcategory dictionary.
         """
         assert self.subcategories, "Subcategories must be initialized."
-        assert subcategory_name in self.subcategories, "Subcategory must be in subcategories."
+        assert subcategory_name in self.subcategories, (
+            "Subcategory must be in subcategories."
+        )
         self.subcategories[subcategory_name]["value"] = subcategory_value
         self.save()
 
@@ -136,8 +154,12 @@ class PatientFindingClassification(models.Model):
         Raises:
             ValueError: If the descriptor's distribution type is not supported.
         """
-        assert self.numerical_descriptors is not None, "Numerical descriptors must be initialized."
-        assert descriptor_name in self.numerical_descriptors, "Descriptor must be in numerical descriptors."
+        assert self.numerical_descriptors is not None, (
+            "Numerical descriptors must be initialized."
+        )
+        assert descriptor_name in self.numerical_descriptors, (
+            "Descriptor must be in numerical descriptors."
+        )
         descriptor = self.numerical_descriptors[descriptor_name]
         min_val = descriptor.get("min", 0)
         max_val = descriptor.get("max", 1)
@@ -173,7 +195,9 @@ class PatientFindingClassification(models.Model):
             raise ValueError("Descriptor name must be in numerical descriptors.")
 
         value = self.get_random_value_for_numerical_descriptor(descriptor_name)
-        assert self.numerical_descriptors is not None, "Numerical descriptors must be initialized."
+        assert self.numerical_descriptors is not None, (
+            "Numerical descriptors must be initialized."
+        )
         self.numerical_descriptors[descriptor_name]["value"] = value
         if save:
             self.save()
@@ -190,11 +214,16 @@ class PatientFindingClassification(models.Model):
         if not self.subcategories or not self.numerical_descriptors:
             self.save()
 
-        assert self.numerical_descriptors is not None, "Numerical descriptors must be initialized."
+        assert self.numerical_descriptors is not None, (
+            "Numerical descriptors must be initialized."
+        )
 
         numerical_descriptors = self.numerical_descriptors
 
-        for numerical_descriptor_name, _numerical_descriptor_dict in numerical_descriptors.items():
+        for (
+            numerical_descriptor_name,
+            _numerical_descriptor_dict,
+        ) in numerical_descriptors.items():
             self.set_random_numerical_descriptor(numerical_descriptor_name, save=False)
 
         self.save()

endoreg_db/models/medical/patient/patient_finding_intervention.py

@@ -8,9 +8,17 @@ if TYPE_CHECKING:
 
 
 class PatientFindingIntervention(models.Model):
-    finding = models.ForeignKey("PatientFinding", on_delete=models.CASCADE, related_name="interventions")
-    intervention = models.ForeignKey("FindingIntervention", on_delete=models.CASCADE, related_name="patient_finding_interventions")
-    is_active = models.BooleanField(default=True, help_text="Indicates if the intervention is currently active.")
+    finding = models.ForeignKey(
+        "PatientFinding", on_delete=models.CASCADE, related_name="interventions"
+    )
+    intervention = models.ForeignKey(
+        "FindingIntervention",
+        on_delete=models.CASCADE,
+        related_name="patient_finding_interventions",
+    )
+    is_active = models.BooleanField(
+        default=True, help_text="Indicates if the intervention is currently active."
+    )
     state = models.CharField(max_length=100, blank=True, null=True)
     time_start = models.DateTimeField(blank=True, null=True)
     time_end = models.DateTimeField(blank=True, null=True)

endoreg_db/models/medical/patient/patient_lab_sample.py

@@ -60,7 +60,9 @@ class PatientLabSample(models.Model):
         values (PatientLabValue; One2Many): The value of the lab sample.
     """
 
-    patient = models.ForeignKey("Patient", on_delete=models.CASCADE, related_name="lab_samples")
+    patient = models.ForeignKey(
+        "Patient", on_delete=models.CASCADE, related_name="lab_samples"
+    )
     sample_type = models.ForeignKey("PatientLabSampleType", on_delete=models.CASCADE)
     date = models.DateTimeField()
 
@@ -125,7 +127,9 @@ class PatientLabSample(models.Model):
         if not date:
             date = dt.now(timezone.utc)
 
-        patient_lab_sample = cls.objects.create(patient=patient, sample_type=sample_type, date=date)
+        patient_lab_sample = cls.objects.create(
+            patient=patient, sample_type=sample_type, date=date
+        )
 
         if save:
             patient_lab_sample.save()

endoreg_db/models/medical/patient/patient_lab_value.py

@@ -23,11 +23,23 @@ class PatientLabValue(models.Model):
         date (datetime): The date of the lab value.
     """
 
-    patient = models.ForeignKey("Patient", on_delete=models.CASCADE, related_name="lab_values", blank=True, null=True)
+    patient = models.ForeignKey(
+        "Patient",
+        on_delete=models.CASCADE,
+        related_name="lab_values",
+        blank=True,
+        null=True,
+    )
     lab_value = models.ForeignKey("LabValue", on_delete=models.CASCADE)
     value = models.FloatField(blank=True, null=True)
     value_str = models.CharField(max_length=255, blank=True, null=True)
-    sample = models.ForeignKey("PatientLabSample", on_delete=models.CASCADE, blank=True, null=True, related_name="values")
+    sample = models.ForeignKey(
+        "PatientLabSample",
+        on_delete=models.CASCADE,
+        blank=True,
+        null=True,
+        related_name="values",
+    )
     datetime = models.DateTimeField(  # if not set, use now
         auto_now_add=True
     )
@@ -57,7 +69,12 @@ class PatientLabValue(models.Model):
 
     @classmethod
     def create_lab_value_by_sample(
-        cls, sample: "PatientLabSample", lab_value_name: str, value: Optional[float] = None, value_str: Optional[str] = None, unit: Optional["Unit"] = None
+        cls,
+        sample: "PatientLabSample",
+        lab_value_name: str,
+        value: Optional[float] = None,
+        value_str: Optional[str] = None,
+        unit: Optional["Unit"] = None,
     ):
         from ..laboratory import LabValue
 
@@ -80,7 +97,9 @@ class PatientLabValue(models.Model):
     def __str__(self):
         formatted_datetime = self.datetime.strftime("%Y-%m-%d %H:%M")
         # normal_range = self.get_normal_range()
-        norm_range_string = f"[{self.normal_range.get('min', '')} - {self.normal_range.get('max', '')}]"
+        norm_range_string = (
+            f"[{self.normal_range.get('min', '')} - {self.normal_range.get('max', '')}]"
+        )
         _str = f"{self.lab_value} - {self.value} {self.unit} - {norm_range_string} ({formatted_datetime})"
         return _str
 
@@ -157,16 +176,29 @@ class PatientLabValue(models.Model):
             distribution = lab_value.get_default_default_distribution()
 
             if not distribution:
-                warnings.warn(f"No distribution set for lab value {lab_value}, assuming uniform numeric distribution based on normal values")
+                warnings.warn(
+                    f"No distribution set for lab value {lab_value}, assuming uniform numeric distribution based on normal values"
+                )
 
-                if not self.normal_range.get("min", None) or not self.normal_range.get("max", None):
+                if not self.normal_range.get("min", None) or not self.normal_range.get(
+                    "max", None
+                ):
                     self.set_norm_values_from_default()
                 _min = self.normal_range.get("min", 0.0001)
                 _max = self.normal_range.get("max", 100)
-                _name = "auto-" + self.lab_value_safe.name + "-distribution-default-uniform"
-                distribution = NumericValueDistribution(name=_name, min_descriptor=_min, max_max_desciptor=_max, distribution_type="uniform")
-
-                value = distribution.generate_value(lab_value=lab_value, patient=patient)
+                _name = (
+                    "auto-" + self.lab_value_safe.name + "-distribution-default-uniform"
+                )
+                distribution = NumericValueDistribution(
+                    name=_name,
+                    min_descriptor=_min,
+                    max_max_desciptor=_max,
+                    distribution_type="uniform",
+                )
+
+                value = distribution.generate_value(
+                    lab_value=lab_value, patient=patient
+                )
                 self.value = value
                 if save:
                     self.save()

endoreg_db/models/medical/patient/patient_medication.py

@@ -1,10 +1,12 @@
-from typing import TYPE_CHECKING, List, Optional, cast  # Added List
+from typing import TYPE_CHECKING, List, cast  # Added List
 
 from django.db import models
 
 # Added imports for type hints
 if TYPE_CHECKING:
-    from ....utils.links.requirement_link import RequirementLinks  # Added RequirementLinks
+    from ....utils.links.requirement_link import (
+        RequirementLinks,
+    )  # Added RequirementLinks
     from ...administration.person.patient import Patient
     from ...other.unit import Unit
     from ..medication import Medication, MedicationIndication, MedicationIntakeTime
@@ -18,9 +20,19 @@ class PatientMedication(models.Model):
     """
 
     patient = models.ForeignKey("Patient", on_delete=models.CASCADE)
-    medication_indication = models.ForeignKey("MedicationIndication", on_delete=models.CASCADE, related_name="indication_patient_medications", null=True)
+    medication_indication = models.ForeignKey(
+        "MedicationIndication",
+        on_delete=models.CASCADE,
+        related_name="indication_patient_medications",
+        null=True,
+    )
 
-    medication = models.ForeignKey("Medication", on_delete=models.CASCADE, blank=True, related_name="medication_patient_medications")
+    medication = models.ForeignKey(
+        "Medication",
+        on_delete=models.CASCADE,
+        blank=True,
+        related_name="medication_patient_medications",
+    )
 
     intake_times = models.ManyToManyField(
         "MedicationIntakeTime",
@@ -39,7 +51,9 @@ class PatientMedication(models.Model):
         medication_indication: models.ForeignKey["MedicationIndication|None"]
         medication: models.ForeignKey["Medication|None"]
 
-        intake_times = cast(models.manager.RelatedManager["MedicationIntakeTime"], intake_times)
+        intake_times = cast(
+            models.manager.RelatedManager["MedicationIntakeTime"], intake_times
+        )
         unit: models.ForeignKey["Unit|None"]
         dosage = cast(models.JSONField, dosage)
 
@@ -73,10 +87,14 @@ class PatientMedication(models.Model):
         verbose_name_plural = "Patient Medications"
 
     @classmethod
-    def create_by_patient_and_indication(cls, patient, medication_indication: "MedicationIndication"):
+    def create_by_patient_and_indication(
+        cls, patient, medication_indication: "MedicationIndication"
+    ):
         """Creates a PatientMedication instance linking a patient and an indication."""
 
-        patient_medication = cls.objects.create(patient=patient, medication_indication=medication_indication)
+        patient_medication = cls.objects.create(
+            patient=patient, medication_indication=medication_indication
+        )
         patient_medication.save()
 
         return patient_medication

endoreg_db/models/medical/patient/patient_medication_schedule.py

@@ -7,7 +7,11 @@ if TYPE_CHECKING:
     from .patient_medication import PatientMedication
     from ..medication import MedicationSchedule
     from ....utils.links.requirement_link import RequirementLinks  # Added
-    from ..medication import Medication, MedicationIndication, MedicationIntakeTime  # Added
+    from ..medication import (
+        Medication,
+        MedicationIndication,
+        MedicationIntakeTime,
+    )  # Added
 
 
 class PatientMedicationSchedule(models.Model):
@@ -16,7 +20,9 @@ class PatientMedicationSchedule(models.Model):
     """
 
     patient = models.ForeignKey("Patient", on_delete=models.CASCADE)
-    medication = models.ManyToManyField("PatientMedication", related_name="patient_medication_schedules", blank=True)
+    medication = models.ManyToManyField(
+        "PatientMedication", related_name="patient_medication_schedules", blank=True
+    )
 
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
@@ -24,7 +30,9 @@ class PatientMedicationSchedule(models.Model):
     if TYPE_CHECKING:
         patient: models.ForeignKey["Patient"]
 
-        medication = cast(models.manager.RelatedManager["PatientMedication"], medication)
+        medication = cast(
+            models.manager.RelatedManager["PatientMedication"], medication
+        )
 
     @property
     def links(self) -> "RequirementLinks":
@@ -37,14 +45,20 @@ class PatientMedicationSchedule(models.Model):
         aggregated_medication_indications: List["MedicationIndication"] = []
         aggregated_medication_intake_times: List["MedicationIntakeTime"] = []
 
-        patient_meds_in_schedule: List["PatientMedication"] = list(self.medication.all())
+        patient_meds_in_schedule: List["PatientMedication"] = list(
+            self.medication.all()
+        )
 
         for pm_instance in patient_meds_in_schedule:
             pm_links_obj = pm_instance.links
 
             aggregated_medications.extend(pm_links_obj.medications)
-            aggregated_medication_indications.extend(pm_links_obj.medication_indications)
-            aggregated_medication_intake_times.extend(pm_links_obj.medication_intake_times)
+            aggregated_medication_indications.extend(
+                pm_links_obj.medication_indications
+            )
+            aggregated_medication_intake_times.extend(
+                pm_links_obj.medication_intake_times
+            )
 
         return RequirementLinks(
             medications=list(set(aggregated_medications)),
@@ -64,12 +78,16 @@ class PatientMedicationSchedule(models.Model):
         from ..medication import MedicationIndicationType
         from .patient_medication import PatientMedication
 
-        medication_indication = MedicationIndicationType.get_random_indication_by_type(name=indication_type)
+        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 = PatientMedication.create_by_patient_and_indication(
+            patient, medication_indication
+        )
         patient_medication_schedule.medication.add(patient_medication)
         patient_medication_schedule.save()
 
@@ -87,7 +105,9 @@ class PatientMedicationSchedule(models.Model):
         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 = PatientMedication.create_by_patient_and_indication(
+            patient, medication_indication
+        )
         patient_medication_schedule.medication.add(patient_medication)
         patient_medication_schedule.save()
 
@@ -112,7 +132,11 @@ class PatientMedicationSchedule(models.Model):
         intake_times = medication_schedule.get_intake_times()
 
         patient_medication = PatientMedication.objects.create(
-            patient=self.patient, medication=drug, medication_indication=medication_indication, unit=unit, dosage=dosage
+            patient=self.patient,
+            medication=drug,
+            medication_indication=medication_indication,
+            unit=unit,
+            dosage=dosage,
         )
 
         patient_medication.intake_times.set(intake_times)