moviewyrm/bookwyrm/views/interaction.py

""" boosts and favs """
from django.db import IntegrityError
from django.contrib.auth.decorators import login_required
from django.http import HttpResponseBadRequest, HttpResponseNotFound
from django.shortcuts import redirect
from django.utils.decorators import method_decorator
from django.views import View

from bookwyrm import models


# pylint: disable= no-self-use
@method_decorator(login_required, name="dispatch")
class Favorite(View):
    """like a status"""

    def post(self, request, status_id):
        """create a like"""
        status = models.Status.objects.get(id=status_id)
        try:
            models.Favorite.objects.create(status=status, user=request.user)
        except IntegrityError:
            # you already fav'ed that
            return HttpResponseBadRequest()

        return redirect(request.headers.get("Referer", "/"))


@method_decorator(login_required, name="dispatch")
class Unfavorite(View):
    """take back a fav"""

    def post(self, request, status_id):
        """unlike a status"""
        status = models.Status.objects.get(id=status_id)
        try:
            favorite = models.Favorite.objects.get(status=status, user=request.user)
        except models.Favorite.DoesNotExist:
            # can't find that status, idk
            return HttpResponseNotFound()

        favorite.delete()
        return redirect(request.headers.get("Referer", "/"))


@method_decorator(login_required, name="dispatch")
class Boost(View):
    """boost a status"""

    def post(self, request, status_id):
        """boost a status"""
        status = models.Status.objects.get(id=status_id)
        # is it boostable?
        if not status.boostable:
            return HttpResponseBadRequest()

        if models.Boost.objects.filter(
            boosted_status=status, user=request.user
        ).exists():
            # you already boosted that.
            return redirect(request.headers.get("Referer", "/"))

        models.Boost.objects.create(
            boosted_status=status,
            privacy=status.privacy,
            user=request.user,
        )
        return redirect(request.headers.get("Referer", "/"))


@method_decorator(login_required, name="dispatch")
class Unboost(View):
    """boost a status"""

    def post(self, request, status_id):
        """boost a status"""
        status = models.Status.objects.get(id=status_id)
        boost = models.Boost.objects.filter(
            boosted_status=status, user=request.user
        ).first()

        boost.delete()
        return redirect(request.headers.get("Referer", "/"))