Improve types.

This commit is contained in:
Tom Hacohen 2021-01-04 11:56:17 +02:00
parent 9bf118225d
commit 9f0430a6da
2 changed files with 5 additions and 5 deletions

View File

@ -125,7 +125,7 @@ def get_login_user(request: Request, challenge: LoginChallengeIn) -> UserType:
raise AuthenticationFailed(code="user_not_found", detail="User not found")
def get_encryption_key(salt):
def get_encryption_key(salt: bytes):
key = nacl.hash.blake2b(settings.SECRET_KEY.encode(), encoder=nacl.encoding.RawEncoder)
return nacl.hash.blake2b(
b"",

View File

@ -50,21 +50,21 @@ def is_collection_admin(collection, user):
return (member is not None) and (member.accessLevel == AccessLevels.ADMIN)
def msgpack_encode(content):
def msgpack_encode(content) -> bytes:
ret = msgpack.packb(content, use_bin_type=True)
assert ret is not None
return ret
def msgpack_decode(content):
def msgpack_decode(content: bytes):
return msgpack.unpackb(content, raw=False)
def b64encode(value):
def b64encode(value: bytes):
return base64.urlsafe_b64encode(value).decode("ascii").strip("=")
def b64decode(data):
def b64decode(data: str):
data += "=" * ((4 - len(data) % 4) % 4)
return base64.urlsafe_b64decode(data)