diff --git a/src/server.py b/src/server.py index 973423a..6e2b245 100644 --- a/src/server.py +++ b/src/server.py @@ -96,7 +96,12 @@ truth_router = APIRouter( @truth_router.get("", response_model=list[TruthPayload]) -def get_all_truths(rich: bool = True, db: redis.Redis = Depends(get_db_factory())): +def get_all_truths( + rich: bool = True, + limit: int = -1, + page: int = 0, + db: redis.Redis = Depends(get_db_factory()) +): """Retrieves all stored truths""" keys = db.keys() if rich is False: @@ -105,13 +110,15 @@ def get_all_truths(rich: bool = True, db: redis.Redis = Depends(get_db_factory() for key in keys ] truths = [json.loads(db.get(key)) for key in keys] + if limit >= 0: + return truths[page * limit:(page + 1) * limit] return truths @truth_router.get("/all", deprecated=True, response_model=list[TruthPayload]) def get_all_truths_deprecated(response: JSONResponse, rich: bool = True, db: redis.Redis = Depends(get_db_factory())): """DEPRECATED - USE get_all_truths INSTEAD""" - return get_all_truths(rich, db) + return get_all_truths(rich=rich, db=db) @truth_router.get("/{truth_id}", response_model=TruthPayload)