reformat by pre-commit

szigyi · szigyi · commit a7c62647e17b · 2024-09-25T17:18:06.000+01:00
diff --git a/tmh_registry/registry/admin.py b/tmh_registry/registry/admin.py
@@ -39,10 +39,12 @@ class PatientAdmin(ExportMixin, admin.ModelAdmin):
 class PatientHospitalMappingAdmin(ExportMixin, admin.ModelAdmin):
     model = PatientHospitalMapping
 
+
 @admin.register(PreferredHospital)
 class PreferredHospitalAdmin(ExportMixin, admin.ModelAdmin):
     model = PreferredHospital
 
+
 @admin.register(Episode)
 class EpisodeAdmin(ExportMixin, admin.ModelAdmin):
     model = Episode
diff --git a/tmh_registry/registry/api/serializers.py b/tmh_registry/registry/api/serializers.py
@@ -246,6 +246,7 @@ def to_representation(self, instance):
 
         return data
 
+
 class PreferredHospitalReadSerializer(ModelSerializer):
     hospital = SerializerMethodField()
 
@@ -257,13 +258,16 @@ def get_hospital(self, obj):
         request = self.context.get("request")
         if request and request.user:
             try:
-                medical_personnel = MedicalPersonnel.objects.get(user=request.user)
+                medical_personnel = MedicalPersonnel.objects.get(
+                    user=request.user
+                )
                 if obj.medical_personnel == medical_personnel:
                     return {"id": obj.hospital.id}
             except MedicalPersonnel.DoesNotExist:
                 pass
         return None
 
+
 class PatientHospitalMappingWriteSerializer(ModelSerializer):
     patient_id = PrimaryKeyRelatedField(queryset=Patient.objects.all())
     hospital_id = PrimaryKeyRelatedField(queryset=Hospital.objects.all())
diff --git a/tmh_registry/registry/api/viewsets.py b/tmh_registry/registry/api/viewsets.py
@@ -19,6 +19,7 @@
 from rest_framework.response import Response
 from rest_framework.viewsets import GenericViewSet
 
+from ...users.models import MedicalPersonnel
 from ..models import (
     Discharge,
     Episode,
@@ -42,31 +43,34 @@
     PreferredHospitalReadSerializer,
     ReadPatientSerializer,
 )
-from ...users.models import MedicalPersonnel
 
 
 class HospitalViewSet(viewsets.ReadOnlyModelViewSet):
     queryset = Hospital.objects.all()
     serializer_class = HospitalSerializer
 
+
 class PreferredHospitalViewSet(viewsets.GenericViewSet):
     serializer_class = PreferredHospitalReadSerializer
     permission_classes = [IsAuthenticated]
 
-    @action(detail=False, methods=['get'])
+    @action(detail=False, methods=["get"])
     def retrieve_for_current_user(self, request, *args, **kwargs):
         user = request.user
         try:
             medical_personnel = MedicalPersonnel.objects.get(user=user)
         except MedicalPersonnel.DoesNotExist:
             return Response({}, status=200)
         try:
-            preferred_hospital = PreferredHospital.objects.get(medical_personnel=medical_personnel)
+            preferred_hospital = PreferredHospital.objects.get(
+                medical_personnel=medical_personnel
+            )
         except PreferredHospital.DoesNotExist:
             return Response({}, status=200)
         serializer = self.get_serializer(preferred_hospital)
         return Response(serializer.data)
 
+
 class PatientFilterSet(FilterSet):
     hospital_id = NumberFilter(
         method="filter_hospital",
diff --git a/tmh_registry/registry/migrations/0036_followup_recurrence.py b/tmh_registry/registry/migrations/0036_followup_recurrence.py
@@ -6,13 +6,13 @@
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('registry', '0035_episode_comments'),
+        ("registry", "0035_episode_comments"),
     ]
 
     operations = [
         migrations.AddField(
-            model_name='followup',
-            name='recurrence',
+            model_name="followup",
+            name="recurrence",
             field=models.BooleanField(null=True),
         ),
     ]
