python/fastapi-users
1
0
Fork 0
mirror of https://github.com/fastapi-users/fastapi-users.git synced 2025-08-15 03:04:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
François Voron
2024-11-03 12:46:46 +00:00
committed by GitHub
parent f37a48f97d
commit 7f92a82e07
3 changed files with 17 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
fastapi_users/authentication/authenticator.py
View File

@ -31,7 +31,9 @@ class DuplicateBackendNamesError(Exception):
pass pass
EnabledBackendsDependency = DependencyCallable[Sequence[AuthenticationBackend[models.UP, models.ID]]] EnabledBackendsDependency = DependencyCallable[
Sequence[AuthenticationBackend[models.UP, models.ID]]
]
class Authenticator(Generic[models.UP, models.ID]): class Authenticator(Generic[models.UP, models.ID]):
@ -62,7 +64,9 @@ class Authenticator(Generic[models.UP, models.ID]):
active: bool = False, active: bool = False,
verified: bool = False, verified: bool = False,
superuser: bool = False, superuser: bool = False,
get_enabled_backends: Optional[EnabledBackendsDependency[models.UP, models.ID]] = None, get_enabled_backends: Optional[
EnabledBackendsDependency[models.UP, models.ID]
] = None,
): ):
""" """
Return a dependency callable to retrieve currently authenticated user and token. Return a dependency callable to retrieve currently authenticated user and token.
@ -106,7 +110,9 @@ class Authenticator(Generic[models.UP, models.ID]):
active: bool = False, active: bool = False,
verified: bool = False, verified: bool = False,
superuser: bool = False, superuser: bool = False,
get_enabled_backends: Optional[EnabledBackendsDependency[models.UP, models.ID]] = None, get_enabled_backends: Optional[
EnabledBackendsDependency[models.UP, models.ID]
] = None,
): ):
""" """
Return a dependency callable to retrieve currently authenticated user. Return a dependency callable to retrieve currently authenticated user.
@ -157,8 +163,8 @@ class Authenticator(Generic[models.UP, models.ID]):
) -> Tuple[Optional[models.UP], Optional[str]]: ) -> Tuple[Optional[models.UP], Optional[str]]:
user: Optional[models.UP] = None user: Optional[models.UP] = None
token: Optional[str] = None token: Optional[str] = None
enabled_backends: Sequence[AuthenticationBackend[models.UP, models.ID]] = kwargs.get( enabled_backends: Sequence[AuthenticationBackend[models.UP, models.ID]] = (
"enabled_backends", self.backends kwargs.get("enabled_backends", self.backends)
) )
for backend in self.backends: for backend in self.backends:
if backend in enabled_backends: if backend in enabled_backends:

4
fastapi_users/fastapi_users.py
View File

@ -72,7 +72,9 @@ class FastAPIUsers(Generic[models.UP, models.ID]):
return get_reset_password_router(self.get_user_manager) return get_reset_password_router(self.get_user_manager)
def get_auth_router( def get_auth_router(
self, backend: AuthenticationBackend[models.UP, models.ID], requires_verification: bool = False self,
backend: AuthenticationBackend[models.UP, models.ID],
requires_verification: bool = False,
) -> APIRouter: ) -> APIRouter:
""" """
Return an auth router for a given authentication backend. Return an auth router for a given authentication backend.

6
tests/test_manager.py
View File

@ -139,7 +139,7 @@ class TestCreateUser:
): ):
user = UserCreate(email=email, password="guinevere") user = UserCreate(email=email, password="guinevere")
created_user = await user_manager.create(user) created_user = await user_manager.create(user)
assert type(created_user) == UserModel assert isinstance(created_user, UserModel)
assert user_manager.on_after_register.called is True assert user_manager.on_after_register.called is True
@ -151,7 +151,7 @@ class TestCreateUser:
email="lancelot@camelot.b", password="guinevere", is_superuser=True email="lancelot@camelot.b", password="guinevere", is_superuser=True
) )
created_user = await user_manager.create(user, safe) created_user = await user_manager.create(user, safe)
assert type(created_user) == UserModel assert isinstance(created_user, UserModel)
assert created_user.is_superuser is result assert created_user.is_superuser is result
assert user_manager.on_after_register.called is True assert user_manager.on_after_register.called is True
@ -164,7 +164,7 @@ class TestCreateUser:
email="lancelot@camelot.b", password="guinevere", is_active=False email="lancelot@camelot.b", password="guinevere", is_active=False
) )
created_user = await user_manager.create(user, safe) created_user = await user_manager.create(user, safe)
assert type(created_user) == UserModel assert isinstance(created_user, UserModel)
assert created_user.is_active is result assert created_user.is_active is result
assert user_manager.on_after_register.called is True assert user_manager.on_after_register.called is True