Update user.py

This commit is contained in:
Egor
2026-01-31 20:12:25 +03:00
committed by GitHub
parent 7bd5d13cc8
commit 9cc2a285dc

View File

@@ -686,6 +686,7 @@ async def get_users_list(
offset: int = 0,
limit: int = 50,
search: str | None = None,
email: str | None = None,
status: UserStatus | None = None,
order_by_balance: bool = False,
order_by_traffic: bool = False,
@@ -722,6 +723,9 @@ async def get_users_list(
query = query.where(or_(*conditions))
if email:
query = query.where(User.email.ilike(f'%{email}%'))
sort_flags = [
order_by_balance,
order_by_traffic,
@@ -777,7 +781,9 @@ async def get_users_list(
return users
async def get_users_count(db: AsyncSession, status: UserStatus | None = None, search: str | None = None) -> int:
async def get_users_count(
db: AsyncSession, status: UserStatus | None = None, search: str | None = None, email: str | None = None
) -> int:
query = select(func.count(User.id))
if status:
@@ -803,6 +809,9 @@ async def get_users_count(db: AsyncSession, status: UserStatus | None = None, se
query = query.where(or_(*conditions))
if email:
query = query.where(User.email.ilike(f'%{email}%'))
result = await db.execute(query)
return result.scalar()