python/fastapi-users
1
0
Fork 0
mirror of https://github.com/fastapi-users/fastapi-users.git synced 2025-11-02 12:21:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Native model and generic ID (#971)

Browse Source 
* Use a generic Protocol model for User instead of Pydantic

* Remove UserDB Pydantic schema

* Harmonize schema variable naming to avoid confusions

* Revamp OAuth account model management

* Revamp AccessToken DB strategy to adopt generic model approach

* Make ID a generic instead of forcing UUIDs

* Improve generic typing

* Improve Strategy typing

* Tweak base DB typing

* Don't set Pydantic schemas on FastAPIUsers class: pass it directly on router creation

* Add IntegerIdMixin and export related classes

* Start to revamp doc for V10

* Revamp OAuth documentation

* Fix code highlights

* Write the 9.x.x ➡️ 10.x.x migration doc

* Fix pyproject.toml
This commit is contained in:
François Voron
2022-05-05 14:51:19 +02:00
committed by GitHub
parent b7734fc8b0
commit 72aa68c462
124 changed files with 2144 additions and 2114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_authentication_authenticator.py
View File

@ -12,7 +12,7 @@ from fastapi_users.authentication.strategy import Strategy
from fastapi_users.authentication.transport import Transport
from fastapi_users.manager import BaseUserManager
from fastapi_users.types import DependencyCallable
from tests.conftest import UserDB
from tests.conftest import User, UserModel
class MockSecurityScheme(SecurityBase):
@ -29,18 +29,18 @@ class MockTransport(Transport):
class NoneStrategy(Strategy):
async def read_token(
self, token: Optional[str], user_manager: BaseUserManager[models.UC, models.UD]
) -> Optional[models.UD]:
self, token: Optional[str], user_manager: BaseUserManager[models.UP, models.ID]
) -> Optional[models.UP]:
return None
class UserStrategy(Strategy, Generic[models.UC, models.UD]):
def __init__(self, user: models.UD):
class UserStrategy(Strategy, Generic[models.UP]):
def __init__(self, user: models.UP):
self.user = user
async def read_token(
self, token: Optional[str], user_manager: BaseUserManager[models.UC, models.UD]
) -> Optional[models.UD]:
self, token: Optional[str], user_manager: BaseUserManager[models.UP, models.ID]
) -> Optional[models.UP]:
return self.user
@ -55,7 +55,7 @@ def get_backend_none():
@pytest.fixture
def get_backend_user(user: UserDB):
def get_backend_user(user: UserModel):
def _get_backend_user(name: str = "user"):
return AuthenticationBackend(
name=name,
@ -78,17 +78,17 @@ def get_test_auth_client(get_user_manager, get_test_client):
app = FastAPI()
authenticator = Authenticator(backends, get_user_manager)
@app.get("/test-current-user")
@app.get("/test-current-user", response_model=User)
def test_current_user(
user: UserDB = Depends(
user: UserModel = Depends(
authenticator.current_user(get_enabled_backends=get_enabled_backends)
),
):
return user
@app.get("/test-current-active-user")
@app.get("/test-current-active-user", response_model=User)
def test_current_active_user(
user: UserDB = Depends(
user: UserModel = Depends(
authenticator.current_user(
active=True, get_enabled_backends=get_enabled_backends
)
@ -96,9 +96,9 @@ def get_test_auth_client(get_user_manager, get_test_client):
):
return user
@app.get("/test-current-superuser")
@app.get("/test-current-superuser", response_model=User)
def test_current_superuser(
user: UserDB = Depends(
user: UserModel = Depends(
authenticator.current_user(
active=True,
superuser=True,