From c9641978d7ee598ecaf2dd1aa8a0f7273b7ebbc0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mika=C3=ABl=20Ates?= <mates@entrouvert.com>
Date: Fri, 22 Jan 2016 11:39:40 +0100
Subject: [PATCH] Add cbv to manage Room model.

---
 src/biomon/views.py | 59 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/src/biomon/views.py b/src/biomon/views.py
index b9c7964..07cd816 100644
--- a/src/biomon/views.py
+++ b/src/biomon/views.py
@@ -20,6 +20,8 @@
 '''
 
 
+import requests
+
 from datetime import datetime
 
 from django.views.generic import (TemplateView, FormView, View, ListView,
@@ -220,3 +222,60 @@ class AllAlertsCheckView(View):
     def post(self, request, *args, **kwargs):
         medibot_models.Episode.objects.all().update(checked=True, checker=request.user)
         return JsonResponse({'message': 'ok',})
+
+
+class RoomList(ListView):
+    model = models.Room
+
+    def get_context_data(self, **kwargs):
+        context = super(RoomList, self).get_context_data(**kwargs)
+        try:
+            context['lastpatient'] = models.Patient.objects.get(
+                    pk=int(self.request.COOKIES.get(LAST_PATIENT_COOKIE)))
+        except:
+            pass
+        return context
+
+
+class RoomCreate(CreateView):
+    model = models.Room
+    success_url = reverse_lazy('room_list')
+    fields = ['number']
+
+    def get_context_data(self, **kwargs):
+        context = super(RoomCreate, self).get_context_data(**kwargs)
+        try:
+            context['lastpatient'] = models.Patient.objects.get(
+                    pk=int(self.request.COOKIES.get(LAST_PATIENT_COOKIE)))
+        except:
+            pass
+        return context
+
+
+class RoomUpdate(UpdateView):
+    model = models.Room
+    success_url = reverse_lazy('room_list')
+    fields = ['number']
+
+    def get_context_data(self, **kwargs):
+        context = super(RoomUpdate, self).get_context_data(**kwargs)
+        try:
+            context['lastpatient'] = models.Patient.objects.get(
+                    pk=int(self.request.COOKIES.get(LAST_PATIENT_COOKIE)))
+        except:
+            pass
+        return context
+
+
+class RoomDelete(DeleteView):
+    model = models.Room
+    success_url = reverse_lazy('room_list')
+
+    def get_context_data(self, **kwargs):
+        context = super(RoomDelete, self).get_context_data(**kwargs)
+        try:
+            context['lastpatient'] = models.Patient.objects.get(
+                    pk=int(self.request.COOKIES.get(LAST_PATIENT_COOKIE)))
+        except:
+            pass
+        return context