klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2024-11-22 06:03:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Initial StockItem create form value addition

Browse Source
This commit is contained in:
guillaume-renaud 2016-11-11 16:55:52 +01:00
parent 887893fb2d
commit 587ad96326
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
stock/views.py
View File

@ -1,4 +1,4 @@
from django.shortcuts import render
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 _
@ -39,7 +39,7 @@ class StockEditForm(forms.ModelForm):
"""
class Meta:
model = Stock
fields = ['name']
fields = ['name', 'counter']
def __init__(self, *args, **kwargs):
super(StockEditForm, self).__init__(*args, **kwargs)
@ -81,5 +81,11 @@ class StockItemCreateView(CounterAdminTabsMixin, CanCreateMixin, CreateView):
pk_url_kwarg = "stock_id"
current_tab = "stocks"
def get_initial(self):
ret = super(StockItemCreateView, self).get_initial()
if 'stock' in self.request.GET.keys():
ret['stock_owner'] = self.request.GET['stock']
return ret
def get_success_url(self):
return reverse_lazy('stock:main', kwargs={'stock_id': self.object.stock_owner.id})