Skip to content

Commit

Permalink
linting fix
Browse files Browse the repository at this point in the history
Signed-off-by: PatStLouis <patrick.st-louis@opsecid.ca>
  • Loading branch information
PatStLouis committed Sep 20, 2024
1 parent 1a752bb commit a1731d4
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 4 deletions.
1 change: 1 addition & 0 deletions aries_cloudagent/wallet/did_info.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@


class KeyInfo(NamedTuple):
"""Class returning key information."""
verkey: str
metadata: dict
key_type: KeyType
Expand Down
5 changes: 4 additions & 1 deletion aries_cloudagent/wallet/keys/manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,16 +24,19 @@ def __init__(self, profile=Profile):
self.profile = profile

def _multikey_to_verkey(self, multikey, alg="ed25519"):
"""Transform multikey to verkey."""
prefix_lenght = ALG_MAPPINGS[alg]["prefix_lenght"]
public_bytes = bytes(bytearray(multibase.decode(multikey))[prefix_lenght:])
return bytes_to_b58(public_bytes)

def _verkey_to_multikey(self, verkey, alg="ed25519"):
"""Transform verkey to multikey."""
prefix_hex = ALG_MAPPINGS[alg]["prefix_hex"]
prefixed_key_hex = f"{prefix_hex}{b58_to_bytes(verkey).hex()}"
return multibase.encode(bytes.fromhex(prefixed_key_hex), "base58btc")

async def kid_exists(self, kid):
"""Check if kid exists."""
async with self.profile.session() as session:
wallet = session.inject(BaseWallet)
try:
Expand Down Expand Up @@ -88,7 +91,7 @@ async def update(self, multikey: str, kid: str):
wallet = session.inject(BaseWallet)
try:
verkey = self._multikey_to_verkey(multikey)
except:
except Exception:
raise MultikeyManagerError(f"Invalid multikey value {multikey}.")
key_info = await wallet.assign_kid_to_key(verkey=verkey, kid=kid)
return {
Expand Down
3 changes: 2 additions & 1 deletion aries_cloudagent/wallet/keys/routes.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,8 @@ class CreateKeyRequestSchema(OpenAPISchema):
seed = fields.Str(
required=False,
metadata={
"description": "Optional seed to generate the key pair. Must enable insecure wallet mode.",
"description": "Optional seed to generate the key pair. \
Must enable insecure wallet mode.",
"example": "00000000000000000000000000000000",
},
)
Expand Down
4 changes: 2 additions & 2 deletions aries_cloudagent/wallet/keys/tests/test_key_operations.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,10 @@ class TestKeyOperations(IsolatedAsyncioTestCase):
async def test_key_creation(self):
key_info = await self.manager.create(seed=self.seed)
assert key_info["multikey"] == self.multikey
assert key_info["kid"] == None
assert key_info["kid"] is None
key_info = await self.manager.from_multikey(multikey=self.multikey)
assert key_info["multikey"] == self.multikey
assert key_info["kid"] == None
assert key_info["kid"] is None
key_info = await self.manager.update(multikey=self.multikey, kid=self.kid)
assert key_info["multikey"] == self.multikey
assert key_info["kid"] == self.kid
Expand Down

0 comments on commit a1731d4

Please sign in to comment.