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

Improve generic typing

Browse Source
This commit is contained in:
François Voron
2021-09-14 11:53:43 +02:00
parent 90aee2d487
commit fdc8e54253
15 changed files with 60 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
fastapi_users/manager.py
View File

@@ -37,7 +37,7 @@ class InvalidPasswordException(FastAPIUsersException):
class ValidatePasswordProtocol(Protocol): # pragma: no cover
def __call__(
self, password: str, user: Union[models.BaseUserCreate, models.BaseUserDB]
self, password: str, user: Union[models.UC, models.UD]
) -> Awaitable[None]:
pass
@@ -82,9 +82,7 @@ class UserManager(Generic[models.UD]):
return user
async def create(
self, user: models.BaseUserCreate, safe: bool = False
) -> models.UD:
async def create(self, user: models.UC, safe: bool = False) -> models.UD:
if self.validate_password:
await self.validate_password(user.password, user)
@@ -107,7 +105,7 @@ class UserManager(Generic[models.UD]):
return await self.user_db.update(user)
async def update(
self, updated_user: models.BaseUserUpdate, user: models.UD, safe: bool = False
self, updated_user: models.UU, user: models.UD, safe: bool = False
) -> models.UD:
if safe:
updated_user_data = updated_user.create_update_dict()