Address incorrect typing in pycryptodome dependency

This commit is contained in:
Alex Stokes 2019-09-04 10:19:27 -07:00
parent a099b9c65d
commit 451ec2664a
No known key found for this signature in database
GPG Key ID: 51CE1721B245C086

View File

@ -24,7 +24,8 @@ class RSAPublicKey(PublicKey):
def verify(self, data: bytes, signature: bytes) -> bool:
h = SHA256.new(data)
try:
pkcs1_15.new(self.impl).verify(h, signature)
# NOTE: the typing in ``pycryptodome`` is wrong on the arguments to ``verify``.
pkcs1_15.new(self.impl).verify(h, signature) # type: ignore
except (ValueError, TypeError):
return False
return True
@ -47,7 +48,8 @@ class RSAPrivateKey(PrivateKey):
def sign(self, data: bytes) -> bytes:
h = SHA256.new(data)
return pkcs1_15.new(self.impl).sign(h)
# NOTE: the typing in ``pycryptodome`` is wrong on the arguments to ``sign``.
return pkcs1_15.new(self.impl).sign(h) # type: ignore
def get_public_key(self) -> PublicKey:
return RSAPublicKey(self.impl.publickey())