Improve launderette plannings and admin part

launderette/views.py

@@ -1,5 +1,6 @@
 from datetime import datetime, timedelta
 from collections import OrderedDict
+import pytz
 
 from django.shortcuts import render
 from django.views.generic import ListView, DetailView, RedirectView, TemplateView
@@ -7,9 +8,10 @@ from django.views.generic.edit import UpdateView, CreateView, DeleteView, Proces
 from django.forms.models import modelform_factory
 from django.forms import CheckboxSelectMultiple
 from django.utils.translation import ugettext as _
-from django.utils.timezone import make_aware
 from django.utils import dateparse
+from django.core.urlresolvers import reverse_lazy
 from django.conf import settings
+from django.db import transaction
 
 from core.models import Page
 from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, CanCreateMixin
@@ -33,27 +35,50 @@ class LaunderetteBookMainView(CanViewMixin, ListView):
     model = Launderette
     template_name = 'launderette/launderette_book_choose.jinja'
 
-class LaunderetteBookView(DetailView):
+class LaunderetteBookView(CanViewMixin, DetailView):
     """Display the launderette schedule"""
     model = Launderette
     pk_url_kwarg = "launderette_id"
     template_name = 'launderette/launderette_book.jinja'
 
-    def post(self, request, *args, **kwargs):
-        self.object = self.get_object()
-        if 'slot' in request.POST.keys() and request.user.is_authenticated():
-            subscriber = get_subscriber(request.user)
-            if subscriber.is_subscribed():
-                date = dateparse.parse_datetime(request.POST['slot'])
-                for m in self.object.machines.filter(is_working=True).all():
-                    slot = Slot.objects.filter(start_date=date, machine=m).first()
-                    print(slot)
-                    if slot is None:
-                        Slot(user=subscriber, start_date=date, machine=m, type="WASHING").save()
-                        print("Saved")
-                        break
+    def get(self, request, *args, **kwargs):
+        self.slot_type = "BOTH"
+        self.machines = {}
         return super(LaunderetteBookView, self).get(request, *args, **kwargs)
 
+    def post(self, request, *args, **kwargs):
+        self.slot_type = "BOTH"
+        self.machines = {}
+        with transaction.atomic():
+            self.object = self.get_object()
+            if 'slot_type' in request.POST.keys():
+                self.slot_type = request.POST['slot_type']
+            if 'slot' in request.POST.keys() and request.user.is_authenticated():
+                self.subscriber = get_subscriber(request.user)
+                if self.subscriber.is_subscribed():
+                    self.date = dateparse.parse_datetime(request.POST['slot']).replace(tzinfo=pytz.UTC)
+                    if self.slot_type == "WASHING":
+                        if self.check_slot(self.slot_type):
+                            Slot(user=self.subscriber, start_date=self.date, machine=self.machines[self.slot_type], type=self.slot_type).save()
+                    elif self.slot_type == "DRYING":
+                        if self.check_slot(self.slot_type):
+                            Slot(user=self.subscriber, start_date=self.date, machine=self.machines[self.slot_type], type=self.slot_type).save()
+                    else:
+                        if self.check_slot("WASHING") and self.check_slot("DRYING", self.date + timedelta(hours=1)):
+                            Slot(user=self.subscriber, start_date=self.date, machine=self.machines["WASHING"], type="WASHING").save()
+                            Slot(user=self.subscriber, start_date=self.date + timedelta(hours=1),
+                                    machine=self.machines["DRYING"], type="DRYING").save()
+        return super(LaunderetteBookView, self).get(request, *args, **kwargs)
+
+    def check_slot(self, type, date=None):
+        if date is None: date = self.date
+        for m in self.object.machines.filter(is_working=True, type=type).all():
+            slot = Slot.objects.filter(start_date=date, machine=m).first()
+            if slot is None:
+                self.machines[type] = m
+                return True
+        return False
+
     @staticmethod
     def date_iterator(startDate, endDate, delta=timedelta(days=1)):
         currentDate = startDate
@@ -65,16 +90,20 @@ class LaunderetteBookView(DetailView):
         """ Add page to the context """
         kwargs = super(LaunderetteBookView, self).get_context_data(**kwargs)
         kwargs['planning'] = OrderedDict()
-        start_date = make_aware(datetime.now().replace(hour=0, minute=0, second=0, microsecond=0))
+        kwargs['slot_type'] = self.slot_type
+        start_date = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0, tzinfo=pytz.UTC)
         for date in LaunderetteBookView.date_iterator(start_date, start_date+timedelta(days=6), timedelta(days=1)):
             kwargs['planning'][date] = []
             for h in LaunderetteBookView.date_iterator(date, date+timedelta(days=1), timedelta(hours=1)):
                 free = False
-                for m in self.object.machines.filter(is_working=True).all():
-                    s = Slot.objects.filter(start_date=h, machine=m).first()
-                    if s is None:
-                        free = True
-                if free and make_aware(datetime.now()) < h:
+                if self.slot_type == "BOTH" and self.check_slot("WASHING", h) and self.check_slot("DRYING", h + timedelta(hours=1)):
+                    print("GUY")
+                    free = True
+                elif self.slot_type == "WASHING" and self.check_slot("WASHING", h):
+                    free = True
+                elif self.slot_type == "DRYING" and self.check_slot("DRYING", h):
+                    free = True
+                if free and datetime.now().replace(tzinfo=pytz.UTC) < h:
                     kwargs['planning'][date].append(h)
                 else:
                     kwargs['planning'][date].append(None)
@@ -105,5 +134,37 @@ class LaunderetteEditView(CanViewMixin, UpdateView):
 class LaunderetteCreateView(CanCreateMixin, CreateView):
     """Create a new launderette"""
     model = Launderette
-    fields = ['name', 'sellers']
+    fields = ['name']
     template_name = 'core/create.jinja'
+
+
+class MachineEditView(CanEditPropMixin, UpdateView):
+    """Edit a machine"""
+    model = Machine
+    pk_url_kwarg = "machine_id"
+    fields = ['name', 'launderette', 'type', 'is_working']
+    template_name = 'core/edit.jinja'
+
+class MachineDeleteView(CanEditPropMixin, DeleteView):
+    """Edit a machine"""
+    model = Machine
+    pk_url_kwarg = "machine_id"
+    template_name = 'core/delete_confirm.jinja'
+    success_url = reverse_lazy('launderette:launderette_list')
+
+class MachineCreateView(CanCreateMixin, CreateView):
+    """Create a new machine"""
+    model = Machine
+    fields = ['name', 'launderette', 'type']
+    template_name = 'core/create.jinja'
+
+    def get_initial(self):
+        ret = super(MachineCreateView, self).get_initial()
+        if 'launderette' in self.request.GET.keys():
+            obj = Launderette.objects.filter(id=int(self.request.GET['launderette'])).first()
+            if obj is not None:
+                ret['launderette'] = obj.id
+        return ret
+
+
+