feat: add origin filter to tree endpoint

kernelci · Aug 15, 2024 · e0b3343 · e0b3343
1 parent 01833fd
commit e0b3343
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/backend/kernelCI_app/views/treeView.py b/backend/kernelCI_app/views/treeView.py
@@ -5,9 +5,13 @@
 from kernelCI_app.utils import getQueryTimeInterval
 
 
+DEFAULT_ORIGIN = '0dayci'
+
+
 class TreeView(View):
+    def get(self, request):
+        origin = request.GET.get('origin', DEFAULT_ORIGIN)
 
-    def get(self, _):
         checkouts = Checkouts.objects.raw(
             """
             WITH
@@ -19,7 +23,7 @@ def get(self, _):
                             start_time
                         FROM
                             checkouts
-                        WHERE start_time  >= TO_TIMESTAMP(%s)
+                        WHERE origin = %s AND start_time  >= TO_TIMESTAMP(%s)
                         ) AS selection
                     ORDER BY
                         start_time DESC
@@ -56,7 +60,7 @@ def get(self, _):
             GROUP BY
                 checkouts.git_commit_hash, checkouts.patchset_hash
             ;
-            """, [getQueryTimeInterval().timestamp()]
+            """, [origin, getQueryTimeInterval().timestamp()]
         )
 
         serializer = TreeSerializer(checkouts, many=True)