Runs black

This commit is contained in:
Mouse Reeve
2021-03-08 08:49:10 -08:00
parent a07f955781
commit 70296e760b
198 changed files with 10239 additions and 8572 deletions

View File

@ -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", "/"))