Add query parameter to search preset by suite

Signed-off-by: Devansh Singh <[email protected]>

src/teuthology_api/routes/presets.py

@@ -1,4 +1,5 @@
 import logging
+from typing import Optional
 
 from fastapi import status, APIRouter, Depends, HTTPException, Request
 from sqlmodel import Session
@@ -25,8 +26,13 @@ def read_preset(username: str, name: str, db: Session = Depends(get_db)):
 
 
 @router.get("/list", status_code=status.HTTP_200_OK)
-def read_all_presets(username: str, db: Session = Depends(get_db)):
-    db_presets = PresetsService(db).get_by_username(username)
+def read_all_presets(
+    username: str, suite: Optional[str] = None, db: Session = Depends(get_db)
+):
+    if suite:
+        db_presets = PresetsService(db).get_by_username_and_suite(username, suite)
+    else:
+        db_presets = PresetsService(db).get_by_username(username)
     if not db_presets:
         raise HTTPException(
             status_code=status.HTTP_404_NOT_FOUND,

src/teuthology_api/services/presets.py

@@ -22,6 +22,13 @@ def get_by_username(self, username: str):
         db_presets = self.db.exec(statement).all()
         return db_presets
 
+    def get_by_username_and_suite(self, username: str, suite: str):
+        statement = select(Presets).where(
+            Presets.username == username, Presets.suite == suite
+        )
+        db_presets = self.db.exec(statement).all()
+        return db_presets
+
     def get_by_username_and_name(
         self, username: str, preset_name: str
     ) -> Union[Presets, None]: