forked from mirrors/bookwyrm
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
''' tagging views'''
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.http import HttpResponseNotFound
|
|
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 bookwyrm.broadcast import broadcast
|
|
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 = models.Tag.objects.filter(identifier=tag_id).first()
|
|
if not tag_obj:
|
|
return HttpResponseNotFound()
|
|
|
|
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 = {
|
|
'title': tag_obj.name,
|
|
'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, created = models.Tag.objects.get_or_create(
|
|
name=name,
|
|
)
|
|
user_tag, _ = models.UserTag.objects.get_or_create(
|
|
user=request.user,
|
|
book=book,
|
|
tag=tag_obj,
|
|
)
|
|
|
|
if created:
|
|
broadcast(request.user, user_tag.to_add_activity(request.user))
|
|
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)
|
|
tag_activity = user_tag.to_remove_activity(request.user)
|
|
user_tag.delete()
|
|
|
|
broadcast(request.user, tag_activity)
|
|
return redirect('/book/%s' % book_id)
|