Runs black
This commit is contained in:
@ -1,4 +1,4 @@
|
||||
''' boosts and favs '''
|
||||
""" boosts and favs """
|
||||
from django.db import IntegrityError
|
||||
from django.contrib.auth.decorators import login_required
|
||||
from django.http import HttpResponseBadRequest, HttpResponseNotFound
|
||||
@ -10,75 +10,74 @@ from bookwyrm import models
|
||||
|
||||
|
||||
# pylint: disable= no-self-use
|
||||
@method_decorator(login_required, name='dispatch')
|
||||
@method_decorator(login_required, name="dispatch")
|
||||
class Favorite(View):
|
||||
''' like a status '''
|
||||
""" like a status """
|
||||
|
||||
def post(self, request, status_id):
|
||||
''' create a like '''
|
||||
""" create a like """
|
||||
status = models.Status.objects.get(id=status_id)
|
||||
try:
|
||||
models.Favorite.objects.create(
|
||||
status=status,
|
||||
user=request.user
|
||||
)
|
||||
models.Favorite.objects.create(status=status, user=request.user)
|
||||
except IntegrityError:
|
||||
# you already fav'ed that
|
||||
return HttpResponseBadRequest()
|
||||
|
||||
return redirect(request.headers.get('Referer', '/'))
|
||||
return redirect(request.headers.get("Referer", "/"))
|
||||
|
||||
|
||||
@method_decorator(login_required, name='dispatch')
|
||||
@method_decorator(login_required, name="dispatch")
|
||||
class Unfavorite(View):
|
||||
''' take back a fav '''
|
||||
""" take back a fav """
|
||||
|
||||
def post(self, request, status_id):
|
||||
''' unlike a status '''
|
||||
""" unlike a status """
|
||||
status = models.Status.objects.get(id=status_id)
|
||||
try:
|
||||
favorite = models.Favorite.objects.get(
|
||||
status=status,
|
||||
user=request.user
|
||||
)
|
||||
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', '/'))
|
||||
return redirect(request.headers.get("Referer", "/"))
|
||||
|
||||
|
||||
@method_decorator(login_required, name='dispatch')
|
||||
@method_decorator(login_required, name="dispatch")
|
||||
class Boost(View):
|
||||
''' boost a status '''
|
||||
""" boost a status """
|
||||
|
||||
def post(self, request, status_id):
|
||||
''' boost a status '''
|
||||
""" 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():
|
||||
boosted_status=status, user=request.user
|
||||
).exists():
|
||||
# you already boosted that.
|
||||
return redirect(request.headers.get('Referer', '/'))
|
||||
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', '/'))
|
||||
return redirect(request.headers.get("Referer", "/"))
|
||||
|
||||
|
||||
@method_decorator(login_required, name='dispatch')
|
||||
@method_decorator(login_required, name="dispatch")
|
||||
class Unboost(View):
|
||||
''' boost a status '''
|
||||
""" boost a status """
|
||||
|
||||
def post(self, request, status_id):
|
||||
''' boost a status '''
|
||||
""" 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', '/'))
|
||||
return redirect(request.headers.get("Referer", "/"))
|
||||
|
Reference in New Issue
Block a user