from django.shortcuts import render, get_object_or_404 from django.views.generic import ListView, DetailView, RedirectView, TemplateView 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 from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, CanCreateMixin, TabedViewMixin from counter.views import CounterAdminTabsMixin, CounterTabsMixin from counter.models import Counter, ProductType from stock.models import Stock, StockItem, ShoppingList class StockItemList(CounterAdminTabsMixin, CanCreateMixin, ListView): """ The stockitems list view for the counter owner """ model = Stock template_name = 'stock/stock_item_list.jinja' pk_url_kwarg = "stock_id" current_tab = "stocks" def can_be_viewed_by(self, user): return user.is_in_group(settings.SITH_GROUP_COUNTER_ADMIN_ID) class StockListView(CounterAdminTabsMixin, CanViewMixin, ListView): """ A list view for the admins """ model = Stock template_name = 'stock/stock_list.jinja' current_tab = "stocks" def can_be_viewed_by(self, user): return user.is_in_group(settings.SITH_GROUP_COUNTER_ADMIN_ID) class StockEditForm(forms.ModelForm): """ A form to change stock's characteristics """ class Meta: model = Stock fields = ['name', 'counter'] def __init__(self, *args, **kwargs): super(StockEditForm, self).__init__(*args, **kwargs) def save(self, *args, **kwargs): return super(StockEditForm, self).save(*args, **kwargs) class StockEditView(CounterAdminTabsMixin, CanEditPropMixin, UpdateView): """ An edit view for the stock """ model = Stock form_class = StockEditForm pk_url_kwarg = "stock_id" current_tab = "stocks" def get_context_data(self, **kwargs): context = super(StockItemList, self).get_context_data(**kwargs) if 'stock' in self.request.GET.keys(): context['stock'] = Stock.objects.filter(id=self.request.GET['stock']).first() return context def can_be_viewed_by(self, user): return user.is_in_group(settings.SITH_GROUP_COUNTER_ADMIN_ID) class StockItemEditView(CounterAdminTabsMixin, CanEditPropMixin, UpdateView): """ An edit view for a stock item """ model = StockItem form_class = modelform_factory(StockItem, fields=['name', 'unit_quantity', 'effective_quantity', 'type', 'stock_owner']) pk_url_kwarg = "item_id" template_name = 'core/edit.jinja' current_tab = "stocks" class StockCreateView(CounterAdminTabsMixin, CanCreateMixin, CreateView): """ A create view for a new Stock """ model = Stock form_class = modelform_factory(Stock, fields=['name', 'counter']) template_name = 'core/create.jinja' pk_url_kwarg = "counter_id" current_tab = "stocks" success_url = reverse_lazy('stock:list') def get_initial(self): ret = super(StockCreateView, self).get_initial() if 'counter_id' in self.kwargs.keys(): ret['counter'] = self.kwargs['counter_id'] return ret class StockItemCreateView(CounterAdminTabsMixin, CanCreateMixin, CreateView): """ A create view for a new StockItem """ model = StockItem form_class = modelform_factory(StockItem, fields=['name', 'unit_quantity', 'effective_quantity', 'type', 'stock_owner']) template_name = 'core/create.jinja' pk_url_kwarg = "stock_id" current_tab = "stocks" def get_initial(self): ret = super(StockItemCreateView, self).get_initial() if 'stock_id' in self.kwargs.keys(): ret['stock_owner'] = self.kwargs['stock_id'] return ret def get_success_url(self): return reverse_lazy('stock:items_list', kwargs={'stock_id':self.object.stock_owner.id}) class StockShoppingListView(CounterAdminTabsMixin, CanViewMixin, ListView): """ A list view for the people to know the item to buy """ model = Stock template_name = "stock/stock_shopping_list.jinja" pk_url_kwarg = "stock_id" current_tab = "stocks" def get_context_data(self): ret = super(StockShoppingListView, self).get_context_data() if 'stock_id' in self.kwargs.keys(): ret['stock'] = Stock.objects.filter(id=self.kwargs['stock_id']).first(); return ret 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 int(t) > 0 : 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): return super(StockItemQuantityBaseFormView, self).form_valid(form) def get_context_data(self, **kwargs): 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})) class StockUpdateAfterShopppingForm(forms.BaseForm): def clean(self): with transaction.atomic(): self.shoppinglist = ShoppingList.objects.filter(id=self.shoppinglist_id).first() for k,t in self.cleaned_data.items(): item_id = int(k[5:]) if int(t) > 0 : item = StockItem.objects.filter(id=item_id).first() item.effective_quantity += int(t) item.save() self.shoppinglist.todo = False self.shoppinglist.save() return self.cleaned_data class StockUpdateAfterShopppingBaseFormView(CounterAdminTabsMixin, CanEditMixin, DetailView, BaseFormView): """ docstring for StockUpdateAfterShopppingBaseFormView """ model = ShoppingList template_name = "stock/update_after_shopping.jinja" pk_url_kwarg = "shoppinglist_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.shoppinglist.items_to_buy.filter(type=t).order_by('name').all(): field_name = "item-%s" % (str(i.id)) fields[field_name] = forms.CharField(max_length=30, required=True, label=str(i), help_text=str(i.tobuy_quantity) + " asked") kwargs['shoppinglist_id'] = self.shoppinglist.id kwargs['base_fields'] = fields return type('StockUpdateAfterShopppingForm', (StockUpdateAfterShopppingForm,), kwargs) def get(self, request, *args, **kwargs): self.shoppinglist = ShoppingList.objects.filter(id=self.kwargs['shoppinglist_id']).first() return super(StockUpdateAfterShopppingBaseFormView, 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.shoppinglist = ShoppingList.objects.filter(id=self.kwargs['shoppinglist_id']).first() return super(StockUpdateAfterShopppingBaseFormView, self).post(request, *args, **kwargs) def form_valid(self, form): """ We handle here the redirection """ return super(StockUpdateAfterShopppingBaseFormView, self).form_valid(form) def get_context_data(self, **kwargs): kwargs = super(StockUpdateAfterShopppingBaseFormView, self).get_context_data(**kwargs) if 'form' not in kwargs.keys(): kwargs['form'] = self.get_form() kwargs['shoppinglist'] = self.shoppinglist kwargs['stock'] = self.shoppinglist.stock_owner return kwargs def get_success_url(self): self.kwargs.pop('shoppinglist_id', None) return reverse_lazy('stock:shoppinglist_list', args=self.args, kwargs=self.kwargs) class StockTakeItemsForm(forms.BaseForm): """ docstring for StockTakeItemsFormView """ def clean(self): with transaction.atomic(): for k,t in self.cleaned_data.items(): item_id = int(k[5:]) if int(t) > 0 : item = StockItem.objects.filter(id=item_id).first() item.effective_quantity -= int(t) item.save() return self.cleaned_data class StockTakeItemsBaseFormView(CounterTabsMixin, CanEditMixin, DetailView, BaseFormView): """ docstring for StockTakeItemsBaseFormView """ model = StockItem template_name = "stock/stock_take_items.jinja" pk_url_kwarg = "stock_id" current_tab = "take_items_from_stock" 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(): field_name = "item-%s" % (str(i.id)) fields[field_name] = forms.CharField(max_length=30, required=True, label=str(i)) kwargs[field_name] = i.effective_quantity kwargs['stock_id'] = self.stock.id kwargs['counter_id'] = self.stock.counter.id kwargs['base_fields'] = fields return type('StockTakeItemsForm', (StockTakeItemsForm,), kwargs) def get(self, request, *args, **kwargs): """ Simple get view """ self.stock = Stock.objects.filter(id=self.kwargs['stock_id']).first() return super(StockTakeItemsBaseFormView, 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() if self.stock.counter.type == "BAR" and not ('counter_token' in self.request.session.keys() and self.request.session['counter_token'] == self.stock.counter.token): # Also check the token to avoid the bar to be stolen return HttpResponseRedirect(reverse_lazy('counter:details', args=self.args, kwargs={'counter_id': self.stock.counter.id})+'?bad_location') return super(StockTakeItemsBaseFormView, self).post(request, *args, **kwargs) def form_valid(self, form): return super(StockTakeItemsBaseFormView, self).form_valid(form) def get_context_data(self, **kwargs): kwargs = super(StockTakeItemsBaseFormView, self).get_context_data(**kwargs) if 'form' not in kwargs.keys(): kwargs['form'] = self.get_form() kwargs['stock'] = self.stock kwargs['counter'] = self.stock.counter return kwargs def get_success_url(self): stock = Stock.objects.filter(id=self.kwargs['stock_id']).first() self.kwargs['counter_id'] = stock.counter.id self.kwargs.pop('stock_id', None) return reverse_lazy('counter:details', args=self.args, kwargs=self.kwargs)