Fix typing
This commit is contained in:
parent
497cfdc561
commit
bca05ac185
@ -13,7 +13,7 @@ from uuid import UUID, uuid4
|
|||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from tempfile import NamedTemporaryFile
|
from tempfile import NamedTemporaryFile
|
||||||
|
|
||||||
from typing import Optional, TYPE_CHECKING, Dict, Callable, List, AsyncGenerator
|
from typing import Optional, TYPE_CHECKING, Dict, Callable, List, AsyncGenerator, Any
|
||||||
|
|
||||||
from pymongo import ReturnDocument
|
from pymongo import ReturnDocument
|
||||||
from pymongo.errors import AutoReconnect, DuplicateKeyError
|
from pymongo.errors import AutoReconnect, DuplicateKeyError
|
||||||
@ -140,7 +140,7 @@ class OrgOps:
|
|||||||
role: UserRole = UserRole.VIEWER,
|
role: UserRole = UserRole.VIEWER,
|
||||||
page_size: int = DEFAULT_PAGE_SIZE,
|
page_size: int = DEFAULT_PAGE_SIZE,
|
||||||
page: int = 1,
|
page: int = 1,
|
||||||
sort_by: Optional[str] = "name",
|
sort_by: str = "name",
|
||||||
sort_direction: int = 1,
|
sort_direction: int = 1,
|
||||||
):
|
):
|
||||||
"""Get all orgs a user is a member of"""
|
"""Get all orgs a user is a member of"""
|
||||||
@ -150,12 +150,11 @@ class OrgOps:
|
|||||||
page = page - 1
|
page = page - 1
|
||||||
skip = page_size * page
|
skip = page_size * page
|
||||||
|
|
||||||
if user.is_superuser:
|
query: Dict[str, Any] = {}
|
||||||
query: Dict[str, object] = {}
|
if not user.is_superuser:
|
||||||
else:
|
query[f"users.{user.id}"] = {"$gte": role.value}
|
||||||
query: Dict[str, object] = {f"users.{user.id}": {"$gte": role.value}}
|
|
||||||
|
|
||||||
aggregate = [{"$match": query}]
|
aggregate: List[Dict[str, Any]] = [{"$match": query}]
|
||||||
|
|
||||||
# Ensure default org is always first, then sort on sort_by if set
|
# Ensure default org is always first, then sort on sort_by if set
|
||||||
sort_query = {"default": -1}
|
sort_query = {"default": -1}
|
||||||
@ -1132,8 +1131,8 @@ def init_orgs_api(
|
|||||||
user: User = Depends(user_dep),
|
user: User = Depends(user_dep),
|
||||||
pageSize: int = DEFAULT_PAGE_SIZE,
|
pageSize: int = DEFAULT_PAGE_SIZE,
|
||||||
page: int = 1,
|
page: int = 1,
|
||||||
sortBy: Optional[str] = "name",
|
sortBy: str = "name",
|
||||||
sortDirection: Optional[int] = 1,
|
sortDirection: int = 1,
|
||||||
):
|
):
|
||||||
results, total = await ops.get_orgs_for_user(
|
results, total = await ops.get_orgs_for_user(
|
||||||
user,
|
user,
|
||||||
|
Loading…
Reference in New Issue
Block a user