python/fastapi-users
1
0
Fork 0
mirror of https://github.com/fastapi-users/fastapi-users.git synced 2025-11-01 10:25:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add get by id method on DB adapter

Browse Source
This commit is contained in:
François Voron
2019-10-07 07:53:56 +02:00
parent c4ff6ca3fe
commit 51b67b0739
3 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fastapi_users/db/__init__.py
View File

@ -15,6 +15,9 @@ class UserDBInterface:
async def list(self) -> List[UserDB]:
raise NotImplementedError()
async def get(self, id: str) -> UserDB:
raise NotImplementedError()
async def get_by_email(self, email: str) -> UserDB:
raise NotImplementedError()

4
fastapi_users/db/sqlalchemy.py
View File

@ -34,6 +34,10 @@ class SQLAlchemyUserDB(UserDBInterface):
query = users.select()
return await self.database.fetch_all(query)
async def get(self, id: str) -> UserDB:
query = users.select().where(User.id == id)
return await self.database.fetch_one(query)
async def get_by_email(self, email: str) -> UserDB:
query = users.select().where(User.email == email)
return await self.database.fetch_one(query)