From eb23c0e1a20fac8493260f1c4e6179e5144fdacf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mika=C3=ABl=20Ates?= <mates@entrouvert.com>
Date: Fri, 22 Jan 2016 12:26:39 +0100
Subject: [PATCH] Add endpoint to convert room number to patient id.

---
 src/biomon/urls.py  |  1 +
 src/biomon/views.py | 13 +++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/src/biomon/urls.py b/src/biomon/urls.py
index 497656c..001c59d 100644
--- a/src/biomon/urls.py
+++ b/src/biomon/urls.py
@@ -48,6 +48,7 @@ urlpatterns += (
     url(r'^room/add/$', login_required(views.RoomCreate.as_view()), name='room_add'),
     url(r'^room/(?P<pk>\d+)/$', login_required(views.RoomUpdate.as_view()), name='room_update'),
     url(r'^room/(?P<pk>\d+)/delete/$', login_required(views.RoomDelete.as_view()), name='room_delete'),
+    url(r'^room-number-to-pid/(?P<number>\w+)', views.RoomNumberToPid.as_view(), name='room_number_to_pid'),
 )
 
 urlpatterns += (
diff --git a/src/biomon/views.py b/src/biomon/views.py
index 07cd816..109e8cb 100644
--- a/src/biomon/views.py
+++ b/src/biomon/views.py
@@ -34,6 +34,7 @@ from django.contrib import messages
 from django.utils.translation import ugettext_lazy as _
 from django.conf import settings
 from django.http import JsonResponse
+from django.shortcuts import get_object_or_404
 
 from . import cbv
 from . import models
@@ -279,3 +280,15 @@ class RoomDelete(DeleteView):
         except:
             pass
         return context
+
+
+class RoomNumberToPid(View):
+    def get(self, request, *args, **kwargs):
+        room = get_object_or_404(models.Room,
+                number=kwargs.get('number', None))
+        infos = {'pid' : ''}
+        try:
+            infos = {'pid' : room.patient.id}
+        except models.Patient.DoesNotExist:
+            pass
+        return JsonResponse(infos)