def add_s(request):
"""Добавление нового складского помещения.
"""
context = dict()
if request.method == 'POST':
form =AddStorage(request.POST)
if form.is_valid():
try:
m1 = request.user.departament.pk
except AttributeError:
# если пользователь не ассоциирован с организацией,
# то сообщаем об ошибке
messages.error(request, _('User not assosiate with organization unit!<br/>Error code: 101.'))
context['form'] = form
return render(request, 'storages/add_s.html', context);
data = form.cleaned_data
title = data.get('title')
address = data.get('address')
description = data.get('description')
# первый создаваемый объект устанавливается по умолчанию
flag = False
try:
Storages.objects.get(pk=1)
except Storages.DoesNotExist:
flag = True
m1 = Storages(
title = title,
address = address,
departament = request.user.departament,
description = description,
default = flag
)
m1.save()
messages.success(request, _('Starage "%(starage_name)s" success added.') % {'starage_name': title})
context['form'] = form
else:
context['form'] = form
messages.error(request, _('Starage not added.'))
else:
form = AddStorage()
context['form'] = form
return render(request, 'storages/add_s.html', context);
def edit_s(request):
"""Редактирование информации о складском помещении.
"""
context = dict()
select = request.GET.get('select', 0);
try:
select = int(select)
except ValueError:
select = 0
try:
m1 = Storages.objects.get(pk=select)
except Storages.DoesNotExist:
messages.error(request, _('Storage by id not found.'))
m1 = False
if request.method == 'POST':
form =AddStorage(request.POST)
if form.is_valid():
data = form.cleaned_data
title = data.get('title')
address = data.get('address')
description = data.get('description')
m1.title = title
m1.address = address
m1.description = description
m1.save()
messages.success(request, _('Starage "%(starage_name)s" success edited.') % {'starage_name': title})
context['form'] = form
else:
context['form'] = form
messages.error(request, _('Starage not edited.'))
else:
if m1:
form =AddStorage(initial={
'title': m1.title,
'address': m1.address,
'description': m1.description})
context['form'] = form
return render(request, 'storages/edit_s.html', context);
