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 @@
''' book list views'''
""" book list views"""
from django.contrib.auth.decorators import login_required
from django.core.paginator import Paginator
from django.db import IntegrityError
@ -18,51 +18,57 @@ from .helpers import get_user_from_username
# pylint: disable=no-self-use
class Lists(View):
''' book list page '''
""" book list page """
def get(self, request):
''' display a book list '''
""" display a book list """
try:
page = int(request.GET.get('page', 1))
page = int(request.GET.get("page", 1))
except ValueError:
page = 1
user = request.user if request.user.is_authenticated else None
# hide lists with no approved books
lists = models.List.objects.filter(
~Q(user=user),
).annotate(
item_count=Count('listitem', filter=Q(listitem__approved=True))
).filter(
item_count__gt=0
).distinct().all()
lists = (
models.List.objects.filter(
~Q(user=user),
)
.annotate(item_count=Count("listitem", filter=Q(listitem__approved=True)))
.filter(item_count__gt=0)
.distinct()
.all()
)
lists = privacy_filter(
request.user, lists, privacy_levels=['public', 'followers'])
request.user, lists, privacy_levels=["public", "followers"]
)
paginated = Paginator(lists, 12)
data = {
'lists': paginated.page(page),
'list_form': forms.ListForm(),
'path': '/list',
"lists": paginated.page(page),
"list_form": forms.ListForm(),
"path": "/list",
}
return TemplateResponse(request, 'lists/lists.html', data)
return TemplateResponse(request, "lists/lists.html", data)
@method_decorator(login_required, name='dispatch')
@method_decorator(login_required, name="dispatch")
# pylint: disable=unused-argument
def post(self, request):
''' create a book_list '''
""" create a book_list """
form = forms.ListForm(request.POST)
if not form.is_valid():
return redirect('lists')
return redirect("lists")
book_list = form.save()
return redirect(book_list.local_path)
class UserLists(View):
''' a user's book list page '''
""" a user's book list page """
def get(self, request, username):
''' display a book list '''
""" display a book list """
try:
page = int(request.GET.get('page', 1))
page = int(request.GET.get("page", 1))
except ValueError:
page = 1
user = get_user_from_username(request.user, username)
@ -71,19 +77,20 @@ class UserLists(View):
paginated = Paginator(lists, 12)
data = {
'user': user,
'is_self': request.user.id == user.id,
'lists': paginated.page(page),
'list_form': forms.ListForm(),
'path': user.local_path + '/lists',
"user": user,
"is_self": request.user.id == user.id,
"lists": paginated.page(page),
"list_form": forms.ListForm(),
"path": user.local_path + "/lists",
}
return TemplateResponse(request, 'user/lists.html', data)
return TemplateResponse(request, "user/lists.html", data)
class List(View):
''' book list page '''
""" book list page """
def get(self, request, list_id):
''' display a book list '''
""" display a book list """
book_list = get_object_or_404(models.List, id=list_id)
if not object_visible_to_user(request.user, book_list):
return HttpResponseNotFound()
@ -91,7 +98,7 @@ class List(View):
if is_api_request(request):
return ActivitypubResponse(book_list.to_activity(**request.GET))
query = request.GET.get('q')
query = request.GET.get("q")
suggestions = None
if query and request.user.is_authenticated:
# search for books
@ -104,89 +111,85 @@ class List(View):
suggestions = [s.book for s in suggestions[:5]]
if len(suggestions) < 5:
suggestions += [
s.default_edition for s in \
models.Work.objects.filter(
~Q(editions__in=book_list.books.all()),
).order_by('-updated_date')
][:5 - len(suggestions)]
s.default_edition
for s in models.Work.objects.filter(
~Q(editions__in=book_list.books.all()),
).order_by("-updated_date")
][: 5 - len(suggestions)]
data = {
'list': book_list,
'items': book_list.listitem_set.filter(approved=True),
'pending_count': book_list.listitem_set.filter(
approved=False).count(),
'suggested_books': suggestions,
'list_form': forms.ListForm(instance=book_list),
'query': query or ''
"list": book_list,
"items": book_list.listitem_set.filter(approved=True),
"pending_count": book_list.listitem_set.filter(approved=False).count(),
"suggested_books": suggestions,
"list_form": forms.ListForm(instance=book_list),
"query": query or "",
}
return TemplateResponse(request, 'lists/list.html', data)
return TemplateResponse(request, "lists/list.html", data)
@method_decorator(login_required, name='dispatch')
@method_decorator(login_required, name="dispatch")
# pylint: disable=unused-argument
def post(self, request, list_id):
''' edit a list '''
""" edit a list """
book_list = get_object_or_404(models.List, id=list_id)
form = forms.ListForm(request.POST, instance=book_list)
if not form.is_valid():
return redirect('list', book_list.id)
return redirect("list", book_list.id)
book_list = form.save()
return redirect(book_list.local_path)
class Curate(View):
''' approve or discard list suggestsions '''
@method_decorator(login_required, name='dispatch')
""" approve or discard list suggestsions """
@method_decorator(login_required, name="dispatch")
def get(self, request, list_id):
''' display a pending list '''
""" display a pending list """
book_list = get_object_or_404(models.List, id=list_id)
if not book_list.user == request.user:
# only the creater can curate the list
return HttpResponseNotFound()
data = {
'list': book_list,
'pending': book_list.listitem_set.filter(approved=False),
'list_form': forms.ListForm(instance=book_list),
"list": book_list,
"pending": book_list.listitem_set.filter(approved=False),
"list_form": forms.ListForm(instance=book_list),
}
return TemplateResponse(request, 'lists/curate.html', data)
return TemplateResponse(request, "lists/curate.html", data)
@method_decorator(login_required, name='dispatch')
@method_decorator(login_required, name="dispatch")
# pylint: disable=unused-argument
def post(self, request, list_id):
''' edit a book_list '''
""" edit a book_list """
book_list = get_object_or_404(models.List, id=list_id)
suggestion = get_object_or_404(
models.ListItem, id=request.POST.get('item'))
approved = request.POST.get('approved') == 'true'
suggestion = get_object_or_404(models.ListItem, id=request.POST.get("item"))
approved = request.POST.get("approved") == "true"
if approved:
suggestion.approved = True
suggestion.save()
else:
suggestion.delete()
return redirect('list-curate', book_list.id)
return redirect("list-curate", book_list.id)
@require_POST
def add_book(request, list_id):
''' put a book on a list '''
""" put a book on a list """
book_list = get_object_or_404(models.List, id=list_id)
if not object_visible_to_user(request.user, book_list):
return HttpResponseNotFound()
book = get_object_or_404(models.Edition, id=request.POST.get('book'))
book = get_object_or_404(models.Edition, id=request.POST.get("book"))
# do you have permission to add to the list?
try:
if request.user == book_list.user or book_list.curation == 'open':
if request.user == book_list.user or book_list.curation == "open":
# go ahead and add it
models.ListItem.objects.create(
book=book,
book_list=book_list,
user=request.user,
)
elif book_list.curation == 'curated':
elif book_list.curation == "curated":
# make a pending entry
models.ListItem.objects.create(
approved=False,
@ -201,17 +204,17 @@ def add_book(request, list_id):
# if the book is already on the list, don't flip out
pass
return redirect('list', list_id)
return redirect("list", list_id)
@require_POST
def remove_book(request, list_id):
''' put a book on a list '''
""" put a book on a list """
book_list = get_object_or_404(models.List, id=list_id)
item = get_object_or_404(models.ListItem, id=request.POST.get('item'))
item = get_object_or_404(models.ListItem, id=request.POST.get("item"))
if not book_list.user == request.user and not item.user == request.user:
return HttpResponseNotFound()
item.delete()
return redirect('list', list_id)
return redirect("list", list_id)