feat: allow category in /all

r3tr0ananas · Aug 26, 2024 · c61ba52 · c61ba52
1 parent 4831c16
commit c61ba52
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/api/__init__.py b/api/__init__.py
@@ -1 +1 @@
-__version__ = "1.1.8"
+__version__ = "1.1.9"
diff --git a/api/main.py b/api/main.py
@@ -1,4 +1,4 @@
-from typing import List, Tuple
+from typing import List, Tuple, Optional
 
 from fastapi import FastAPI
 from fastapi.responses import RedirectResponse, JSONResponse, FileResponse
@@ -102,7 +102,12 @@ async def info():
         }
     },
 )
-async def all():
+async def all(category: Optional[str] = None):
+    if category is not None:
+        return [
+            image.to_dict() for image in agac.images if image.category == category
+        ]
+
     return [
         image.to_dict() for image in agac.images
     ]