Sith/core/views/site.py

131 lines
4.0 KiB
Python
Raw Normal View History

#
# Copyright 2016,2017
# - Skia <skia@libskia.so>
# - Sli <antoine@bartuccio.fr>
#
# Ce fichier fait partie du site de l'Association des Étudiants de l'UTBM,
# http://ae.utbm.fr.
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License a published by the Free Software
# Foundation; either version 3 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
2024-08-10 12:23:01 +00:00
# this program; if not, write to the Free Software Foundation, Inc., 59 Temple
# Place - Suite 330, Boston, MA 02111-1307, USA.
#
#
2024-06-24 11:07:36 +00:00
import json
from django.conf import settings
2016-08-19 00:53:44 +00:00
from django.contrib.auth.decorators import login_required
2024-06-24 11:07:36 +00:00
from django.core import serializers
from django.db.models.query import QuerySet
from django.http import JsonResponse
from django.shortcuts import redirect, render
from django.utils import html
from django.utils.text import slugify
2024-08-10 12:23:01 +00:00
from django.views.generic import ListView
2016-12-19 20:00:09 +00:00
from haystack.query import SearchQuerySet
2016-08-19 00:53:44 +00:00
from club.models import Club
2024-06-24 11:07:36 +00:00
from core.models import Notification, User
2016-08-10 03:48:06 +00:00
2017-06-12 07:42:03 +00:00
2015-11-24 15:09:46 +00:00
def index(request, context=None):
from com.views import NewsListView
2018-10-04 19:29:19 +00:00
return NewsListView.as_view()(request)
2016-08-19 00:53:44 +00:00
2017-06-12 07:42:03 +00:00
2016-12-09 23:06:17 +00:00
class NotificationList(ListView):
model = Notification
template_name = "core/notification_list.jinja"
def get_queryset(self) -> QuerySet[Notification]:
if self.request.user.is_anonymous:
return Notification.objects.none()
# TODO: Bulk update in django 2.2
2018-10-04 19:29:19 +00:00
if "see_all" in self.request.GET.keys():
for n in self.request.user.notifications.filter(viewed=False):
n.viewed = True
n.save()
2018-10-04 19:29:19 +00:00
return self.request.user.notifications.order_by("-date")[:20]
2016-12-09 23:06:17 +00:00
2017-06-12 07:42:03 +00:00
2016-12-08 18:47:28 +00:00
def notification(request, notif_id):
notif = Notification.objects.filter(id=notif_id).first()
if notif:
if notif.type not in settings.SITH_PERMANENT_NOTIFICATIONS:
notif.viewed = True
else:
notif.callback()
notif.save()
2016-12-08 18:47:28 +00:00
return redirect(notif.url)
return redirect("/")
2017-06-12 07:42:03 +00:00
2024-06-27 12:57:40 +00:00
def search_user(query):
try:
# slugify turns everything into ascii and every whitespace into -
2020-04-21 13:36:13 +00:00
# it ends by removing duplicate - (so ' - ' will turn into '-')
# replace('-', ' ') because search is whitespace based
query = slugify(query).replace("-", " ")
2020-04-21 13:36:13 +00:00
# TODO: is this necessary?
query = html.escape(query)
2020-04-21 13:36:13 +00:00
res = (
SearchQuerySet()
.models(User)
.autocomplete(auto=query)
2024-09-29 21:01:55 +00:00
.order_by("-last_login")
.load_all()[:20]
2020-04-21 13:36:13 +00:00
)
return [r.object for r in res]
except TypeError:
return []
2017-06-12 07:42:03 +00:00
def search_club(query, *, as_json=False):
clubs = []
if query:
clubs = Club.objects.filter(name__icontains=query).all()
clubs = clubs[:5]
if as_json:
# Re-loads json to avoid double encoding by JsonResponse, but still benefit from serializers
2018-10-04 19:29:19 +00:00
clubs = json.loads(serializers.serialize("json", clubs, fields=("name")))
else:
2016-08-19 01:12:20 +00:00
clubs = list(clubs)
return clubs
2016-08-19 00:53:44 +00:00
2017-06-12 07:42:03 +00:00
2016-08-19 00:53:44 +00:00
@login_required
def search_view(request):
result = {
2018-10-04 19:29:19 +00:00
"users": search_user(request.GET.get("query", "")),
"clubs": search_club(request.GET.get("query", "")),
2017-06-12 07:42:03 +00:00
}
2018-10-04 19:29:19 +00:00
return render(request, "core/search.jinja", context={"result": result})
2017-06-12 07:42:03 +00:00
@login_required
def search_user_json(request):
2024-06-27 12:57:40 +00:00
result = {"users": search_user(request.GET.get("query", ""))}
return JsonResponse(result)
2016-08-19 00:53:44 +00:00
2017-06-12 07:42:03 +00:00
2016-08-19 00:53:44 +00:00
@login_required
def search_json(request):
result = {
2024-06-27 12:57:40 +00:00
"users": search_user(request.GET.get("query", "")),
"clubs": search_club(request.GET.get("query", ""), as_json=True),
2017-06-12 07:42:03 +00:00
}
return JsonResponse(result)