feat: add filter capabilities to the treeDetails endpoint

backend/kernelCI_app/serializers.py

@@ -74,7 +74,7 @@ class TreeDetailsSerializer(serializers.Serializer):
     architecture = serializers.CharField()
     config_name = serializers.CharField()
     valid = serializers.BooleanField()
-    start_time = serializers.CharField()
+    start_time = serializers.DateTimeField()
     duration = serializers.CharField()
     compiler = serializers.CharField()
     config_url = serializers.CharField()

backend/kernelCI_app/views.py

@@ -1,8 +1,8 @@
 from django.http import JsonResponse
 from django.views import View
-
+from querybuilder.query import Query
 from kernelCI_app.models import Checkouts, Builds
-from kernelCI_app.serializers import TreeSerializer, TreeDetailsSerializer
+from kernelCI_app.serializers import TreeSerializer
 from kernelCI_app.utils import get_visible_record_identifiers
 
 
@@ -108,25 +108,30 @@ def get_test_staus(self, build_id):
         return {k: getattr(builds[0], k) for k in status_keys}
 
     def get(self, request, commit_hash):
-        builds = Builds.objects.raw(
-            """
-            SELECT
-                builds.id, builds.architecture, builds.config_name, builds.misc,
-                builds.config_url, builds.compiler, builds.valid,
-                builds.start_time, builds.duration, builds.log_url
-            FROM
-                builds
-            INNER JOIN
-                checkouts ON checkouts.id = builds.checkout_id
-            WHERE checkouts.git_commit_hash = %s;
-            """,
-            [commit_hash]
-        )
-
-        for build in builds:
-            build.test_status = self.get_test_staus(build.id)
-
-        data = TreeDetailsSerializer(builds, many=True).data
-        summary = self.create_summary(data)
-        resp = {"builds": data, "summary": summary}
-        return JsonResponse(resp, safe=False)
+        build_fields = [
+            'id', 'architecture', 'config_name', 'misc', 'config_url',
+            'compiler', 'valid', 'duration', 'log_url', 'start_time']
+        checkout_fields = [
+            'git_repository_branch', 'git_repository_url', 'git_repository_branch']
+
+        query = Query().from_table(Builds, build_fields).join(
+            'checkouts',
+            condition='checkouts.id = builds.checkout_id',
+            fields=checkout_fields
+        ).where(git_commit_hash__eq=commit_hash)
+
+        for k in request.GET.keys():
+            if k.startswith('filter_'):
+                field = k[7:]
+                if field in build_fields or field in checkout_fields:
+                    filter_list = request.GET.getlist(k)
+                    query.where({field: filter_list})
+
+        records = query.select()
+        for r in records:
+            status = self.get_test_staus(r.get('id'))
+            r['status'] = status
+
+        summary = self.create_summary(records)
+
+        return JsonResponse({"builds": records, "summary": summary}, safe=False)

backend/pyproject.toml

@@ -12,6 +12,7 @@ django = "^5.0.6"
 djangorestframework = "^3.15.2"
 gunicorn = "^22.0.0"
 psycopg = "^3.1.19"
+django-query-builder = "^3.2.0"
 
 
 [tool.poetry.group.dev.dependencies]