update trainee / supervisor management

generic/tables.py

@@ -1,3 +1,4 @@
+from django.urls import reverse
 import django_tables2 as tables
 from django_tables2.utils import A
 
@@ -312,8 +313,9 @@ class SupervisorTable(tables.Table):
         verbose_name="Name",
     )
 
-    edit = tables.LinkColumn(
-        "generic:supervisor_edit", text="Edit", args=[A("pk")], orderable=False
+    # We don't use a link column as we want to add a redirect parameter
+    edit = tables.Column(
+        accessor="pk", orderable=False
     )
 
     class Meta:
@@ -321,4 +323,8 @@ class SupervisorTable(tables.Table):
         template_name = "django_tables2/bootstrap4.html"
         
         sequence = ("name", "email")
-        exclude = ("id",)
+        exclude = ("id",)
+
+    def render_edit(self, value, record):
+        url = f"{reverse("generic:supervisor_edit", args=[record.pk])}?redirect={self.request.path}"
+        return format_html("<a href='{}'>Edit</a>", url)