pagination

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException, Request, Query
 from fastapi.responses import JSONResponse
 import logging
 import os
@@ -51,8 +51,36 @@ async def get_music_store():
     return load_balancer.MUSIC_STORE
 
 @app.get("/api/get/music/all")
-async def get_all_music_api():
-    return load_balancer.get_all_music()
+async def get_all_music_api(
+    page: int = Query(None, ge=1),  # Default to None, but must be at least 1 if provided
+    limit: int = Query(None, ge=1, le=100)  # Default to None, but must be between 1 and 100 if provided
+):
+    # Fetch all music files from the load balancer
+    all_music = load_balancer.get_all_music()
+
+    # If pagination parameters are not provided, return all music
+    if page is None or limit is None:
+        return {"total_files": len(all_music), "files": all_music}
+
+    # Calculate the starting index and the end index for pagination
+    start_index = (page - 1) * limit
+    end_index = start_index + limit
+
+    # Handle the case where the requested page exceeds available files
+    if start_index >= len(all_music):
+        raise HTTPException(status_code=404, detail="No more files available for the requested page.")
+
+    # Get the paginated music files
+    paginated_music = all_music[start_index:end_index]
+
+    # Prepare the response
+    return {
+        "page": page,
+        "limit": limit,
+        "total_files": len(all_music),
+        "files": paginated_music
+    }
+
 
 @app.get('/api/get/instances')
 async def get_instances():
@@ -88,7 +116,28 @@ async def get_categories():
     return load_balancer.get_all_categories()
 
 @app.get("/api/get/category/{category}")
-async def get_all_from_a_category(category: str):
+async def get_all_from_a_category(
+    category: str, 
+    page: int = Query(1, ge=1),  # Default to page 1, minimum value is 1
+    limit: int = Query(10, ge=1, le=100)  # Default limit is 10, max limit is 100
+):
     if not category:
         raise HTTPException(status_code=400, detail="category parameter is required")
-    return load_balancer.get_files_from_category(category)
+
+    # Fetch all files from the category
+    all_files = load_balancer.get_files_from_category(category)
+
+    # Calculate the starting index and the end index for pagination
+    start_index = (page - 1) * limit
+    end_index = start_index + limit
+
+    # Get the paginated files
+    paginated_files = all_files[start_index:end_index]
+
+    # Prepare the response
+    return {
+        "page": page,
+        "limit": limit,
+        "total_files": len(all_files),
+        "files": paginated_files
+    }