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