mirror of
https://gitlab.com/pulsechaincom/staking-deposit-cli.git
synced 2025-01-10 04:51:21 +00:00
4542faca87
Co-authored-by: Hsiao-Wei Wang <hwwang156@gmail.com>
30 lines
870 B
Python
30 lines
870 B
Python
from typing import List
|
|
|
|
from .mnemonic import get_seed
|
|
from .tree import (
|
|
derive_master_SK,
|
|
derive_child_SK,
|
|
)
|
|
|
|
|
|
def path_to_nodes(path: str) -> List[int]:
|
|
"""
|
|
Maps from a path string to a list of indices where each index represents the corresponding level in the path.
|
|
"""
|
|
path = path.replace(' ', '')
|
|
assert set(path).issubset(set('m1234567890/'))
|
|
indices = path.split('/')
|
|
assert indices.pop(0) == 'm'
|
|
return [int(index) for index in indices]
|
|
|
|
|
|
def mnemonic_and_path_to_key(*, mnemonic: str, path: str, password: str) -> int:
|
|
"""
|
|
Return the SK at position `path`, secured with `password` derived from `mnemonic`.
|
|
"""
|
|
seed = get_seed(mnemonic=mnemonic, password=password)
|
|
sk = derive_master_SK(seed)
|
|
for node in path_to_nodes(path):
|
|
sk = derive_child_SK(parent_SK=sk, index=node)
|
|
return sk
|