Galaxy (#575)

Co-authored-by: Skia <florent.jacquet@eshard.com>

galaxy/views.py

@@ -0,0 +1,104 @@
+# -*- coding:utf-8 -*
+#
+# Copyright 2023
+# - Skia <skia@hya.sk>
+#
+# 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
+# this program; if not, write to the Free Sofware Foundation, Inc., 59 Temple
+# Place - Suite 330, Boston, MA 02111-1307, USA.
+#
+#
+
+from django.views.generic import DetailView, View
+from django.http import JsonResponse, Http404
+from django.db.models import Q, Case, F, When, Value
+from django.db.models.functions import Concat
+from django.utils.translation import gettext_lazy as _
+
+from core.views import (
+    CanViewMixin,
+    FormerSubscriberMixin,
+)
+from core.models import User
+from core.views import UserTabsMixin
+from galaxy.models import Galaxy, GalaxyLane
+
+
+class GalaxyUserView(CanViewMixin, UserTabsMixin, DetailView):
+    model = User
+    pk_url_kwarg = "user_id"
+    template_name = "galaxy/user.jinja"
+    current_tab = "galaxy"
+
+    def get_object(self, *args, **kwargs):
+        user: User = super(GalaxyUserView, self).get_object(*args, **kwargs)
+        if not hasattr(user, "galaxy_user"):
+            raise Http404(_("This citizen has not yet joined the galaxy"))
+        return user
+
+    def get_queryset(self):
+        return super(GalaxyUserView, self).get_queryset().select_related("galaxy_user")
+
+    def get_context_data(self, **kwargs):
+        kwargs = super(GalaxyUserView, self).get_context_data(**kwargs)
+        kwargs["lanes"] = (
+            GalaxyLane.objects.filter(
+                Q(star1=self.object.galaxy_user) | Q(star2=self.object.galaxy_user)
+            )
+            .order_by("distance")
+            .annotate(
+                other_star_id=Case(
+                    When(star1=self.object.galaxy_user, then=F("star2__owner__id")),
+                    default=F("star1__owner__id"),
+                ),
+                other_star_mass=Case(
+                    When(star1=self.object.galaxy_user, then=F("star2__mass")),
+                    default=F("star1__mass"),
+                ),
+                other_star_name=Case(
+                    When(
+                        star1=self.object.galaxy_user,
+                        then=Case(
+                            When(
+                                star2__owner__nick_name=None,
+                                then=Concat(
+                                    F("star2__owner__first_name"),
+                                    Value(" "),
+                                    F("star2__owner__last_name"),
+                                ),
+                            )
+                        ),
+                    ),
+                    default=Case(
+                        When(
+                            star1__owner__nick_name=None,
+                            then=Concat(
+                                F("star1__owner__first_name"),
+                                Value(" "),
+                                F("star1__owner__last_name"),
+                            ),
+                        )
+                    ),
+                ),
+            )
+            .select_related("star1", "star2")
+        )
+        return kwargs
+
+
+class GalaxyDataView(FormerSubscriberMixin, View):
+    def get(self, request, *args, **kwargs):
+        return JsonResponse(Galaxy.objects.first().state)