mirror of
https://github.com/ae-utbm/sith.git
synced 2025-11-02 10:03:05 +00:00
Compare commits
1 Commits
fix-poster
...
trim-galax
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c041aed227 |
@@ -59,7 +59,7 @@ from com.views import (
|
||||
PosterEditBaseView,
|
||||
PosterListBaseView,
|
||||
)
|
||||
from core.auth.mixins import CanEditMixin, PermissionOrClubBoardRequiredMixin
|
||||
from core.auth.mixins import CanEditMixin
|
||||
from core.models import PageRev
|
||||
from core.views import DetailFormView, PageEditViewBase, UseFragmentsMixin
|
||||
from core.views.mixins import FragmentMixin, FragmentRenderer, TabedViewMixin
|
||||
@@ -758,13 +758,11 @@ class MailingAutoGenerationView(View):
|
||||
return redirect("club:mailing", club_id=club.id)
|
||||
|
||||
|
||||
class PosterListView(
|
||||
PermissionOrClubBoardRequiredMixin, ClubTabsMixin, PosterListBaseView
|
||||
):
|
||||
class PosterListView(ClubTabsMixin, PosterListBaseView):
|
||||
"""List communication posters."""
|
||||
|
||||
current_tab = "posters"
|
||||
permission_required = "com.view_poster"
|
||||
extra_context = {"app": "club"}
|
||||
|
||||
def get_queryset(self):
|
||||
return super().get_queryset().filter(club=self.club.id)
|
||||
@@ -772,17 +770,6 @@ class PosterListView(
|
||||
def get_object(self):
|
||||
return self.club
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
return super().get_context_data(**kwargs) | {
|
||||
"create_url": reverse_lazy(
|
||||
"club:poster_create", kwargs={"club_id": self.club.id}
|
||||
),
|
||||
"get_edit_url": lambda poster: reverse(
|
||||
"club:poster_edit",
|
||||
kwargs={"club_id": self.club.id, "poster_id": poster.id},
|
||||
),
|
||||
}
|
||||
|
||||
|
||||
class PosterCreateView(ClubTabsMixin, PosterCreateBaseView):
|
||||
"""Create communication poster."""
|
||||
|
||||
@@ -402,7 +402,9 @@ class Poster(models.Model):
|
||||
groups__id__in=[settings.SITH_GROUP_COM_ADMIN_ID]
|
||||
):
|
||||
Notification.objects.create(
|
||||
user=user, url=reverse("com:poster_list"), type="POSTER_MODERATION"
|
||||
user=user,
|
||||
url=reverse("com:poster_moderate_list"),
|
||||
type="POSTER_MODERATION",
|
||||
)
|
||||
return super().save(*args, **kwargs)
|
||||
|
||||
|
||||
@@ -20,7 +20,33 @@
|
||||
position: absolute;
|
||||
display: flex;
|
||||
bottom: 5px;
|
||||
left: 0;
|
||||
|
||||
&.left {
|
||||
left: 0;
|
||||
}
|
||||
|
||||
&.right {
|
||||
right: 0;
|
||||
}
|
||||
|
||||
.link {
|
||||
padding: 5px;
|
||||
padding-left: 20px;
|
||||
padding-right: 20px;
|
||||
margin-left: 5px;
|
||||
border-radius: 20px;
|
||||
background-color: hsl(40, 100%, 50%);
|
||||
color: black;
|
||||
|
||||
&:hover {
|
||||
color: black;
|
||||
background-color: hsl(40, 58%, 50%);
|
||||
}
|
||||
|
||||
&.delete {
|
||||
background-color: hsl(0, 100%, 40%);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -117,15 +143,43 @@
|
||||
}
|
||||
}
|
||||
|
||||
.actions {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
align-items: stretch;
|
||||
form {
|
||||
margin: unset;
|
||||
padding: unset;
|
||||
button {
|
||||
width: 100%;
|
||||
.edit,
|
||||
.moderate,
|
||||
.slideshow {
|
||||
padding: 5px;
|
||||
border-radius: 20px;
|
||||
background-color: hsl(40, 100%, 50%);
|
||||
color: black;
|
||||
|
||||
&:hover {
|
||||
color: black;
|
||||
background-color: hsl(40, 58%, 50%);
|
||||
}
|
||||
|
||||
&:nth-child(2n) {
|
||||
margin-top: 5px;
|
||||
margin-bottom: 5px;
|
||||
}
|
||||
}
|
||||
|
||||
.tooltip {
|
||||
visibility: hidden;
|
||||
width: 120px;
|
||||
background-color: hsl(210, 20%, 98%);
|
||||
color: hsl(0, 0%, 0%);
|
||||
text-align: center;
|
||||
padding: 5px 0;
|
||||
border-radius: 6px;
|
||||
position: absolute;
|
||||
z-index: 10;
|
||||
|
||||
ul {
|
||||
margin-left: 0;
|
||||
display: inline-block;
|
||||
|
||||
li {
|
||||
display: list-item;
|
||||
list-style-type: none;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -13,53 +13,53 @@
|
||||
|
||||
<div id="title">
|
||||
<h3>{% trans %}Posters{% endtrans %}</h3>
|
||||
<div id="links">
|
||||
<a id="create" class="btn btn-blue" href="{{ create_url }}">
|
||||
<i class="fa fa-plus"></i>
|
||||
{% trans %}Create{% endtrans %}
|
||||
</a>
|
||||
<div id="links" class="right">
|
||||
{% if app == "com" %}
|
||||
<a id="create" class="link" href="{{ url(app + ":poster_create") }}">{% trans %}Create{% endtrans %}</a>
|
||||
<a id="moderation" class="link" href="{{ url("com:poster_moderate_list") }}">{% trans %}Moderation{% endtrans %}</a>
|
||||
{% elif app == "club" %}
|
||||
<a id="create" class="link" href="{{ url(app + ":poster_create", club.id) }}">{% trans %}Create{% endtrans %}</a>
|
||||
{% endif %}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div id="posters">
|
||||
{% for poster in poster_list %}
|
||||
<div class="poster{% if not poster.is_moderated %} not_moderated{% endif %}">
|
||||
<div class="name">{{ poster.name }}</div>
|
||||
<div
|
||||
class="image"
|
||||
hover="{% trans %}Click to expand{% endtrans %}"
|
||||
@click="active = $el.firstElementChild"
|
||||
tooltip="{%- for screen in poster.screens.all() -%}
|
||||
{{ screen }}
|
||||
{% endfor %}"
|
||||
>
|
||||
<img src="{{ poster.file.url }}" alt="{{ poster.name }}">
|
||||
</div>
|
||||
<div class="dates">
|
||||
<div class="begin">{{ poster.date_begin | localtime | date("d/M/Y H:m") }}</div>
|
||||
<div class="end">{{ poster.date_end | localtime | date("d/M/Y H:m") }}</div>
|
||||
</div>
|
||||
<div class="actions">
|
||||
{% if poster.is_editable %}
|
||||
<a class="btn btn-blue" href="{{ get_edit_url(poster) }}">
|
||||
<i class="fa fa-pen-to-square"></i>
|
||||
{% trans %}Edit{% endtrans %}
|
||||
</a>
|
||||
{% endif %}
|
||||
{% if not poster.is_moderated and user.has_perm("com.moderate_poster") %}
|
||||
<form action="{{ url("com:poster_moderate", object_id=poster.id) }}" method="post">
|
||||
{% csrf_token %}
|
||||
<button type="submit" class="btn btn-green">
|
||||
<i class="fa fa-check"></i>
|
||||
{% trans %}Moderate{% endtrans %}
|
||||
</button>
|
||||
</form>
|
||||
{% endif %}
|
||||
</div>
|
||||
</div>
|
||||
{% else %}
|
||||
|
||||
{% if poster_list.count() == 0 %}
|
||||
<div id="no-posters">{% trans %}No posters{% endtrans %}</div>
|
||||
{% endfor %}
|
||||
{% else %}
|
||||
|
||||
{% for poster in poster_list %}
|
||||
<div class="poster{% if not poster.is_moderated %} not_moderated{% endif %}">
|
||||
<div class="name">{{ poster.name }}</div>
|
||||
<div
|
||||
class="image"
|
||||
hover="{% trans %}Click to expand{% endtrans %}"
|
||||
@click="active = $el.firstElementChild"
|
||||
>
|
||||
<img src="{{ poster.file.url }}"></img>
|
||||
</div>
|
||||
<div class="dates">
|
||||
<div class="begin">{{ poster.date_begin | localtime | date("d/M/Y H:m") }}</div>
|
||||
<div class="end">{{ poster.date_end | localtime | date("d/M/Y H:m") }}</div>
|
||||
</div>
|
||||
{% if app == "com" %}
|
||||
<a class="edit" href="{{ url(app + ":poster_edit", poster.id) }}">{% trans %}Edit{% endtrans %}</a>
|
||||
{% elif app == "club" %}
|
||||
<a class="edit" href="{{ url(app + ":poster_edit", club.id, poster.id) }}">{% trans %}Edit{% endtrans %}</a>
|
||||
{% endif %}
|
||||
<div class="tooltip">
|
||||
<ul>
|
||||
{% for screen in poster.screens.all() %}
|
||||
<li>{{ screen }}</li>
|
||||
{% endfor %}
|
||||
</ul>
|
||||
</div>
|
||||
</div>
|
||||
{% endfor %}
|
||||
|
||||
{% endif %}
|
||||
|
||||
</div>
|
||||
|
||||
<div
|
||||
@@ -68,9 +68,7 @@
|
||||
@click="active = null"
|
||||
:class="{active: active !== null}"
|
||||
>
|
||||
<div id="placeholder">
|
||||
<img :src="active?.src" :alt="active?.name">
|
||||
</div>
|
||||
<div id="placeholder"><img :src="active?.src"></div>
|
||||
</div>
|
||||
|
||||
</div>
|
||||
|
||||
43
com/templates/com/poster_moderate.jinja
Normal file
43
com/templates/com/poster_moderate.jinja
Normal file
@@ -0,0 +1,43 @@
|
||||
{% extends "core/base.jinja" %}
|
||||
|
||||
{% block script %}
|
||||
{{ super() }}
|
||||
<script src="{{ static('com/js/poster_list.js') }}"></script>
|
||||
{% endblock %}
|
||||
|
||||
{% block additional_css %}
|
||||
<link rel="stylesheet" href="{{ static('com/css/posters.scss') }}">
|
||||
{% endblock %}
|
||||
|
||||
{% block content %}
|
||||
<div id="poster_list">
|
||||
|
||||
<div id="title">
|
||||
<div id="links" class="left">
|
||||
<a id="list" class="link" href="{{ url("com:poster_list") }}">{% trans %}List{% endtrans %}</a>
|
||||
</div>
|
||||
<h3>{% trans %}Posters - moderation{% endtrans %}</h3>
|
||||
</div>
|
||||
|
||||
<div id="posters">
|
||||
|
||||
{% if object_list.count == 0 %}
|
||||
<div id="no-posters">{% trans %}No objects{% endtrans %}</div>
|
||||
{% else %}
|
||||
|
||||
{% for poster in object_list %}
|
||||
<div class="poster{% if not poster.is_moderated %} not_moderated{% endif %}">
|
||||
<div class="name"> {{ poster.name }} </div>
|
||||
<div class="image"> <img src="{{ poster.file.url }}"></img> </div>
|
||||
<a class="moderate" href="{{ url("com:poster_moderate", object_id=poster.id) }}">Moderate</a>
|
||||
</div>
|
||||
{% endfor %}
|
||||
|
||||
{% endif %}
|
||||
|
||||
</div>
|
||||
|
||||
<div id="view"><div id="placeholder"></div></div>
|
||||
|
||||
</div>
|
||||
{% endblock %}
|
||||
@@ -17,9 +17,7 @@ from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
from django.conf import settings
|
||||
from django.contrib.auth.models import Permission
|
||||
from django.contrib.sites.models import Site
|
||||
from django.core.files.uploadedfile import SimpleUploadedFile
|
||||
from django.test import Client, TestCase
|
||||
from django.urls import reverse
|
||||
from django.utils import html
|
||||
@@ -29,10 +27,9 @@ from model_bakery import baker
|
||||
from pytest_django.asserts import assertNumQueries, assertRedirects
|
||||
|
||||
from club.models import Club, Membership
|
||||
from com.models import News, NewsDate, Poster, Sith, Weekmail, WeekmailArticle
|
||||
from com.models import News, NewsDate, Sith, Weekmail, WeekmailArticle
|
||||
from core.baker_recipes import subscriber_user
|
||||
from core.models import AnonymousUser, Group, User
|
||||
from core.utils import RED_PIXEL_PNG
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
@@ -317,6 +314,7 @@ def test_feed(client: Client):
|
||||
[
|
||||
reverse("com:poster_list"),
|
||||
reverse("com:poster_create"),
|
||||
reverse("com:poster_moderate_list"),
|
||||
],
|
||||
)
|
||||
def test_poster_management_views_crash_test(client: Client, url: str):
|
||||
@@ -327,37 +325,3 @@ def test_poster_management_views_crash_test(client: Client, url: str):
|
||||
client.force_login(user)
|
||||
res = client.get(url)
|
||||
assert res.status_code == 200
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
@pytest.mark.parametrize(
|
||||
"referer",
|
||||
[
|
||||
None,
|
||||
reverse("com:poster_list"),
|
||||
reverse("club:poster_list", kwargs={"club_id": settings.SITH_MAIN_CLUB_ID}),
|
||||
],
|
||||
)
|
||||
def test_moderate_poster(client: Client, referer: str | None):
|
||||
poster = baker.make(
|
||||
Poster,
|
||||
is_moderated=False,
|
||||
file=SimpleUploadedFile("test.png", content=RED_PIXEL_PNG),
|
||||
club_id=settings.SITH_MAIN_CLUB_ID,
|
||||
)
|
||||
user = baker.make(
|
||||
User,
|
||||
user_permissions=Permission.objects.filter(
|
||||
codename__in=["view_poster", "moderate_poster"]
|
||||
),
|
||||
)
|
||||
client.force_login(user)
|
||||
headers = {"REFERER": f"https://{settings.SITH_URL}{referer}"} if referer else {}
|
||||
response = client.post(
|
||||
reverse("com:poster_moderate", kwargs={"object_id": poster.id}), headers=headers
|
||||
)
|
||||
result_url = referer or reverse("com:poster_list")
|
||||
assertRedirects(response, result_url)
|
||||
poster.refresh_from_db()
|
||||
assert poster.is_moderated
|
||||
assert poster.moderator == user
|
||||
|
||||
@@ -33,6 +33,7 @@ from com.views import (
|
||||
PosterDeleteView,
|
||||
PosterEditView,
|
||||
PosterListView,
|
||||
PosterModerateListView,
|
||||
PosterModerateView,
|
||||
ScreenCreateView,
|
||||
ScreenDeleteView,
|
||||
@@ -101,6 +102,11 @@ urlpatterns = [
|
||||
PosterDeleteView.as_view(),
|
||||
name="poster_delete",
|
||||
),
|
||||
path(
|
||||
"poster/moderate/",
|
||||
PosterModerateListView.as_view(),
|
||||
name="poster_moderate_list",
|
||||
),
|
||||
path(
|
||||
"poster/<int:object_id>/moderate/",
|
||||
PosterModerateView.as_view(),
|
||||
|
||||
67
com/views.py
67
com/views.py
@@ -25,7 +25,6 @@ import itertools
|
||||
from datetime import date, timedelta
|
||||
from smtplib import SMTPRecipientsRefused
|
||||
from typing import Any
|
||||
from urllib.parse import urlparse
|
||||
|
||||
from dateutil.relativedelta import relativedelta
|
||||
from django.conf import settings
|
||||
@@ -35,7 +34,7 @@ from django.contrib.auth.mixins import (
|
||||
)
|
||||
from django.contrib.syndication.views import Feed
|
||||
from django.core.exceptions import PermissionDenied, ValidationError
|
||||
from django.db.models import Exists, Max, OuterRef, Value
|
||||
from django.db.models import Max
|
||||
from django.forms.models import modelform_factory
|
||||
from django.http import HttpResponseRedirect
|
||||
from django.shortcuts import get_object_or_404, redirect
|
||||
@@ -46,7 +45,7 @@ from django.utils.translation import gettext_lazy as _
|
||||
from django.views.generic import DetailView, ListView, TemplateView, View
|
||||
from django.views.generic.edit import CreateView, DeleteView, UpdateView
|
||||
|
||||
from club.models import Club, Mailing, Membership
|
||||
from club.models import Club, Mailing
|
||||
from com.forms import NewsDateForm, NewsForm, PosterForm
|
||||
from com.ics_calendar import IcsCalendar
|
||||
from com.models import News, NewsDate, Poster, Screen, Sith, Weekmail, WeekmailArticle
|
||||
@@ -562,26 +561,16 @@ class MailingModerateView(View):
|
||||
raise PermissionDenied
|
||||
|
||||
|
||||
class PosterListBaseView(ListView):
|
||||
class PosterListBaseView(PermissionOrClubBoardRequiredMixin, ListView):
|
||||
"""List communication posters."""
|
||||
|
||||
model = Poster
|
||||
template_name = "com/poster_list.jinja"
|
||||
permission_required = "com.view_poster"
|
||||
ordering = ["-date_begin"]
|
||||
|
||||
def get_queryset(self):
|
||||
qs = Poster.objects.prefetch_related("screens")
|
||||
if self.request.user.has_perm("com.edit_poster"):
|
||||
qs = qs.annotate(is_editable=Value(value=True))
|
||||
else:
|
||||
qs = qs.annotate(
|
||||
is_editable=Exists(
|
||||
Membership.objects.ongoing()
|
||||
.board()
|
||||
.filter(user=self.request.user, club=OuterRef("club_id"))
|
||||
)
|
||||
)
|
||||
return qs.order_by("-date_begin")
|
||||
def get_context_data(self, **kwargs):
|
||||
return super().get_context_data(**kwargs) | {"club": self.club}
|
||||
|
||||
|
||||
class PosterCreateBaseView(PermissionOrClubBoardRequiredMixin, CreateView):
|
||||
@@ -644,17 +633,21 @@ class PosterDeleteBaseView(
|
||||
permission_required = "com.delete_poster"
|
||||
|
||||
|
||||
class PosterListView(PermissionRequiredMixin, ComTabsMixin, PosterListBaseView):
|
||||
class PosterListView(ComTabsMixin, PosterListBaseView):
|
||||
"""List communication posters."""
|
||||
|
||||
current_tab = "posters"
|
||||
extra_context = {
|
||||
"create_url": reverse_lazy("com:poster_create"),
|
||||
"get_edit_url": lambda poster: reverse(
|
||||
"com:poster_edit", kwargs={"poster_id": poster.id}
|
||||
),
|
||||
}
|
||||
permission_required = "com.view_poster"
|
||||
|
||||
def get_queryset(self):
|
||||
qs = super().get_queryset()
|
||||
if self.request.user.has_perm("com.view_poster"):
|
||||
return qs
|
||||
return qs.filter(club=self.club.id)
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
kwargs = super().get_context_data(**kwargs)
|
||||
kwargs["app"] = "com"
|
||||
return kwargs
|
||||
|
||||
|
||||
class PosterCreateView(ComTabsMixin, PosterCreateBaseView):
|
||||
@@ -679,6 +672,17 @@ class PosterDeleteView(PosterDeleteBaseView):
|
||||
success_url = reverse_lazy("com:poster_list")
|
||||
|
||||
|
||||
class PosterModerateListView(PermissionRequiredMixin, ComTabsMixin, ListView):
|
||||
"""Moderate list communication poster."""
|
||||
|
||||
current_tab = "posters"
|
||||
model = Poster
|
||||
template_name = "com/poster_moderate.jinja"
|
||||
queryset = Poster.objects.filter(is_moderated=False).all()
|
||||
permission_required = "com.moderate_poster"
|
||||
extra_context = {"app": "com"}
|
||||
|
||||
|
||||
class PosterModerateView(PermissionRequiredMixin, ComTabsMixin, View):
|
||||
"""Moderate communication poster."""
|
||||
|
||||
@@ -686,21 +690,12 @@ class PosterModerateView(PermissionRequiredMixin, ComTabsMixin, View):
|
||||
permission_required = "com.moderate_poster"
|
||||
extra_context = {"app": "com"}
|
||||
|
||||
def post(self, request, *args, **kwargs):
|
||||
def get(self, request, *args, **kwargs):
|
||||
obj = get_object_or_404(Poster, pk=kwargs["object_id"])
|
||||
obj.is_moderated = True
|
||||
obj.moderator = request.user
|
||||
obj.save()
|
||||
# The moderation request may be originated from a club context (/club/poster)
|
||||
# or a global context (/com/poster),
|
||||
# so the redirection URL will be the URL of the page that called this view,
|
||||
# as long as the latter belongs to the sith.
|
||||
referer = self.request.META.get("HTTP_REFERER")
|
||||
if referer:
|
||||
parsed = urlparse(referer)
|
||||
if parsed.netloc == settings.SITH_URL:
|
||||
return redirect(parsed.path)
|
||||
return redirect(reverse("com:poster_list"))
|
||||
return redirect("com:poster_moderate_list")
|
||||
|
||||
|
||||
class ScreenListView(PermissionRequiredMixin, ComTabsMixin, ListView):
|
||||
|
||||
@@ -45,8 +45,9 @@ class Command(BaseCommand):
|
||||
"verbosity level should be between 0 and 2 included", stacklevel=2
|
||||
)
|
||||
|
||||
if options["verbosity"] == 2:
|
||||
if options["verbosity"] >= 2:
|
||||
logger.setLevel(logging.DEBUG)
|
||||
logging.getLogger("django.db.backends").setLevel(logging.DEBUG)
|
||||
elif options["verbosity"] == 1:
|
||||
logger.setLevel(logging.INFO)
|
||||
else:
|
||||
@@ -59,6 +60,3 @@ class Command(BaseCommand):
|
||||
Galaxy.objects.filter(state__isnull=True).delete()
|
||||
|
||||
logger.info("Ruled the galaxy in {} queries.".format(len(connection.queries)))
|
||||
if options["verbosity"] > 2:
|
||||
for q in connection.queries:
|
||||
logger.debug(q)
|
||||
|
||||
@@ -31,13 +31,14 @@ from collections import defaultdict
|
||||
from typing import NamedTuple, TypedDict
|
||||
|
||||
from django.db import models
|
||||
from django.db.models import Count, F, Q, QuerySet
|
||||
from django.utils.timezone import localdate
|
||||
from django.db.models import Count, Exists, F, OuterRef, Q, QuerySet
|
||||
from django.utils.timezone import localdate, now
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
|
||||
from club.models import Membership
|
||||
from core.models import User
|
||||
from sas.models import PeoplePictureRelation, Picture
|
||||
from subscription.models import Subscription
|
||||
|
||||
|
||||
class GalaxyStar(models.Model):
|
||||
@@ -198,8 +199,16 @@ class Galaxy(models.Model):
|
||||
cls, picture_count_threshold: int = DEFAULT_PICTURE_COUNT_THRESHOLD
|
||||
) -> QuerySet[User]:
|
||||
return (
|
||||
User.objects.exclude(subscriptions=None)
|
||||
.annotate(pictures_count=Count("pictures"))
|
||||
User.objects.filter(is_subscriber_viewable=True)
|
||||
.exclude(subscriptions=None)
|
||||
.annotate(
|
||||
pictures_count=Count("pictures"),
|
||||
is_active_in_galaxy=Exists(
|
||||
Subscription.objects.filter(
|
||||
member=OuterRef("id"), subscription_end__gt=now()
|
||||
)
|
||||
),
|
||||
)
|
||||
.filter(pictures_count__gt=picture_count_threshold)
|
||||
.distinct()
|
||||
)
|
||||
@@ -290,9 +299,9 @@ class Galaxy(models.Model):
|
||||
31/12/2022 (also two years, but with an offset of one year), then their
|
||||
club score is 365.
|
||||
"""
|
||||
memberships = user.memberships.only("start_date", "end_date", "club_id")
|
||||
memberships = user.memberships.values("start_date", "end_date", "club_id")
|
||||
result = defaultdict(int)
|
||||
now = localdate()
|
||||
today = localdate()
|
||||
for membership in memberships:
|
||||
# This is a N+1 query, but 92% of galaxy users have less than 10 memberships.
|
||||
# Only 5 users have more than 30 memberships.
|
||||
@@ -300,23 +309,23 @@ class Galaxy(models.Model):
|
||||
Membership.objects.exclude(user=user)
|
||||
.filter(
|
||||
Q( # start2 <= start1 <= end2
|
||||
start_date__lte=membership.start_date,
|
||||
end_date__gte=membership.start_date,
|
||||
start_date__lte=membership["start_date"],
|
||||
end_date__gte=membership["start_date"],
|
||||
)
|
||||
| Q( # start2 <= start1 <= now
|
||||
start_date__lte=membership.start_date, end_date=None
|
||||
| Q( # start2 <= start1 <= today
|
||||
start_date__lte=membership["start_date"], end_date=None
|
||||
)
|
||||
| Q( # start1 <= start2 <= end2
|
||||
start_date__gte=membership.start_date,
|
||||
start_date__lte=membership.end_date or now,
|
||||
start_date__gte=membership["start_date"],
|
||||
start_date__lte=membership["end_date"] or today,
|
||||
),
|
||||
club_id=membership.club_id,
|
||||
club_id=membership["club_id"],
|
||||
)
|
||||
.only("start_date", "end_date", "user_id")
|
||||
)
|
||||
for other in common_memberships:
|
||||
start = max(membership.start_date, other.start_date)
|
||||
end = min(membership.end_date or now, other.end_date or now)
|
||||
start = max(membership["start_date"], other.start_date)
|
||||
end = min(membership["end_date"] or today, other.end_date or today)
|
||||
result[other.user_id] += (end - start).days * cls.CLUBS_POINTS
|
||||
return result
|
||||
|
||||
@@ -382,18 +391,22 @@ class Galaxy(models.Model):
|
||||
# this is memory expensive but prevents a lot of db hits, therefore
|
||||
# is far more time efficient
|
||||
|
||||
rulable_users = list(self.get_rulable_users(picture_count_threshold))
|
||||
rulable_users_count = len(rulable_users)
|
||||
rulable_users_qs = self.get_rulable_users(picture_count_threshold)
|
||||
active_users_count = rulable_users_qs.filter(is_active_in_galaxy=True).count()
|
||||
rulable_users = list(rulable_users_qs)
|
||||
user1_count = 0
|
||||
self.logger.info(
|
||||
f"{rulable_users_count} citizen have been listed. Starting to rule."
|
||||
f" {len(rulable_users)} citizens (with {active_users_count} active ones) "
|
||||
f"have been listed. Starting to rule."
|
||||
)
|
||||
|
||||
self.logger.info("Creating stars for all citizen")
|
||||
individual_scores = self.compute_individual_scores()
|
||||
GalaxyStar.objects.bulk_create(
|
||||
[
|
||||
GalaxyStar(owner=user, galaxy=self, mass=individual_scores[user.id])
|
||||
GalaxyStar(
|
||||
owner_id=user.id, galaxy=self, mass=individual_scores[user.id]
|
||||
)
|
||||
for user in rulable_users
|
||||
]
|
||||
)
|
||||
@@ -405,9 +418,9 @@ class Galaxy(models.Model):
|
||||
t_global_start = time.time()
|
||||
while len(rulable_users) > 0:
|
||||
user1 = rulable_users.pop()
|
||||
if not user1.is_active_in_galaxy:
|
||||
continue
|
||||
user1_count += 1
|
||||
rulable_users_count2 = len(rulable_users)
|
||||
|
||||
star1 = stars[user1.id]
|
||||
|
||||
lanes = []
|
||||
@@ -448,17 +461,20 @@ class Galaxy(models.Model):
|
||||
self.logger.info("")
|
||||
self.logger.info(f" Ruling of {self} ".center(60, "#"))
|
||||
self.logger.info(
|
||||
f"Progression: {user1_count}/{rulable_users_count} "
|
||||
f"citizen -- {rulable_users_count - user1_count} remaining"
|
||||
f"Progression: {user1_count}/{active_users_count} "
|
||||
f"citizen -- {active_users_count - user1_count} remaining"
|
||||
)
|
||||
self.logger.info(f"Speed: {global_avg_speed:.2f} citizen per second")
|
||||
eta = rulable_users_count2 // global_avg_speed
|
||||
eta = len(rulable_users) // global_avg_speed
|
||||
self.logger.info(
|
||||
f"ETA: {int(eta // 60 % 60)} minutes {int(eta % 60)} seconds"
|
||||
)
|
||||
self.logger.info("#" * 60)
|
||||
t_global_start = time.time()
|
||||
|
||||
count, _ = self.stars.filter(Q(lanes1=None) & Q(lanes2=None)).delete()
|
||||
self.logger.info(f"{count} orphan stars have been trimmed.")
|
||||
|
||||
# Here, we get the IDs of the old galaxies that we'll need to delete. In normal operation, only one galaxy
|
||||
# should be returned, and we can't delete it yet, as it's the one still displayed by the Sith.
|
||||
old_galaxies_pks = list(
|
||||
|
||||
@@ -122,7 +122,7 @@ class TestGalaxyModel(TestCase):
|
||||
self.com,
|
||||
]
|
||||
|
||||
with self.assertNumQueries(44):
|
||||
with self.assertNumQueries(38):
|
||||
while len(users) > 0:
|
||||
user1 = users.pop(0)
|
||||
family_scores = Galaxy.compute_user_family_score(user1)
|
||||
@@ -150,7 +150,7 @@ class TestGalaxyModel(TestCase):
|
||||
that the number of queries to rule the galaxy is stable.
|
||||
"""
|
||||
galaxy = Galaxy.objects.create()
|
||||
with self.assertNumQueries(39):
|
||||
with self.assertNumQueries(36):
|
||||
galaxy.rule(0) # We want everybody here
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user