python/fastapi-users
1
0
Fork 0
mirror of https://github.com/fastapi-users/fastapi-users.git synced 2026-03-13 07:49:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make ID a generic instead of forcing UUIDs

Browse Source
This commit is contained in:
François Voron
2022-05-01 11:18:27 +02:00
parent 87ac51a7bd
commit 7093c9e38a
25 changed files with 143 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fastapi_users/db/__init__.py
View File

@@ -1,9 +1,6 @@
from fastapi_users.db.base import BaseUserDatabase, UserDatabaseDependency
__all__ = [
"BaseUserDatabase",
"UserDatabaseDependency",
]
__all__ = ["BaseUserDatabase", "UserDatabaseDependency"]
try: # pragma: no cover
from fastapi_users_db_mongodb import MongoDBUserDatabase # noqa: F401

10
fastapi_users/db/base.py
View File

@@ -1,15 +1,13 @@
from typing import Any, Dict, Generic, Optional
from pydantic import UUID4
from fastapi_users.models import OAP, UP
from fastapi_users.models import ID, OAP, UP
from fastapi_users.types import DependencyCallable
class BaseUserDatabase(Generic[UP]):
class BaseUserDatabase(Generic[UP, ID]):
"""Base adapter for retrieving, creating and updating users from a database."""
async def get(self, id: UUID4) -> Optional[UP]:
async def get(self, id: ID) -> Optional[UP]:
"""Get a single user by id."""
raise NotImplementedError()
@@ -44,4 +42,4 @@ class BaseUserDatabase(Generic[UP]):
raise NotImplementedError()
UserDatabaseDependency = DependencyCallable[BaseUserDatabase[UP]]
UserDatabaseDependency = DependencyCallable[BaseUserDatabase[UP, ID]]