mirror of
https://github.com/onyx-dot-app/onyx.git
synced 2026-10-23 03:42:59 -05:00
19 lines
553 B
Python
19 lines
553 B
Python
from collections.abc import Sequence
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from danswer.db.models import User
|
|
|
|
|
|
def list_users(db_session: Session) -> Sequence[User]:
|
|
"""List all users. No pagination as of now, as the # of users
|
|
is assumed to be relatively small (<< 1 million)"""
|
|
return db_session.scalars(select(User)).unique().all()
|
|
|
|
|
|
def get_user_by_email(email: str, db_session: Session) -> User | None:
|
|
user = db_session.query(User).filter(User.email == email).first() # type: ignore
|
|
|
|
return user
|