Merge pull request #111 from hpi-sam/108-m01-add-basic-triage

108 m01 add basic triage

backend/dps_training_k/game/consumers/abstract_consumer.py

@@ -34,6 +34,7 @@ class ClosureCodes:
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.exercise_code = ""
+        self.exercise = None
         self.REQUESTS_MAP = {}
         self.user = None
 
@@ -147,27 +148,27 @@ def authenticate(self, token):
             token = Token.objects.get(key=token)
             self.user = token.user
             self.user.set_channel_name(self.channel_name)
-            return True
+            return True, self.user.username
         except Token.DoesNotExist:
             self.close(code=self.ClosureCodes.NOT_AUTHENTICATED)
-            return False
+            return False, None
 
     def _send_exercise(self):
-        exercise = Exercise.createExercise()
         patient = Patient.objects.create(
-            name="Max Mustermann", exercise=exercise, patientId=123456
+            name="Max Mustermann", exercise=self.exercise, patientId=123456
         )
         exercise_object = {
             "exercise": {
-                "exerciseId": exercise.exerciseId,
+                "exerciseId": self.exercise.exerciseId,
                 "areas": [
                     {
                         "areaName": "X",
                         "patients": [
                             {
                                 "patientId": patient.patientId,
                                 "patientName": patient.name,
-                                "patientCode": 0
+                                "patientCode": 0,
+                                "triage": patient.triage
                             }
                         ],
                         "personnel": [

backend/dps_training_k/game/consumers/patient_consumer.py

@@ -1,11 +1,13 @@
 from .abstract_consumer import AbstractConsumer
 from urllib.parse import parse_qs
+from game.models import Patient
 
 
 class PatientConsumer(AbstractConsumer):
     class PatientIncomingMessageTypes:
         EXAMPLE = "example"
         TEST_PASSTHROUGH = "test-passthrough"
+        TRIAGE = "triage"
 
     class PatientOutgoingMessageTypes:
         RESPONSE = "response"
@@ -14,7 +16,8 @@ class PatientOutgoingMessageTypes:
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.patient_code = ""
+        self.patientId = ""
+        self.patient = None
         self.REQUESTS_MAP = {
             self.PatientIncomingMessageTypes.EXAMPLE: (
                 self.handle_example,
@@ -24,25 +27,37 @@ def __init__(self, *args, **kwargs):
             self.PatientIncomingMessageTypes.TEST_PASSTHROUGH: (
                 self.handle_test_passthrough,
             ),
+            self.PatientIncomingMessageTypes.TRIAGE: (
+                self.handle_triage,
+                "triage",
+            ),
         }
 
     def connect(self):
         query_string = parse_qs(self.scope["query_string"].decode())
         token = query_string.get("token", [None])[0]
-        if self.authenticate(token):
+        success, patientId = self.authenticate(token)
+        if success:
+            self.patient = Patient.objects.get(patientId=patientId)
+            self.patientId = patientId
+            self.exercise = self.patient.exercise
             self.accept()
             self._send_exercise()
 
     def handle_example(self, exercise_code, patient_code):
         self.exercise_code = exercise_code
-        self.patient_code = patient_code
+        self.patientId = patient_code
         self.send_event(
             self.PatientOutgoingMessageTypes.RESPONSE,
-            content=f"exercise_code {self.exercise_code} & patient_code {self.patient_code}",
+            content=f"exercise_code {self.exercise_code} & patient_code {self.patientId}",
         )
 
     def handle_test_passthrough(self):
         self.send_event(
             self.PatientOutgoingMessageTypes.TEST_PASSTHROUGH,
             message="received test event",
         )
+
+    def handle_triage(self, triage):
+        self.patient.triage = triage
+        self._send_exercise()

backend/dps_training_k/game/consumers/trainer_consumer.py

@@ -1,5 +1,5 @@
 from .abstract_consumer import AbstractConsumer
-
+from game.models import Exercise
 
 class TrainerConsumer(AbstractConsumer):
     class TrainerIncomingMessageTypes:
@@ -59,6 +59,7 @@ def handle_example(self, exercise_code):
         )
 
     def handle_create_exercise(self):
+        self.exercise = Exercise.createExercise()
         self._send_exercise()
 
     def handle_test_passthrough(self):

backend/dps_training_k/game/migrations/0006_patient_triage.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.0.1 on 2024-03-25 14:08
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('game', '0005_rename_invitation_code_exercise_exerciseid_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='patient',
+            name='triage',
+            field=models.CharField(choices=[('-', 'undefined'), ('R', 'red'), ('Y', 'yellow'), ('G', 'green'), ('A', 'airway'), ('B', 'breathing'), ('C', 'circulation'), ('D', 'disability'), ('E', 'exposure')], default='-'),
+        ),
+    ]

backend/dps_training_k/game/models/patient.py

@@ -4,6 +4,17 @@
 
 
 class Patient(Eventable, models.Model):
+    class Triage(models.TextChoices):
+        UNDEFINED = "-", "undefined"
+        RED = "R", "red"
+        YELLOW = "Y", "yellow"
+        GREEN = "G", "green"
+        Airway = "A", "airway"
+        BREATHING = "B", "breathing"
+        CIRCULATION = "C", "circulation"
+        DISABILITY = "D", "disability"
+        EXPOSURE = "E", "exposure"
+
     name = models.CharField(
         max_length=100, default="Max Mustermann"
     )  # technically patientData but kept here for simplicity for now
@@ -14,6 +25,10 @@ class Patient(Eventable, models.Model):
     patientId = models.IntegerField(
         help_text="patientId used to log into patient - therefore part of authentication"
     )
+    triage = models.CharField(
+        choices=Triage.choices,
+        default=Triage.UNDEFINED,
+    )
 
     def __str__(self):
         return f"Patient #{self.id} called {self.name} with ID {self.patientId}"

backend/dps_training_k/game/tests/factories/exercise_factory.py

@@ -16,8 +16,8 @@ class Meta:
 class ExerciseFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = Exercise
-        django_get_or_create = ("config", "invitation_code", "state")
+        django_get_or_create = ("config", "exerciseId", "state")
 
     config = factory.SubFactory(SavedExerciseFactory)
-    invitation_code = "a" * settings.INVITATION_LOGIC.code_length
+    exerciseId = "a" * settings.INVITATION_LOGIC.code_length
     state = Exercise.ExerciseStateTypes.CONFIGURATION

backend/dps_training_k/game/tests/factories/patient_factory.py

@@ -6,8 +6,8 @@
 class PatientFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = Patient
-        django_get_or_create = ("name", "exercise", "patientCode")
+        django_get_or_create = ("name", "exercise", "patientId")
 
     name = "Max Mustermann"
     exercise = factory.SubFactory(ExerciseFactory)
-    patientCode = 123456
+    patientId = 123456

backend/dps_training_k/game/tests/test_websocket_authentication.py

@@ -1,5 +1,6 @@
 from django.test import TransactionTestCase
 from game.models import User
+from .factories import PatientFactory
 from rest_framework.authtoken.models import Token
 from configuration.asgi import application
 from channels.testing import WebsocketCommunicator
@@ -9,8 +10,9 @@ class PatientWebSocketTest(TransactionTestCase):
     def setUp(self):
         super().setUp()
         # Create a user and token for testing
-        self.user = User.objects.create_user(username="testpatient123", password="test")
+        self.user = User.objects.create_user(username="123456", password="test")
         self.token, _ = Token.objects.get_or_create(user=self.user)
+        self.patient = PatientFactory()
 
     async def test_authenticated_websocket_connection(self):
         # Connect to the WebSocket