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})