Add bulk operations for processes

main/tables.py

@@ -33,6 +33,4 @@ class ProcessTable(tables.Table):
     status_code = tables.Column(verbose_name="Status Code")
     exit_code = tables.Column(verbose_name="Exit Code")
     logs = tables.TemplateColumn(logs_column_template, verbose_name="Logs")
-    restart = tables.TemplateColumn(restart_column_template, verbose_name="Restart")
-    flush = tables.TemplateColumn(flush_column_template, verbose_name="Flush")
-    kill = tables.TemplateColumn(kill_column_template, verbose_name="Kill")
+    select = tables.CheckBoxColumn(accessor="uuid", verbose_name="Select")

main/templates/main/index.html

@@ -5,8 +5,14 @@
 
 {% block content %}
 <div class="col">
-  <a href="{% url 'main:boot_process' %}" class="btn btn-primary">Boot</a>
-  {% render_table table %}
+  <form method="post", action="{% url 'main:process_action' %}">
+    {% csrf_token %}
+    <a href="{% url 'main:boot_process' %}" class="btn btn-primary">Boot</a>
+    <input type="submit" value="Restart" class="btn btn-success", name="action", onclick="return confirm('Restart selected processes?')">
+    <input type="submit" value="Flush" class="btn btn-warning", name="action", onclick="return confirm('Flush selected processes?')">
+    <input type="submit" value="Kill" class="btn btn-danger", name="action", onclick="return confirm('Kill selected processes?')">
+    {% render_table table %}
+  </form>
 </div>
 
 {% endblock content %}

main/urls.py

@@ -8,9 +8,7 @@
 urlpatterns = [
     path("", views.index, name="index"),
     path("accounts/", include("django.contrib.auth.urls")),
-    path("restart/<uuid:uuid>", views.restart_process, name="restart"),
-    path("kill/<uuid:uuid>", views.kill_process, name="kill"),
-    path("flush/<uuid:uuid>", views.flush_process, name="flush"),
+    path("process_action/", views.process_action, name="process_action"),
     path("logs/<uuid:uuid>", views.logs, name="logs"),
     path("boot_process/", views.BootProcessView.as_view(), name="boot_process"),
 ]

main/views.py

@@ -80,68 +80,49 @@ class ProcessAction(Enum):
     FLUSH = "flush"
 
 
-async def _process_call(uuid: str, action: ProcessAction) -> None:
+async def _process_call(uuids: list[str], action: ProcessAction) -> None:
     """Perform an action on a process with a given UUID.
 
     Args:
-        uuid: UUID of the process to be actioned.
-        action: Action to be performed {restart,kill}.
+        uuids: List of UUIDs of the process to be actioned.
+        action: Action to be performed {restart,flush,kill}.
     """
     pmd = get_process_manager_driver()
-    query = ProcessQuery(uuids=[ProcessUUID(uuid=uuid)])
+    uuids_ = [ProcessUUID(uuid=u) for u in uuids]
 
     match action:
         case ProcessAction.RESTART:
-            await pmd.restart(query)
+            for uuid_ in uuids_:
+                query = ProcessQuery(uuids=[uuid_])
+                await pmd.restart(query)
         case ProcessAction.KILL:
+            query = ProcessQuery(uuids=uuids_)
             await pmd.kill(query)
         case ProcessAction.FLUSH:
+            query = ProcessQuery(uuids=uuids_)
             await pmd.flush(query)
 
 
 @login_required
-def restart_process(request: HttpRequest, uuid: uuid.UUID) -> HttpResponse:
-    """Restart the process associated to the given UUID.
-
-    Args:
-        request: HttpRequest object. This is not used in the function, but is required
-            by Django.
-        uuid: UUID of the process to be restarted.
-
-    Returns:
-        HttpResponse, redirecting to the main page.
-    """
-    asyncio.run(_process_call(str(uuid), ProcessAction.RESTART))
-    return HttpResponseRedirect(reverse("main:index"))
-
-
-@login_required
-def kill_process(request: HttpRequest, uuid: uuid.UUID) -> HttpResponse:
-    """Kill the process associated to the given UUID.
+def process_action(request: HttpRequest) -> HttpResponse:
+    """Perform an action on the selected processes.
 
