lib.itmens/common/views.py

99 lines
3.3 KiB
Python
Raw Normal View History

2020-05-01 22:46:15 +08:00
from django.shortcuts import render
from django.contrib.auth.decorators import login_required
from books.models import Book
from common.models import MarkStatusEnum
2020-07-03 15:36:23 +08:00
from common.utils import PageLinksGenerator
2020-05-01 22:46:15 +08:00
from users.models import Report, User
2020-05-05 23:50:48 +08:00
from django.core.paginator import Paginator
2020-07-10 21:28:09 +08:00
from django.db.models import Q, Count
2020-05-01 22:46:15 +08:00
from django.http import HttpResponseBadRequest
# how many books have in each set at the home page
BOOKS_PER_SET = 5
# how many items are showed in one search result page
ITEMS_PER_PAGE = 20
2020-07-03 15:36:23 +08:00
# how many pages links in the pagination
PAGE_LINK_NUMBER = 7
2020-07-10 21:28:09 +08:00
# max tags on list page
TAG_NUMBER_ON_LIST = 5
2020-05-01 22:46:15 +08:00
@login_required
def home(request):
if request.method == 'GET':
2020-05-05 23:50:48 +08:00
do_book_marks = request.user.user_bookmarks.filter(status=MarkStatusEnum.DO)
do_books_more = True if do_book_marks.count() > BOOKS_PER_SET else False
wish_book_marks = request.user.user_bookmarks.filter(status=MarkStatusEnum.WISH)
wish_books_more = True if wish_book_marks.count() > BOOKS_PER_SET else False
collect_book_marks = request.user.user_bookmarks.filter(status=MarkStatusEnum.COLLECT)
collect_books_more = True if collect_book_marks.count() > BOOKS_PER_SET else False
2020-05-01 22:46:15 +08:00
reports = Report.objects.order_by('-submitted_time').filter(is_read=False)
# reports = Report.objects.latest('submitted_time').filter(is_read=False)
return render(
request,
'common/home.html',
{
2020-05-05 23:50:48 +08:00
'do_book_marks': do_book_marks[:BOOKS_PER_SET],
'wish_book_marks': wish_book_marks[:BOOKS_PER_SET],
'collect_book_marks': collect_book_marks[:BOOKS_PER_SET],
2020-05-01 22:46:15 +08:00
'do_books_more': do_books_more,
'wish_books_more': wish_books_more,
'collect_books_more': collect_books_more,
'reports': reports,
}
)
else:
return HttpResponseBadRequest()
2020-05-05 23:50:48 +08:00
@login_required
2020-05-01 22:46:15 +08:00
def search(request):
if request.method == 'GET':
# in the future when more modules are added...
# category = request.GET.get("category")
q = Q()
query_args = []
2020-07-10 21:28:09 +08:00
# keywords
keywords = request.GET.get("q", default='').split()
2020-05-01 22:46:15 +08:00
for keyword in keywords:
q = q | Q(title__icontains=keyword)
q = q | Q(subtitle__istartswith=keyword)
q = q | Q(orig_title__icontains=keyword)
2020-07-10 21:28:09 +08:00
# tag
tag = request.GET.get("tag", default='')
if tag:
q = q | Q(book_tags__content__iexact=tag)
2020-05-01 22:46:15 +08:00
query_args.append(q)
2020-07-10 21:28:09 +08:00
queryset = Book.objects.filter(*query_args).distinct()
2020-05-01 22:46:15 +08:00
paginator = Paginator(queryset, ITEMS_PER_PAGE)
page_number = request.GET.get('page', default=1)
items = paginator.get_page(page_number)
2020-07-03 15:36:23 +08:00
items.pagination = PageLinksGenerator(PAGE_LINK_NUMBER, page_number, paginator.num_pages)
2020-07-10 21:28:09 +08:00
for item in items:
item.tag_list = item.get_tags_manager().values('content').annotate(
tag_frequency=Count('content')).order_by('-tag_frequency')[:TAG_NUMBER_ON_LIST]
2020-05-01 22:46:15 +08:00
return render(
request,
"common/search_result.html",
{
"items": items,
}
)
else:
2020-07-10 21:28:09 +08:00
return HttpResponseBadRequest()