test

api/routes.py

@@ -50,8 +50,8 @@ def setup_routes(app: FastAPI) -> None:
         return {"status": "success", "message": "Источник добавлен", "data": result}
 
     @app.get("/all_sources", summary="Метод получения всех источников")
-    async def get_all_sources():  
-        return wp.get_all_sources()
+    async def get_all_sources(category: str = "all"):  
+        return wp.get_all_sources(category)
 
     @app.delete("/delete_sources", summary="Метод удаления источника")
     async def delete_sources(url: str):

work_parser.py

@@ -345,12 +345,15 @@ def add_sources(url: str, promt: str):
     finally:
         pass   
 
-def get_all_sources():
+def get_all_sources(category: str):
     """Возвращает все записи из таблицы sourse"""
     conn = get_connection()
     try:
         with conn.cursor(cursor_factory=RealDictCursor) as cur:
-            cur.execute("SELECT * FROM sourse")
+            if category == "all":
+                cur.execute("SELECT * FROM sourse")
+            else:
+                cur.execute("SELECT * FROM sourse WHERE category = %s", (category,))
             rows = cur.fetchall()
         sources = [{"url": row["url"], "promt": row["promt"]} for row in rows]
         return {"sources": sources}