fix: add admin arg to crud.get_users

Gozargah · Jul 15, 2024 · 500d770 · 500d770
1 parent 9e078f2
commit 500d770
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/app/__init__.py b/app/__init__.py
@@ -11,7 +11,7 @@
 
 from config import DOCS, XRAY_SUBSCRIPTION_PATH
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 
 
 app = FastAPI(

diff --git a/app/db/crud.py b/app/db/crud.py
@@ -128,6 +128,7 @@ def get_users(db: Session,
               search: Optional[str] = None,
               status: Optional[Union[UserStatus, list]] = None,
               sort: Optional[List[UsersSortingOptions]] = None,
+              admin: Optional[Admin] = None,
               admins: Optional[List[str]] = None,
               reset_strategy: Optional[Union[UserDataLimitResetStrategy, list]] = None,
               return_with_count: bool = False) -> Union[List[User], Tuple[List[User], int]]:
@@ -151,6 +152,9 @@ def get_users(db: Session,
         else:
             query = query.filter(User.data_limit_reset_strategy == reset_strategy)
 
+    if admin:
+        query = query.filter(User.admin == admin)
+
     if admins:
         query = query.filter(User.admin.has(Admin.username.in_(admins)))