def save(request):
if request.POST:
form = CompanyForm(request.POST)
if form.is_valid():
new_company = form.save(commit=False)
new_company.owner = request.user
new_company.save()
return HttpResponseRedirect(reverse('companies:index'))
return HttpResponse("Passed confirmed did not get the post")
def update_company(request, company_id):
results = {}
if request.is_ajax():
company_form = CompanyForm(request.POST, instance=Company.objects.get(id=company_id))
if company_form.is_valid():
company_form.save()
results['return'] = True
else:
results['errors'] = company_form.errors
results['return'] = False
else:
results['return'] = False
return HttpResponse(json.dumps(results))
def update(request, pk=None):
obj = get_object_or_404(Company, pk=pk)
form = CompanyForm(request.POST or None,
request.FILES or None,
instance=obj)
if request.method == 'POST':
if form.owner != request.user:
context = {
'warning_message': 'Only the owner is allowed to update!'
}
return render(request, 'companies/warning.html', context)
if form.is_valid():
return redirect('/companies/')
return render(request, 'companies/update.html', {'form': form})
def add_company(request):
form1 = UserProfileForm(request.POST)
form2 = UserCreateForm(request.POST)
form3 = CompanyForm(request.POST)
if form1.is_valid() and form2.is_valid() and form3.is_valid():
up = form1.save(commit=False)
c = form3.save()
c.active = True
c.favorite = True
c.save()
u = form2.save()
up.user = u
up.save()
up.companies.add(c)
for user in User.objects.filter(is_superuser=True):
user.userprofile.companies.add(c)
user.userprofile.save()
# add dossier global
fy_init = FiscalYear.objects.filter(init=True)[0]
y_init = Year(fiscal_year=fy_init, active=True, refer_company=c, favorite=False)
y_init.save()
c.years.add(y_init)
tt_init = TemplateTrimester.objects.filter(year=fy_init, favorite=True)[0]
tri_init = Trimester(template=tt_init, start_date=tt_init.start_date, active=True, refer_year=y_init, favorite=True)
tri_init.save()
y_init.trimesters.add(tri_init)
tp_init = TypeCategory.objects.filter(priority=10)[0]
cat_init = Category(cat=tp_init, refer_trimester=tri_init, active=True)
cat_init.save()
tri_init.categories.add(cat_init)
# add favorite_year and favorite_trimester
fy_fav = FiscalYear.objects.filter(favorite=True)[0]
y_fav = Year(fiscal_year=fy_fav, active=True, refer_company=c, favorite=True)
y_fav.save()
c.years.add(y_fav)
tt_fav = TemplateTrimester.objects.filter(year=fy_fav, favorite=True)[0]
tri_fav = Trimester(template=tt_fav, start_date=tt_fav.start_date, active=True, refer_year=y_fav, favorite=True)
tri_fav.save()
y_fav.trimesters.add(tri_fav)
for tp in TypeCategory.objects.filter(priority__lt=10).order_by('priority'):
cat_fav = Category(cat=tp, refer_trimester=tri_fav, active=True)
cat_fav.save()
tri_fav.categories.add(cat_fav)
c = {'return': True, 'list': Company.objects.all(), 'form': [UserProfileForm(), UserCreateForm(), CompanyForm()], 'url': '/company/add/'}
return render(request, 'list.tpl', c)
else:
c = {'view_form': True, 'list': Company.objects.all(), 'form': [form1(), form2(), form3()]}
return render(request, 'list.tpl', c)
def add(request):
company_form = CompanyForm()
return render(request, 'companies/add.html', {'CompanyForm': company_form})
