def category_update(request, pk): category = get_object_or_404(Category, pk=pk) if request.method == 'POST': f = CategoryForm(request.POST, instance=category) if f.is_valid(): if request.POST.get('author') == '' and request.user.is_superuser: updated_category = f.save(commit=False) updated_category.author = Author.objects.get( user__username='******') updated_category.save() f.save_m2m() elif request.POST.get('author') and request.user.is_superuser: f.save() else: updated_category = f.save(commit=False) updated_category.author = Author.objects.get( user__username=request.user.username) updated_category.save() f.save_m2m() messages.add_message(request, messages.INFO, 'Category updated') redirect('category_update', pk=pk) else: print(f.errors) else: f = CategoryForm(instance=category) return render(request, 'cadmin/category_update.html', {'form': f})
def category_add(request): if request.method == 'POST': f = CategoryForm(request.POST) if f.is_valid(): if request.POST.get('author') == '' and request.user.is_superuser: new_category = f.save(commit=False) new_category.author = Author.objects.get( user__username='******') new_category.save() f.save_m2m() elif request.POST.get('author') and request.user.is_superuser: f.save() else: new_category = f.save(commit=False) new_category.author = Author.objects.get( user__username=request.user.username) new_category.save() f.save_m2m() messages.add_message(request, messages.INFO, 'Category added') return redirect('category_add') else: print(f.errors) else: f = CategoryForm() return render(request, 'cadmin/category_add.html', {'form': f})