Optimize admin queries

huey_monitor/admin.py

@@ -1,12 +1,14 @@
 from bx_django_utils.templatetags.humanize_time import human_duration
 from django.contrib import admin, messages
 from django.contrib.admin.views.main import ChangeList
+from django.db.models import OuterRef, Prefetch
 from django.shortcuts import redirect
 from django.template.loader import render_to_string
 from django.urls import path, reverse
 from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
 from huey.contrib.djhuey import HUEY
+from huey.signals import SIGNAL_EXECUTING
 
 from huey_monitor.models import SignalInfoModel, TaskModel
 
@@ -17,7 +19,21 @@ def get_queryset(self, request):
         List only the main-tasks (sub-tasks will be inlined)
         """
         qs = super().get_queryset(request)
-        qs = qs.filter(parent_task__isnull=True)
+        executing_dt = SignalInfoModel.objects.filter(
+            task_id=OuterRef("task_id"), signal_name=SIGNAL_EXECUTING
+        ).values('create_dt')[:1]
+        qs = (
+            qs.filter(parent_task__isnull=True)
+            .prefetch_related(
+                Prefetch(
+                    "sub_tasks",
+                    queryset=TaskModel.objects.select_related("state")
+                    .annotate(_executing_dt=executing_dt)
+                    .order_by('-create_dt'),
+                )
+            )
+            .annotate(_executing_dt=executing_dt)
+        )
         return qs
 
 
@@ -27,11 +43,7 @@ def get_changelist(self, request, **kwargs):
         return TaskModelChangeList
 
     def column_name(self, obj):
-        qs = TaskModel.objects.filter(parent_task_id=obj.pk).order_by('-create_dt')
-        context = {
-            'main_task': obj,
-            'sub_tasks': qs
-        }
+        context = {'main_task': obj, 'sub_tasks': obj.sub_tasks.all()}
         return render_to_string(
             template_name='admin/huey_monitor/taskmodel/column_name.html',
             context=context,
@@ -168,6 +180,7 @@ def task_name(self, obj):
     )
     readonly_fields = ('create_dt',)
     list_display_links = ('task_name',)
+    list_select_related = ('task',)
     ordering = ('-create_dt',)
     date_hierarchy = 'create_dt'
     list_filter = ('task__name', 'signal_name', 'hostname')

huey_monitor/migrations/0010_alter_taskmodel_parent_task.py

@@ -0,0 +1,28 @@
+# Generated by Django 3.2.14 on 2022-11-25 12:11
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('huey_monitor', '0009_fix_optinal_fields'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='taskmodel',
+            name='parent_task',
+            field=models.ForeignKey(
+                blank=True,
+                editable=False,
+                help_text='Only set if this task is a sub task started from his parent.',
+                null=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name='sub_tasks',
+                to='huey_monitor.taskmodel',
+                verbose_name='Parent Task',
+            ),
+        ),
+    ]

huey_monitor/models.py

@@ -43,7 +43,7 @@ class TaskModel(TimetrackingBaseModel):
         to='self',
         null=True, blank=True,
         editable=False,
-        related_name='+',
+        related_name='sub_tasks',
         on_delete=models.CASCADE,
         verbose_name=_('Parent Task'),
         help_text=_('Only set if this task is a sub task started from his parent.'),
@@ -106,6 +106,8 @@ class TaskModel(TimetrackingBaseModel):
 
     @cached_property
     def executing_dt(self):
+        if hasattr(self, "_executing_dt"):
+            return self._executing_dt
         executing_signal = SignalInfoModel.objects.filter(
             task_id=self.task_id,
             signal_name=SIGNAL_EXECUTING