예제 #1
0
def tag_update(request, pk):
    tag = get_object_or_404(Tag, pk=pk)
    if request.method == 'POST':
        f = TagForm(request.POST, instance=tag)
        if f.is_valid():
            if request.POST.get('author') == '' and request.user.is_superuser:
                updated_tag = f.save(commit=False)
                updated_tag.author = Author.objects.get(user__username='******')
                updated_tag.save()
                f.save_m2m()

            elif request.POST.get('author') and request.user.is_superuser:
                updated_tag = f.save()

            else:
                updated_tag = f.save(commit=False)
                updated_tag.author = Author.objects.get(
                    user__username=request.user.username)
                updated_tag.save()
                f.save_m2m()

            messages.add_message(request, messages.INFO, 'Tag updated')
            return redirect('tag_update', pk=pk)

        else:
            print(f.errors)

    else:
        f = TagForm(instance=tag)

    return render(request, 'cadmin/tag_update.html', {'form': f})
예제 #2
0
def tag_add(request):
    if request.method == 'POST':
        f = TagForm(request.POST)
        if f.is_valid():
            if request.POST.get('author') == '' and request.user.is_superuser:
                new_tag = f.save(commit=False)
                new_tag.author = Author.objects.get(user__username='******')
                new_tag.save()
                f.save_m2m()

            elif request.POST.get('author') and request.user.is_superuser:
                new_tag = f.save()

            else:
                new_tag = f.save(commit=False)
                new_tag.author = Author.objects.get(
                    user__username=request.user.username)
                new_tag.save()
                f.save_m2m()

            messages.add_message(request, messages.INFO, 'Tag added')
            return redirect('tag_add')

        else:
            print(f.errors)

    else:
        f = TagForm()

    return render(request, 'cadmin/tag_add.html', {'form': f})