moviewyrm/bookwyrm/views/tag.py

''' tagging views'''
from django.contrib.auth.decorators import login_required
from django.shortcuts import get_object_or_404, redirect
from django.template.response import TemplateResponse
from django.utils.decorators import method_decorator
from django.views import View

from bookwyrm import models
from bookwyrm.activitypub import ActivitypubResponse
from .helpers import is_api_request


# pylint: disable= no-self-use
class Tag(View):
    ''' tag page '''
    def get(self, request, tag_id):
        ''' see books related to a tag '''
        tag_obj = get_object_or_404(models.Tag, identifier=tag_id)

        if is_api_request(request):
            return ActivitypubResponse(
                tag_obj.to_activity(**request.GET))

        books = models.Edition.objects.filter(
            usertag__tag__identifier=tag_id
        ).distinct()
        data = {
            'books': books,
            'tag': tag_obj,
        }
        return TemplateResponse(request, 'tag.html', data)


@method_decorator(login_required, name='dispatch')
class AddTag(View):
    ''' add a tag to a book '''
    def post(self, request):
        ''' tag a book '''
        # I'm not using a form here because sometimes "name" is sent as a hidden
        # field which doesn't validate
        name = request.POST.get('name')
        book_id = request.POST.get('book')
        book = get_object_or_404(models.Edition, id=book_id)
        tag_obj, _ = models.Tag.objects.get_or_create(
            name=name,
        )
        models.UserTag.objects.get_or_create(
            user=request.user,
            book=book,
            tag=tag_obj,
        )

        return redirect('/book/%s' % book_id)


@method_decorator(login_required, name='dispatch')
class RemoveTag(View):
    ''' remove a user's tag from a book '''
    def post(self, request):
        ''' untag a book '''
        name = request.POST.get('name')
        tag_obj = get_object_or_404(models.Tag, name=name)
        book_id = request.POST.get('book')
        book = get_object_or_404(models.Edition, id=book_id)

        user_tag = get_object_or_404(
            models.UserTag, tag=tag_obj, book=book, user=request.user)
        user_tag.delete()

        return redirect('/book/%s' % book_id)