diff --git a/tmh_registry/registry/migrations/0037_preferredhospital.py b/tmh_registry/registry/migrations/0037_preferredhospital.py
@@ -7,21 +7,43 @@
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('users', '0008_auto_20210630_1407'),
-        ('registry', '0036_followup_recurrence'),
+        ("users", "0008_auto_20210630_1407"),
+        ("registry", "0036_followup_recurrence"),
     ]
 
     operations = [
         migrations.CreateModel(
-            name='PreferredHospital',
+            name="PreferredHospital",
             fields=[
-                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('hospital', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='preferred_by_medical_personnel', to='registry.hospital')),
-                ('medical_personnel', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='preferred_hospital', to='users.medicalpersonnel')),
+                (
+                    "id",
+                    models.AutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                (
+                    "hospital",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="preferred_by_medical_personnel",
+                        to="registry.hospital",
+                    ),
+                ),
+                (
+                    "medical_personnel",
+                    models.OneToOneField(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="preferred_hospital",
+                        to="users.medicalpersonnel",
+                    ),
+                ),
             ],
             options={
-                'verbose_name_plural': 'Medical Personnel Preferred Hospitals',
-                'unique_together': {('medical_personnel', 'hospital')},
+                "verbose_name_plural": "Medical Personnel Preferred Hospitals",
+                "unique_together": {("medical_personnel", "hospital")},
             },
         ),
     ]
diff --git a/tmh_registry/registry/migrations/0038_auto_20240915_1210.py b/tmh_registry/registry/migrations/0038_auto_20240915_1210.py
@@ -6,12 +6,12 @@
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('registry', '0037_preferredhospital'),
+        ("registry", "0037_preferredhospital"),
     ]
 
     operations = [
         migrations.AlterModelOptions(
-            name='patienthospitalmapping',
-            options={'verbose_name_plural': 'Patient-Hospital Mappings'},
+            name="patienthospitalmapping",
+            options={"verbose_name_plural": "Patient-Hospital Mappings"},
         ),
     ]
diff --git a/tmh_registry/registry/migrations/0039_episode_primary_surgeon.py b/tmh_registry/registry/migrations/0039_episode_primary_surgeon.py
@@ -7,24 +7,45 @@
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('users', '0008_auto_20210630_1407'),
-        ('registry', '0038_auto_20240915_1210'),
+        ("users", "0008_auto_20210630_1407"),
+        ("registry", "0038_auto_20240915_1210"),
     ]
 
     operations = [
         migrations.AddField(
-            model_name='episode',
-            name='primary_surgeon',
-            field=models.ForeignKey(blank=True, default=None, null=True, on_delete=django.db.models.deletion.SET_DEFAULT, related_name='primary_surgeon', to='users.medicalpersonnel'),
+            model_name="episode",
+            name="primary_surgeon",
+            field=models.ForeignKey(
+                blank=True,
+                default=None,
+                null=True,
+                on_delete=django.db.models.deletion.SET_DEFAULT,
+                related_name="primary_surgeon",
+                to="users.medicalpersonnel",
+            ),
         ),
         migrations.AddField(
-            model_name='episode',
-            name='secondary_surgeon',
-            field=models.ForeignKey(blank=True, default=None, null=True, on_delete=django.db.models.deletion.SET_DEFAULT, related_name='secondary_surgeon', to='users.medicalpersonnel'),
+            model_name="episode",
+            name="secondary_surgeon",
+            field=models.ForeignKey(
+                blank=True,
+                default=None,
+                null=True,
+                on_delete=django.db.models.deletion.SET_DEFAULT,
+                related_name="secondary_surgeon",
+                to="users.medicalpersonnel",
+            ),
         ),
         migrations.AddField(
-            model_name='episode',
-            name='tertiary_surgeon',
-            field=models.ForeignKey(blank=True, default=None, null=True, on_delete=django.db.models.deletion.SET_DEFAULT, related_name='tertiary_surgeon', to='users.medicalpersonnel'),
+            model_name="episode",
+            name="tertiary_surgeon",
+            field=models.ForeignKey(
+                blank=True,
+                default=None,
+                null=True,
+                on_delete=django.db.models.deletion.SET_DEFAULT,
+                related_name="tertiary_surgeon",
+                to="users.medicalpersonnel",
+            ),
         ),
     ]
diff --git a/tmh_registry/registry/migrations/0040_populate_primary_surgeon_20240916_1320.py b/tmh_registry/registry/migrations/0040_populate_primary_surgeon_20240916_1320.py
@@ -1,5 +1,6 @@
 from django.db import migrations
 
+
 def get_primary_surgeon_from_episode_surgeons(apps, schema_editor):
     Episode = apps.get_model("registry", "episode")
 
@@ -16,11 +17,13 @@ def get_primary_surgeon_from_episode_surgeons(apps, schema_editor):
             episode.tertiary_surgeon = surgeons[2]
         episode.save()
 
+
 class Migration(migrations.Migration):
 
     dependencies = [
         ("registry", "0039_episode_primary_surgeon"),
     ]
 
