From f5af7cc0ed2f8e6e487b52df2d14e4b6bae8b19e Mon Sep 17 00:00:00 2001
From: Tim Repke <repke@mcc-berlin.net>
Date: Mon, 18 Dec 2023 14:01:25 +0100
Subject: [PATCH] add count nql
---
server/api/routes/search.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/server/api/routes/search.py b/server/api/routes/search.py
index 0c0f2f9..5e1e6c9 100644
--- a/server/api/routes/search.py
+++ b/server/api/routes/search.py
@@ -112,7 +112,7 @@ async def nql_query(query: NQLFilter,
@router.post('/nql/count', response_model=int)
async def nql_query_count(query: NQLFilter,
- permissions: UserPermissions = Depends(UserPermissionChecker('dataset_read'))) -> QueryResult:
+ permissions: UserPermissions = Depends(UserPermissionChecker('dataset_read'))) -> int:
async with db_engine.session() as session: # type: AsyncSession
nql = await _get_query(session=session, query=query, project_id=permissions.permissions.project_id)
- return (await session.execute(func.count(nql.stmt.subquery().c.item_id))).scalar()
+ return (await session.execute(func.count(nql.stmt.subquery().c.item_id))).scalar() # type: ignore[return-value]
--
GitLab