From 7a9689a20d2a8604aaf3ce4f2b1464d440a353d2 Mon Sep 17 00:00:00 2001
From: Skia <skia@libskia.so>
Date: Mon, 18 Apr 2016 17:34:21 +0200
Subject: [PATCH] Improve the click view, the total is computed, but there is
 still a lot of work

---
 counter/templates/counter/counter_click.jinja |  9 ++++
 counter/views.py                              | 49 +++++++++++++++++--
 2 files changed, 53 insertions(+), 5 deletions(-)

diff --git a/counter/templates/counter/counter_click.jinja b/counter/templates/counter/counter_click.jinja
index 8c072c62..16246997 100644
--- a/counter/templates/counter/counter_click.jinja
+++ b/counter/templates/counter/counter_click.jinja
@@ -14,6 +14,15 @@
     <p><strong>Club: </strong> {{ counter.club }}</p>
     <p><strong>Products: </strong> {{ counter.products.all() }}</p>
 
+    <div>
+        Customer: {{ customer }}
+        <form method="post" action="{{ url('counter:click', counter_id=counter.id, user_id=customer.user.id) }}">
+            {% csrf_token %}
+            {{ form.as_p() }}
+        <input type="submit" value="CLICK" />
+        </form>
+    </div>
+
 {% endblock %}
 
 
diff --git a/counter/views.py b/counter/views.py
index 1884b23b..2cde332b 100644
--- a/counter/views.py
+++ b/counter/views.py
@@ -13,7 +13,7 @@ from datetime import timedelta
 
 from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin
 from subscription.models import Subscriber
-from accounting.models import Customer
+from accounting.models import Customer, Product
 from counter.models import Counter
 
 class GetUserForm(forms.Form):
@@ -84,20 +84,57 @@ class CounterMain(DetailView, ProcessFormView, FormMixin):
     def get_success_url(self):
         return reverse_lazy('counter:click', args=self.args, kwargs=self.kwargs)
 
+class BasketForm(forms.Form):
+    def __init__(self, *args, **kwargs):
+        print(kwargs)
+        super(BasketForm, self).__init__(*args, **kwargs)
+        for p in kwargs['initial']['counter'].products.all(): # TODO: filter on the allowed products for this counter
+            self.fields[p.id] = forms.IntegerField(label=p.name, required=False)
+            # TODO ^: add some extra infos for the products (price, or ID to load infos dynamically)
+
+    def clean(self):
+        cleaned_data = super(BasketForm, self).clean()
+        total = 0
+        for pid,q in cleaned_data.items():
+            p = Product.objects.filter(id=pid).first()
+            total += (q or 0)*p.selling_price
+        print(total)
+
 class CounterClick(DetailView, ProcessFormView, FormMixin):
     """
     The click view
     """
-    model = Counter # TODO change that to a basket class
+    model = Counter
     template_name = 'counter/counter_click.jinja'
     pk_url_kwarg = "counter_id"
-    form_class = GetUserForm
+    form_class = BasketForm
+    prefix = "prod"
+
+    def get(self, request, *args, **kwargs):
+        self.customer = Customer.objects.filter(user__id=self.kwargs['user_id']).first()
+        return super(CounterClick, self).get(request, *args, **kwargs)
 
     def post(self, request, *args, **kwargs):
-        # TODO: handle the loading of a user, to display the click view
-        # TODO: Do the form and the template for the click view
+        self.object = self.get_object()
+        self.customer = Customer.objects.filter(user__id=self.kwargs['user_id']).first()
         return super(CounterClick, self).post(request, *args, **kwargs)
 
+    def form_valid(self, form):
+        return super(CounterClick, self).form_valid(form)
+
+    def get_form_kwargs(self):
+        kwargs = super(CounterClick, self).get_form_kwargs()
+        kwargs['initial'].update({'counter': self.object, 'customer': self.customer})
+        return kwargs
+
+    def get_context_data(self, **kwargs):
+        """
+        """
+        kwargs = super(CounterClick, self).get_context_data(**kwargs)
+        kwargs['customer'] = self.customer
+        kwargs['form'] = self.get_form()
+        return kwargs
+
     def get_success_url(self):
         return reverse_lazy('counter:click', args=self.args, kwargs=self.kwargs)
 
@@ -140,6 +177,8 @@ class CounterLogout(RedirectView):
     def get_redirect_url(self, *args, **kwargs):
         return reverse_lazy('counter:details', args=args, kwargs=kwargs)
 
+## Counter admin views
+
 class CounterListView(CanViewMixin, ListView):
     """
     A list view for the admins