rafaelcaricio/moviewyrm
1
0
Fork 0
forked from mirrors/bookwyrm
Code Issues 4 Pull requests Projects Releases Wiki Activity

Use aggregate queries

Browse source
This commit is contained in:
Joachim 2021-12-21 10:21:30 +01:00
parent b2f3f03197
commit a24afdb6bf
1 changed files with 13 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
bookwyrm/views/annual_summary.py
View file

@ -1,4 +1,4 @@
from django.db.models import Case, When
from django.db.models import Case, When, Avg, Sum
from django.shortcuts import get_object_or_404
from django.template.response import TemplateResponse
from django.views import View
@ -14,35 +14,25 @@ class AnnualSummary(View):
user = request.user
read_shelf = get_object_or_404(user.shelf_set, identifier="read")
read_shelf_books_in_year = (
read_book_ids_in_year = (
models.ShelfBook.objects.filter(shelf=read_shelf)
.filter(user=user)
.filter(shelved_date__year=year)
.order_by("shelved_date", "created_date", "updated_date")
.values_list("book", flat=True)
)
read_book_ids_in_year = [i.book.id for i in read_shelf_books_in_year]
preserved = Case(
read_shelf_order = Case(
*[When(pk=pk, then=pos) for pos, pk in enumerate(read_book_ids_in_year)]
)
read_books_in_year = models.Edition.objects.filter(
id__in=read_book_ids_in_year
).order_by(preserved)
).order_by(read_shelf_order)
"""pages stats queries"""
read_books_with_pages = read_books_in_year.filter(pages__gte=0).order_by(
"pages"
)
pages_list = [p.pages for p in read_books_with_pages]
pages_total = 0
pages_average = 0
book_pages_lowest = 0
book_pages_highest = 0
if len(pages_list) > 0:
pages_total = sum(pages_list)
pages_average = round(sum(pages_list) / len(pages_list))
book_pages_lowest = read_books_with_pages.first()
book_pages_highest = read_books_with_pages.last()
page_stats = read_books_in_year.aggregate(Sum("pages"), Avg("pages"))
book_list_by_pages = read_books_in_year.filter(pages__gte=0).order_by("pages")
book_pages_lowest = book_list_by_pages.first()
book_pages_highest = book_list_by_pages.last()
"""books with no pages"""
no_page_list = len(read_books_in_year.filter(pages__exact=None))
@ -55,23 +45,19 @@ class AnnualSummary(View):
.filter(book_id__in=read_book_ids_in_year)
)
best_ratings_books_ids = [review.book.id for review in ratings.filter(rating=5)]
rating_average = 0
if len(ratings) > 0:
ratings_list = [review.rating for review in ratings]
rating_average = round(sum(ratings_list) / len(ratings_list), 2)
ratings_stats = ratings.aggregate(Avg("rating"))
data = {
"year": year,
"books_total": len(read_books_in_year),
"books": read_books_in_year,
"pages_total": pages_total,
"pages_average": pages_average,
"pages_total": page_stats["pages__sum"],
"pages_average": page_stats["pages__avg"],
"book_pages_lowest": book_pages_lowest,
"book_pages_highest": book_pages_highest,
"no_page_number": no_page_list,
"ratings_total": len(ratings),
"rating_average": rating_average,
"rating_average": ratings_stats["rating__avg"],
"book_rating_highest": ratings.order_by("-rating").first(),
"best_ratings_books_ids": best_ratings_books_ids,
}