-    Args:
-        request: Django HttpRequest object (unused, but required by Django).
-        uuid: UUID of the process to be killed.
-
-    Returns:
-        HttpResponse redirecting to the index page.
-    """
-    asyncio.run(_process_call(str(uuid), ProcessAction.KILL))
-    return HttpResponseRedirect(reverse("main:index"))
-
-
-@login_required
-def flush_process(request: HttpRequest, uuid: uuid.UUID) -> HttpResponse:
-    """Flush the process associated to the given UUID.
+    Both the action and the selected processes are retrieved from the request.
 
     Args:
-        request: Django HttpRequest object (unused, but required by Django).
-        uuid: UUID of the process to be flushed.
+        request: Django HttpRequest object.
 
     Returns:
         HttpResponse redirecting to the index page.
     """
-    asyncio.run(_process_call(str(uuid), ProcessAction.FLUSH))
+    try:
+        action = request.POST.get("action", "")
+        action_enum = ProcessAction(action.lower())
+    except ValueError:
+        return HttpResponseRedirect(reverse("main:index"))
+
+    if uuids_ := request.POST.getlist("select"):
+        asyncio.run(_process_call(uuids_, action_enum))
     return HttpResponseRedirect(reverse("main:index"))
 
 

tests/test_views.py

@@ -21,27 +21,6 @@ def test_login_redirect(self, client):
         assertRedirects(response, reverse("main:login") + f"?next={self.endpoint}")
 
 
-class ProcessActionsTest(LoginRequiredTest):
-    """Grouping the tests for the process action views."""
-
-    action: ProcessAction
-
-    @classmethod
-    def setup_class(cls):
-        """Set up the endpoint for the tests."""
-        cls.uuid = uuid4()
-        cls.endpoint = reverse(f"main:{cls.action.value}", kwargs=dict(uuid=cls.uuid))
-
-    def test_process_action_view_authenticated(self, auth_client, mocker):
-        """Test the process action view for an authenticated user."""
-        mock = mocker.patch("main.views._process_call")
-        response = auth_client.get(self.endpoint)
-        assert response.status_code == HTTPStatus.FOUND
-        assert response.url == reverse("main:index")
-
-        mock.assert_called_once_with(str(self.uuid), self.action)
-
-
 class TestIndexView(LoginRequiredTest):
     """Tests for the index view."""
 
@@ -81,22 +60,35 @@ def test_logs_view_authenticated(self, auth_client, mocker):
         assert "log_text" in response.context
 
 
-class TestProcessFlushView(ProcessActionsTest):
-    """Tests for the process flush view."""
-
-    action = ProcessAction.FLUSH
+class TestProcessActionView(LoginRequiredTest):
+    """Tests for the process_action view."""
 
+    endpoint = reverse("main:process_action")
 
-class TestProcessKillView(ProcessActionsTest):
-    """Tests for the process kill view."""
-
-    action = ProcessAction.KILL
+    def test_process_action_no_action(self, auth_client):
+        """Test process_action view with no action provided."""
+        response = auth_client.post(self.endpoint, data={})
+        assert response.status_code == HTTPStatus.FOUND
+        assert response.url == reverse("main:index")
 
+    def test_process_action_invalid_action(self, auth_client):
+        """Test process_action view with an invalid action."""
+        response = auth_client.post(self.endpoint, data={"action": "invalid_action"})
+        assert response.status_code == HTTPStatus.FOUND
+        assert response.url == reverse("main:index")
 
-class TestProcessRestartView(ProcessActionsTest):
-    """Tests for the process restart view."""
+    @pytest.mark.parametrize("action", ["kill", "restart", "flush"])
+    def test_process_action_valid_action(self, action, auth_client, mocker):
+        """Test process_action view with a valid action."""
+        mock = mocker.patch("main.views._process_call")
+        uuids_ = [str(uuid4()), str(uuid4())]
+        response = auth_client.post(
+            self.endpoint, data={"action": action, "select": uuids_}
+        )
+        assert response.status_code == HTTPStatus.FOUND
+        assert response.url == reverse("main:index")
 
-    action = ProcessAction.RESTART
+        mock.assert_called_once_with(uuids_, ProcessAction(action))
 
 
 class TestBootProcess(LoginRequiredTest):