Addition of the ShoppingList view to know the item to buy

stock/views.py

@@ -3,6 +3,7 @@ from django.views.generic import ListView, DetailView, RedirectView, TemplateVie
 from django.views.generic.edit import UpdateView, CreateView, DeleteView, ProcessFormView, FormMixin
 from django.utils.translation import ugettext_lazy as _
 from django import forms
+from django.http import HttpResponseRedirect, HttpResponse
 from django.forms.models import modelform_factory
 from django.core.urlresolvers import reverse_lazy, reverse
 
@@ -159,7 +160,140 @@ class StockShoppingListView(CounterAdminTabsMixin, CanViewMixin, ListView):
 			ret['stock'] = Stock.objects.filter(id=self.kwargs['stock_id']).first();
 		return ret
 
-class StockItemOutEditView(CounterAdminTabsMixin, CanViewMixin, UpdateView):
+
+class StockItemQuantityForm(forms.BaseForm):
+	def clean(self):
+		with transaction.atomic():
+			self.stock = Stock.objects.filter(id=self.stock_id).first()
+			shopping_list = ShoppingList(name="Courses "+self.stock.counter.name, date=timezone.now(), todo=True)
+			shopping_list.save()
+			shopping_list.stock_owner = self.stock
+			shopping_list.save()
+			for k,t in self.cleaned_data.items():
+				if t is not None:
+					item_id = int(k[5:])
+					item = StockItem.objects.filter(id=item_id).first()
+					item.tobuy_quantity = t
+					item.shopping_lists.add(shopping_list)
+					item.save()
+
+		return self.cleaned_data
+
+class StockItemQuantityBaseFormView(CounterAdminTabsMixin, CanEditMixin, DetailView, BaseFormView):
 	"""
 	docstring for StockItemOutList
 	"""
+	model = StockItem
+	template_name = "stock/shopping_list_quantity.jinja"
+	pk_url_kwarg = "stock_id"
+	current_tab = "stocks"
+
+	def get_form_class(self):
+		fields = OrderedDict()
+		kwargs = {}
+		for t in ProductType.objects.order_by('name').all():
+			for i in self.stock.items.filter(type=t).order_by('name').all():
+				if i.effective_quantity <= i.minimal_quantity:	            
+					field_name = "item-%s" % (str(i.id))
+					fields[field_name] = forms.CharField(max_length=30, required=True, label=str(i),
+								help_text=str(i.effective_quantity)+" left")
+					kwargs[field_name] = i.effective_quantity
+		kwargs['stock_id'] = self.stock.id
+		kwargs['base_fields'] = fields
+		return type('StockItemQuantityForm', (StockItemQuantityForm,), kwargs)
+
+	def get(self, request, *args, **kwargs):
+		"""
+		Simple get view
+		"""
+		self.stock = Stock.objects.filter(id=self.kwargs['stock_id']).first()
+		return super(StockItemQuantityBaseFormView, self).get(request, *args, **kwargs)
+	
+	def post(self, request, *args, **kwargs):
+		"""
+		Handle the many possibilities of the post request
+		"""
+		self.object = self.get_object()
+		self.stock = Stock.objects.filter(id=self.kwargs['stock_id']).first()
+		return super(StockItemQuantityBaseFormView, self).post(request, *args, **kwargs)
+
+	def form_valid(self, form):
+		"""
+		We handle here the redirection, passing the user id of the asked customer
+		"""
+		return super(StockItemQuantityBaseFormView, self).form_valid(form)
+
+	def get_context_data(self, **kwargs):
+		"""
+    	We handle here the login form for the barman
+    	"""
+		kwargs = super(StockItemQuantityBaseFormView, self).get_context_data(**kwargs)
+		if 'form' not in kwargs.keys():
+			kwargs['form'] = self.get_form()
+		kwargs['stock'] = self.stock
+		return kwargs
+
+	def get_success_url(self):
+		return reverse_lazy('stock:shoppinglist_list', args=self.args, kwargs=self.kwargs)
+
+
+class StockShoppingListItemListView(CounterAdminTabsMixin, CanViewMixin, ListView):
+	"""docstring for StockShoppingListItemListView"""
+	model = ShoppingList
+	template_name = "stock/shopping_list_items.jinja"
+	pk_url_kwarg = "shoppinglist_id"
+	current_tab = "stocks"
+
+	def get_context_data(self):
+		ret = super(StockShoppingListItemListView, self).get_context_data()
+		if 'shoppinglist_id' in self.kwargs.keys():
+			ret['shoppinglist'] = ShoppingList.objects.filter(id=self.kwargs['shoppinglist_id']).first();
+		return ret
+
+class StockShoppingListDeleteView(CounterAdminTabsMixin, CanEditMixin, DeleteView):
+    """
+    Delete a ShoppingList (for the resonsible account)
+    """
+    model = ShoppingList
+    pk_url_kwarg = "shoppinglist_id"
+    template_name = 'core/delete_confirm.jinja'
+    current_tab = "stocks"
+
+    def get_success_url(self):
+    	return reverse_lazy('stock:shoppinglist_list', kwargs={'stock_id':self.object.stock_owner.id})
+
+
+class StockShopppingListSetDone(CanEditMixin, DetailView):
+    """
+    Set a ShoppingList as done
+    """
+    model = ShoppingList
+    pk_url_kwarg = "shoppinglist_id"
+
+    def get(self, request, *args, **kwargs):
+        self.object = self.get_object()
+        self.object.todo = False
+        self.object.save()
+        return HttpResponseRedirect(reverse('stock:shoppinglist_list', args=self.args, kwargs={'stock_id':self.object.stock_owner.id}))
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+        return HttpResponseRedirect(reverse('stock:shoppinglist_list', args=self.args, kwargs={'stock_id':self.object.stock_owner.id}))
+
+
+class StockShopppingListSetTodo(CanEditMixin, DetailView):
+    """
+    Set a ShoppingList as done
+    """
+    model = ShoppingList
+    pk_url_kwarg = "shoppinglist_id"
+
+    def get(self, request, *args, **kwargs):
+        self.object = self.get_object()
+        self.object.todo = True
+        self.object.save()
+        return HttpResponseRedirect(reverse('stock:shoppinglist_list', args=self.args, kwargs={'stock_id':self.object.stock_owner.id}))
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+        return HttpResponseRedirect(reverse('stock:shoppinglist_list', args=self.args, kwargs={'stock_id':self.object.stock_owner.id}))