-    operations = [migrations.RunPython(get_primary_surgeon_from_episode_surgeons)]
-
+    operations = [
+        migrations.RunPython(get_primary_surgeon_from_episode_surgeons)
+    ]
diff --git a/tmh_registry/registry/models.py b/tmh_registry/registry/models.py
@@ -2,6 +2,7 @@
 
 from django.db.models import (
     CASCADE,
+    SET_DEFAULT,
     BooleanField,
     CharField,
     DateField,
@@ -12,7 +13,7 @@
     OneToOneField,
     PositiveIntegerField,
     TextChoices,
-    TextField, SET_DEFAULT,
+    TextField,
 )
 
 from tmh_registry.common.models import TimeStampMixin
@@ -78,23 +79,25 @@ class Meta:
     def __str__(self):
         return f"Patient {self.patient.full_name} ({self.patient_hospital_id}) - Hospital {self.hospital.name}"
 
+
 class PreferredHospital(Model):
     medical_personnel = OneToOneField(
         MedicalPersonnel, on_delete=CASCADE, related_name="preferred_hospital"
     )
     hospital = ForeignKey(
-        Hospital, on_delete=CASCADE, related_name="preferred_by_medical_personnel"
+        Hospital,
+        on_delete=CASCADE,
+        related_name="preferred_by_medical_personnel",
     )
 
     class Meta:
-        unique_together = (
-            ("medical_personnel", "hospital"),
-        )
+        unique_together = (("medical_personnel", "hospital"),)
         verbose_name_plural = "Medical Personnel Preferred Hospitals"
 
     def __str__(self):
         return f"{self.medical_personnel.user.first_name} {self.medical_personnel.user.last_name} ({self.medical_personnel.user.username}) - Hospital {self.hospital.name}"
 
+
 class Episode(Model):
     class EpisodeChoices(TextChoices):
         INGUINAL = ("INGUINAL", "Inguinal Mesh Hernia Repair")
@@ -157,9 +160,30 @@ class AnaestheticChoices(TextChoices):
     created = DateTimeField(auto_now_add=True)
     surgery_date = DateField(null=True, blank=True)
     episode_type = CharField(max_length=128, choices=EpisodeChoices.choices)
-    primary_surgeon = ForeignKey(MedicalPersonnel, on_delete=SET_DEFAULT, blank=True, null=True, default=None, related_name="primary_surgeon")
-    secondary_surgeon = ForeignKey(MedicalPersonnel, on_delete=SET_DEFAULT, blank=True, null=True, default=None, related_name="secondary_surgeon")
-    tertiary_surgeon = ForeignKey(MedicalPersonnel, on_delete=SET_DEFAULT, blank=True, null=True, default=None, related_name="tertiary_surgeon")
+    primary_surgeon = ForeignKey(
+        MedicalPersonnel,
+        on_delete=SET_DEFAULT,
+        blank=True,
+        null=True,
+        default=None,
+        related_name="primary_surgeon",
+    )
+    secondary_surgeon = ForeignKey(
+        MedicalPersonnel,
+        on_delete=SET_DEFAULT,
+        blank=True,
+        null=True,
+        default=None,
+        related_name="secondary_surgeon",
+    )
+    tertiary_surgeon = ForeignKey(
+        MedicalPersonnel,
+        on_delete=SET_DEFAULT,
+        blank=True,
+        null=True,
+        default=None,
+        related_name="tertiary_surgeon",
+    )
     surgeons = ManyToManyField(MedicalPersonnel)
     cepod = CharField(max_length=16, choices=CepodChoices.choices)
     side = CharField(max_length=16, choices=SideChoices.choices)
diff --git a/tmh_registry/registry/urls.py b/tmh_registry/registry/urls.py
@@ -18,7 +18,11 @@
 router.register(r"episodes", EpisodeViewset)
 router.register(r"discharges", DischargeViewset)
 router.register(r"follow-ups", FollowUpViewset)
-router.register(r"preferred-hospital", PreferredHospitalViewSet, basename='preferred-hospital')
+router.register(
+    r"preferred-hospital",
+    PreferredHospitalViewSet,
+    basename="preferred-hospital",
+)
 
 urlpatterns = [
     path("", include(router.urls)),

Original file line number	Diff line number	Diff line change
`@@ -6,12 +6,12 @@`
`6`	`6`	`class Migration(migrations.Migration):`
`7`	`7`
`8`	`8`	`dependencies = [`
`9`		`- ('registry', '0037_preferredhospital'),`
	`9`	`+ ("registry", "0037_preferredhospital"),`
`10`	`10`	`]`
`11`	`11`
`12`	`12`	`operations = [`
`13`	`13`	`migrations.AlterModelOptions(`
`14`		`- name='patienthospitalmapping',`
`15`		`- options={'verbose_name_plural': 'Patient-Hospital Mappings'},`
	`14`	`+ name="patienthospitalmapping",`
	`15`	`+ options={"verbose_name_plural": "Patient-Hospital Mappings"},`
`16`	`16`	`),`
`17`	`17`	`]`