From 021410e31629001a71fe5dbfcbe456900d2f249d Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@usody.com>
Date: Tue, 17 Oct 2023 10:53:03 +0200
Subject: [PATCH] edit and delete memberships

---
 idhub/admin/views.py                       | 49 +++++++++++++++++++++-
 idhub/models.py                            |  3 ++
 idhub/templates/idhub/admin_user.html      | 25 +++++++++++
 idhub/templates/idhub/admin_user_edit.html | 32 ++++++++++++++
 idhub/urls.py                              |  4 ++
 5 files changed, 111 insertions(+), 2 deletions(-)

diff --git a/idhub/admin/views.py b/idhub/admin/views.py
index edda398..05cc8b3 100644
--- a/idhub/admin/views.py
+++ b/idhub/admin/views.py
@@ -125,7 +125,6 @@ class AdminPeopleRegisterView(People, CreateView):
     success_url = reverse_lazy('idhub:admin_people_list')
 
     def get_success_url(self):
-        # import pdb; pdb.set_trace()
         self.success_url = reverse_lazy(
             'idhub:admin_people_membership_new',
             kwargs={"pk": self.object.id}
@@ -162,7 +161,53 @@ class AdminPeopleMembershipRegisterView(People, CreateView):
         self.object = self.model(user=self.user)
         kwargs = super().get_form_kwargs()
         return kwargs
-        
+
+    def get_success_url(self):
+        self.success_url = reverse_lazy(
+            'idhub:admin_people_edit',
+            kwargs={"pk": self.user.id}
+        )
+        return self.success_url
+
+
+class AdminPeopleMembershipEditView(People, CreateView):
+    template_name = "idhub/admin_people_membership_register.html"
+    subtitle = _('People add membership')
+    icon = 'bi bi-person'
+    model = Membership
+    from_class = MembershipForm
+    fields = ('type', 'start_date', 'end_date')
+    success_url = reverse_lazy('idhub:admin_people_list')
+
+    def get_form(self):
+        form = super().get_form()
+        form.fields['start_date'].widget.input_type = 'date'
+        form.fields['end_date'].widget.input_type = 'date'
+        return form
+
+    def get_form_kwargs(self):
+        pk = self.kwargs.get('pk')
+        if pk:
+            self.object = get_object_or_404(self.model, pk=pk)
+        kwargs = super().get_form_kwargs()
+        return kwargs
+
+
+class AdminPeopleMembershipDeleteView(AdminPeopleView):
+    model = Membership
+
+    def get(self, request, *args, **kwargs):
+        self.pk = kwargs['pk']
+        self.object = get_object_or_404(self.model, pk=self.pk)
+
+        if self.object != self.request.user:
+            user = self.object.user
+            self.object.delete()
+        else:
+            messages.error(self.request, _('Is not possible delete your account!'))
+
+        return redirect('idhub:admin_people_edit', user.id)
+
 
 class AdminRolesView(AccessControl):
     template_name = "idhub/admin_roles.html"
diff --git a/idhub/models.py b/idhub/models.py
index 0db78b7..ba5d953 100644
--- a/idhub/models.py
+++ b/idhub/models.py
@@ -77,3 +77,6 @@ class Membership(models.Model):
         on_delete=models.CASCADE,
         related_name='memberships',
     )
+
+    def get_type(self):
+        return dict(self.Types.choices).get(self.type)
diff --git a/idhub/templates/idhub/admin_user.html b/idhub/templates/idhub/admin_user.html
index 92ee4e1..7706a92 100644
--- a/idhub/templates/idhub/admin_user.html
+++ b/idhub/templates/idhub/admin_user.html
@@ -47,6 +47,31 @@
     </div>
   </div>
 </div>
+<div class="row">
+  <div class="col">
+    <div class="table-responsive">
+      <table class="table table-striped table-sm">
+        <thead>
+          <tr>
+            <th scope="col"><button type="button" class="btn btn-grey border border-dark">{% trans 'Membership' %}</button></th>
+            <th scope="col"><button type="button" class="btn btn-grey border border-dark">{% trans 'From' %}</button></th>
+            <th scope="col"><button type="button" class="btn btn-grey border border-dark">{% trans 'To' %}</button></th>
+          </tr>
+        </thead>
+        <tbody>
+          {% for membership in object.memberships.all %}
+          <tr>
+            <td>{{ membership.get_type }}</td>
+            <td>{{ membership.start_date|default:'' }}</td>
+            <td>{{ membership.end_date|default:'' }}</td>
+          </tr>
+          {% endfor %}
+        </tbody>
+      </table>
+    </div>
+  </div>
+</div>
+
 <!-- Modal -->
 <div class="modal" id="confirm-delete" tabindex="-1" aria-labelledby="exampleModalLabel" aria-hidden="true">
   <div class="modal-dialog">
diff --git a/idhub/templates/idhub/admin_user_edit.html b/idhub/templates/idhub/admin_user_edit.html
index f599d81..1ec6adf 100644
--- a/idhub/templates/idhub/admin_user_edit.html
+++ b/idhub/templates/idhub/admin_user_edit.html
@@ -29,4 +29,36 @@
 </div>
 
 </form>
+
+<div class="row mt-5">
+  <div class="col">
+    <div class="table-responsive">
+      <table class="table table-striped table-sm">
+        <thead>
+          <tr>
+            <th scope="col"><button type="button" class="btn btn-grey border border-dark">{% trans 'Membership' %}</button></th>
+            <th scope="col"><button type="button" class="btn btn-grey border border-dark">{% trans 'From' %}</button></th>
+            <th scope="col"><button type="button" class="btn btn-grey border border-dark">{% trans 'To' %}</button></th>
+            <th scope="col"></th>
+            <th scope="col"></th>
+          </tr>
+        </thead>
+        <tbody>
+          {% for membership in object.memberships.all %}
+          <tr>
+            <td>{{ membership.get_type }}</td>
+            <td>{{ membership.start_date|default:'' }}</td>
+            <td>{{ membership.end_date|default:'' }}</td>
+            <td><a href="{% url 'idhub:admin_people_membership_edit' membership.id %}" title="{% trans 'Edit' %}"><i class="bi bi-pencil-square"></i></a></td>
+            <td><a href="{% url 'idhub:admin_people_membership_del' membership.id %}" title="{% trans 'Delete' %}"><i class="bi bi-trash"></i></a></td>
+          </tr>
+          {% endfor %}
+        </tbody>
+      </table>
+      <div class="form-actions-no-box">
+        <a class="btn btn-green-admin" href="{% url 'idhub:admin_people_membership_new' object.id %}">{% translate "Add membership" %} <i class="bi bi-plus"></i></a>
+      </div>
+    </div>
+  </div>
+</div>
 {% endblock %}
diff --git a/idhub/urls.py b/idhub/urls.py
index 0a87d1a..33e438d 100644
--- a/idhub/urls.py
+++ b/idhub/urls.py
@@ -66,6 +66,10 @@ urlpatterns = [
         name='admin_people_new'),
     path('admin/people/<int:pk>/membership/new/', views_admin.AdminPeopleMembershipRegisterView.as_view(),
         name='admin_people_membership_new'),
+    path('admin/membership/<int:pk>/edit/', views_admin.AdminPeopleMembershipEditView.as_view(),
+        name='admin_people_membership_edit'),
+    path('admin/membership/<int:pk>/del/', views_admin.AdminPeopleMembershipDeleteView.as_view(),
+        name='admin_people_membership_del'),
     path('admin/roles/', views_admin.AdminRolesView.as_view(),
         name='admin_roles'),
     path('admin/services/', views_admin.AdminServicesView.as_view(),