from django.shortcuts import render 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.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 from counter.models import Counter from stock.models import Stock, StockItem class StockMain(CounterAdminTabsMixin, CanCreateMixin, DetailView): """ The stock view for the counter owner """ model = Stock template_name = 'stock/stock_main.jinja' pk_url_kwarg = "stock_id" current_tab = "stocks" class StockListView(CounterAdminTabsMixin, CanViewMixin, ListView): """ A list view for the admins """ model = Stock template_name = 'stock/stock_list.jinja' current_tab = "stocks" class StockEditForm(forms.ModelForm): """ docstring for StockEditForm"forms.ModelForm """ class Meta: model = Stock fields = ['name'] 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): """ A edit view for the stock """ model = Stock form_class = StockEditForm pk_url_kwarg = "stock_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') 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', 'stock_owner']) template_name = 'core/create.jinja' pk_url_kwarg = "stock_id" current_tab = "stocks" def get_success_url(self): return reverse_lazy('stock:main', kwargs={'stock_id': self.object.stock_owner.id})