update custom table page

Author: Cuong Nguyen

admin_extended/base.py

@@ -1,4 +1,5 @@
 import copy
+from dataclasses import dataclass
 
 from django.contrib import messages
 from django.contrib import admin
@@ -12,6 +13,15 @@ def has_search_fields(field):
     model_admin = admin.site._registry.get(field.related_model)
     return model_admin and model_admin.search_fields
 
+@dataclass
+class TableData:
+    header: str
+    table_titles = []
+    table_rows = []
+
+    def add_rows(self, row: list):
+        self.table_rows.append(row)
+
 
 class ExtendedAdminModel(ObjectToolModelAdminMixin, UIUtilsMixin, admin.ModelAdmin):
     """
@@ -146,25 +156,16 @@ def get_urls(self):
             path('', self.custom_view, name=view_name),
         ]
 
-    def get_data(self):
+    def get_table_data(self):
         """
-        return dict of list. Eg:
-        results = {
-            'title': ['A', 'B', 'C'],
-            'rows': [
-                [12, 12, 14],
-                [12, 12, 14],
-                [12, 12, 14],
-                [12, 12, 14],
-            ],
-        }
+        return list of TableData
         """
         raise NotImplementedError()
 
     def custom_view(self, request, *args, **kwargs):
         context = {
             **admin.site.each_context(request),
-            'data': self.get_data(),
+            'tables': self.get_table_data(),
         }
 
         return render(request, 'admin/custom/custom_table_page.html', context)

admin_extended/templates/admin/custom/custom_table_page.html

@@ -18,10 +18,12 @@
     <div class="module filtered" id="changelist">
       <div class="changelist-form-container">
         <div class="results">
-          <table id="result_list">
+        {% for table in tables %}
+          <h2>{{ table.header }}</h2>
+          <table class="result_list" style="margin-bottom: 40px">
             <thead>
             <tr>
-              {% for title in data.title %}
+              {% for title in table.table_titles %}
                 <th scope="col" class="">
                   <div class="text"><span>{{ title }}</span></div>
                   <div class="clear"></div>
@@ -30,7 +32,7 @@
             </tr>
             </thead>
             <tbody>
-            {% for row in data.rows %}
+            {% for row in table.table_rows %}
               <tr>
                 {% for item in row %}
                   <td class="">{{ item }}</td>
@@ -39,6 +41,7 @@
             {% endfor %}
             </tbody>
           </table>
+        {% endfor %}
         </div>
       </div>