bookwyrm/bookwyrm/views/get_started.py
2021-03-31 14:53:15 -07:00

75 lines
2.4 KiB
Python

""" Helping new users figure out the lay of the land """
from django.contrib.auth.decorators import login_required
from django.db.models import Count, Q
from django.template.response import TemplateResponse
from django.utils.decorators import method_decorator
from django.views import View
from bookwyrm import forms, models
from bookwyrm.connectors import connector_manager
from .helpers import get_suggested_users
# pylint: disable= no-self-use
@method_decorator(login_required, name="dispatch")
class GetStartedProfile(View):
""" tell us about yourself """
def get(self, request):
""" basic profile info """
data = {
"form": forms.EditUserForm(instance=request.user),
"next": "get-started-books",
}
return TemplateResponse(request, "get_started/profile.html", data)
@method_decorator(login_required, name="dispatch")
class GetStartedBooks(View):
""" name a book, any book, we gotta start somewhere """
def get(self, request):
""" info about a book """
query = request.GET.get("query")
book_results = []
if query:
book_results = connector_manager.local_search(query, raw=True)[:5]
if len(book_results) < 5:
popular_books = (
models.Edition.objects.exclude(
parent_work__in=[b.parent_work for b in book_results],
)
.annotate(Count("shelfbook"))
.order_by("-shelfbook__count")[: 5 - len(book_results)]
)
data = {
"book_results": book_results,
"popular_books": popular_books,
"next": "get-started-users",
}
return TemplateResponse(request, "get_started/books.html", data)
@method_decorator(login_required, name="dispatch")
class GetStartedUsers(View):
""" find friends """
def get(self, request):
""" basic profile info """
suggested_users = (
get_suggested_users(
request.user,
~Q(id=request.user.id),
~Q(followers=request.user),
bookwyrm_user=True,
)
.order_by("shared_books", "-mutuals", "-last_active_date")
.all()[:5]
)
data = {
"suggested_users": suggested_users,
"next": "get-started-profile",
}
return TemplateResponse(request, "get_started/users.html", data)