Family views

apps/family/views.py

@@ -13,7 +13,7 @@ from django.urls import reverse_lazy
 
 from .models import Family, Challenge, FamilyMembership, User
 from .tables import FamilyTable, ChallengeTable, FamilyMembershipTable
-from .forms import ChallengeUpdateForm
+from .forms import ChallengeUpdateForm, FamilyMembershipForm
 
 
 class FamilyCreateView(ProtectQuerysetMixin, ProtectedCreateView):
@@ -49,6 +49,35 @@ class FamilyDetailView(ProtectQuerysetMixin, LoginRequiredMixin, DetailView):
     context_object_name = "family"
     extra_context = {"title": _('Family detail')}
 
+    def get_context_data(self, **kwargs):
+        """
+        Add members list
+        """
+        context = super().get_context_data(**kwargs)
+
+        family = self.object
+
+        # member list
+        family_member = FamilyMembership.objects.filter(
+            family=family,
+            year=date.today().year,
+        ).filter(PermissionBackend.filter_queryset(self.request, FamilyMembership, "view"))\
+            .order_by("user__username").distinct("user__username")
+
+        membership_table = FamilyMembershipTable(data=family_member)
+        context['member_list'] = membership_table
+
+        # Check if the user has the right to create a membership, to display the button.
+        empty_membership = FamilyMembership(
+            family=family,
+            user=User.objects.first(),
+            year=date.today().year,
+        )
+        context["can_add_members"] = PermissionBackend()\
+            .has_perm(self.request.user, "family.add_membership", empty_membership)
+
+        return context
+
 
 class FamilyUpdateView(ProtectQuerysetMixin, LoginRequiredMixin, UpdateView):
     """
@@ -59,6 +88,30 @@ class FamilyUpdateView(ProtectQuerysetMixin, LoginRequiredMixin, UpdateView):
     extra_context = {"title": _('Update family')}
 
 
+class FamilyAddMemberView(ProtectQuerysetMixin, ProtectedCreateView):
+    """
+    Add a membership to a family
+    """
+    model = FamilyMembership
+    form_class = FamilyMembershipForm
+    template_name = 'family/add_member.html'
+    extra_context = {"title": _("Add a new member to the family")}
+
+    def get_sample_object(self):
+        if "family_pk" in self.kwargs:
+            family = Family.objects.get(pk=self.kwargs["family_pk"])
+        else:
+            family = FamilyMembership.objects.get(pk=self.kwargs["pk"]).family
+        return FamilyMembership(
+            user=self.request.user,
+            family=family,
+            year=date.today().year,
+        )
+
+    def get_success_url(self):
+        return reverse_lazy('family:family_detail', kwargs={'pk': self.object.family.id})
+
+
 class ChallengeCreateView(ProtectQuerysetMixin, ProtectedCreateView):
     """
     Create challenge
@@ -72,7 +125,7 @@ class ChallengeCreateView(ProtectQuerysetMixin, ProtectedCreateView):
             description="Sample challenge",
             points=0,
         )
-    
+
     def get_success_url(self):
         return reverse_lazy('family:challenge_list')
 
@@ -103,7 +156,7 @@ class ChallengeDetailView(ProtectQuerysetMixin, LoginRequiredMixin, DetailView):
         context["fields"] = [(
             Challenge._meta.get_field(field).verbose_name.capitalize(),
             value) for field, value in fields.items()]
-        context["obtained"] = getattr(self.object, "obtained")
+        context["obtained"] = self.object.obtained
         context["update"] = PermissionBackend.check_perm(self.request, "family.change_challenge")
 
         return context
@@ -121,4 +174,4 @@ class ChallengeUpdateView(ProtectQuerysetMixin, LoginRequiredMixin, UpdateView):
 
     def get_success_url(self, **kwargs):
         self.object.refresh_from_db()
-        return reverse_lazy('family:challenge_detail', kwargs={'pk': self.object.pk})
+        return reverse_lazy('family:challenge_detail', kwargs={'pk': self.object.pk})