class WalletManager(object):
def __init__(self, scheme=SignatureScheme.SHA256withECDSA):
self.scheme = scheme
self.wallet_file = WalletData()
self.wallet_in_mem = WalletData()
self.wallet_path = ""
def open_wallet(self, wallet_path: str):
self.wallet_path = wallet_path
if is_file_exist(wallet_path) is False:
# create a new wallet file
self.wallet_in_mem.create_time = datetime.today().strftime("%Y-%m-%d %H:%M:%S")
self.save()
# wallet file exists now
self.wallet_file = self.load()
self.wallet_in_mem = self.wallet_file
return self.wallet_file
def load(self):
with open(self.wallet_path, "rb") as f:
content = f.read()
if content.startswith(codecs.BOM_UTF8):
content = content[len(codecs.BOM_UTF8):]
obj = json.loads(content)
try:
create_time = obj['createTime']
except KeyError:
create_time = ''
try:
default_id = obj['defaultOntid']
except KeyError:
default_id = ''
try:
default_address = obj['defaultAccountAddress']
except KeyError:
default_address = ''
try:
identities = obj['identities']
except KeyError:
identities = list()
try:
scrypt_dict = obj['scrypt']
scrypt_obj = Scrypt(scrypt_dict.get('n', 16384), scrypt_dict.get('r', 8), scrypt_dict.get('p', 8),
scrypt_dict.get('dk_len', 64))
wallet = WalletData(obj['name'], obj['version'], create_time, default_id, default_address,
scrypt_obj, identities, obj['accounts'])
except KeyError as e:
raise SDKException(ErrorCode.param_err('wallet file format error: %s.' % e))
return wallet
def save(self):
with open(self.wallet_path, "w") as f:
json.dump(self.wallet_in_mem, f, default=lambda obj: dict(obj), indent=4)
def get_wallet(self):
return self.wallet_in_mem
def write_wallet(self):
self.save()
self.wallet_file = self.wallet_in_mem
return self.wallet_file
def reset_wallet(self):
self.wallet_in_mem = self.wallet_file.clone()
return self.wallet_in_mem
def get_signature_scheme(self):
return self.scheme
def set_signature_scheme(self, scheme):
self.scheme = scheme
def import_identity(self, label: str, encrypted_pri_key: str, pwd: str, salt: str,
b58_address: str) -> Identity or None:
"""
This interface is used to import identity by providing encrypted private key, password, salt and
base58 encode address which should be correspond to the encrypted private key provided.
:param label: a label for identity.
:param encrypted_pri_key: an encrypted private key in base64 encoding from.
:param pwd: a password which is used to encrypt and decrypt the private key.
:param salt: a salt value which will be used in the process of encrypt private key.
:param b58_address: a base58 encode address which correspond with the encrypted private key provided.
:return: if succeed, an Identity object will be returned.
"""
scrypt_n = Scrypt().get_n()
pri_key = Account.get_gcm_decoded_private_key(encrypted_pri_key, pwd, b58_address, salt, scrypt_n, self.scheme)
info = self.__create_identity(label, pwd, salt, pri_key)
for index in range(len(self.wallet_in_mem.identities)):
if self.wallet_in_mem.identities[index].ont_id == info.ont_id:
return self.wallet_in_mem.identities[index]
return None
def create_identity(self, label: str, pwd: str) -> Identity:
"""
:param label: a label for identity.
:param pwd: a password which will be used to encrypt and decrypt the private key.
:return: if succeed, an Identity object will be returned.
"""
pri_key = get_random_hex_str(64)
salt = get_random_hex_str(16)
return self.__create_identity(label, pwd, salt, pri_key)
def __create_identity(self, label: str, pwd: str, salt: str, private_key: str):
acct = self.__create_account(label, pwd, salt, private_key, False)
info = IdentityInfo()
info.ont_id = did_ont + acct.get_address_base58()
info.pubic_key = acct.serialize_public_key().hex()
info.private_key = acct.serialize_private_key().hex()
info.pri_key_wif = acct.export_wif().encode('ascii')
info.encrypted_pri_key = acct.export_gcm_encrypted_private_key(pwd, salt, Scrypt().get_n())
info.address_u160 = acct.get_address().to_array().hex()
return self.wallet_in_mem.get_identity_by_ont_id(info.ont_id)
def create_identity_from_private_key(self, label: str, pwd: str, private_key: str) -> Identity:
"""
This interface is used to create identity based on given label, password and private key.
:param label: a label for identity.
:param pwd: a password which will be used to encrypt and decrypt the private key.
:param private_key: a private key in the form of string.
:return: if succeed, an Identity object will be returned.
"""
salt = get_random_hex_str(16)
identity = self.__create_identity(label, pwd, salt, private_key)
return identity
def create_account(self, label: str, pwd: str) -> AccountData:
"""
This interface is used to create account based on given password and label.
:param label: a label for account.
:param pwd: a password which will be used to encrypt and decrypt the private key
:return: if succeed, return an data structure which contain the information of a wallet account.
"""
pri_key = get_random_hex_str(64)
salt = get_random_hex_str(16)
account = self.__create_account(label, pwd, salt, pri_key, True)
return self.wallet_in_mem.get_account_by_address(account.get_address_base58())
def __create_account(self, label: str, pwd: str, salt: str, private_key: str, account_flag: bool):
account = Account(private_key, self.scheme)
# initialization
if self.scheme == SignatureScheme.SHA256withECDSA:
acct = AccountData()
else:
raise ValueError("scheme type is error")
# set key
if pwd is not None:
acct.key = account.export_gcm_encrypted_private_key(pwd, salt, Scrypt().get_n())
else:
acct.key = account.serialize_private_key().hex()
acct.address = account.get_address_base58()
# set label
if label is None or label == "":
label = str(uuid.uuid4())[0:8]
if account_flag:
for index in range(len(self.wallet_in_mem.accounts)):
if acct.address == self.wallet_in_mem.accounts[index].address:
raise ValueError("wallet account exists")
if len(self.wallet_in_mem.accounts) == 0:
acct.is_default = True
self.wallet_in_mem.default_account_address = acct.address
acct.label = label
acct.salt = base64.b64encode(salt.encode('latin-1')).decode('ascii')
acct.public_key = account.serialize_public_key().hex()
self.wallet_in_mem.accounts.append(acct)
# else:
for index in range(len(self.wallet_in_mem.identities)):
if self.wallet_in_mem.identities[index].ont_id == did_ont + acct.address:
raise ValueError("wallet identity exists")
idt = Identity()
idt.ont_id = did_ont + acct.address
idt.label = label
if len(self.wallet_in_mem.identities) == 0:
idt.is_default = True
self.wallet_in_mem.default_ont_id = idt.ont_id
ctl = Control(id="keys-1", key=acct.key, salt=base64.b64encode(salt.encode()).decode('ascii'),
address=acct.address,
public_key=account.serialize_public_key().hex())
idt.controls.append(ctl)
self.wallet_in_mem.identities.append(idt)
return account
def import_account(self, label: str, encrypted_pri_key: str, pwd: str, base58_address: str,
base64_salt: str) -> AccountData or None:
"""
This interface is used to import account by providing account data.
:param label: str, wallet label
:param encrypted_pri_key: str, an encrypted private key in base64 encoding from
:param pwd: str, a password which is used to encrypt and decrypt the private key
:param base58_address: str, a base58 encode wallet address value
:param base64_salt: str, a base64 encode salt value which is used in the encryption of private key
:return:
if succeed, return an data structure which contain the information of a wallet account.
if failed, return a None object.
"""
salt = base64.b64decode(base64_salt.encode('ascii')).decode('latin-1')
private_key = Account.get_gcm_decoded_private_key(encrypted_pri_key, pwd, base58_address, salt,
Scrypt().get_n(), self.scheme)
info = self.create_account_info(label, pwd, salt, private_key)
for index in range(len(self.wallet_in_mem.accounts)):
if info.address_base58 == self.wallet_in_mem.accounts[index].address:
return self.wallet_in_mem.accounts[index]
return None
def create_account_info(self, label: str, pwd: str, salt: str, private_key: str) -> AccountInfo:
acct = self.__create_account(label, pwd, salt, private_key, True)
info = AccountInfo()
info.address_base58 = Address.address_from_bytes_pubkey(acct.serialize_public_key()).b58encode()
info.public_key = acct.serialize_public_key().hex()
info.encrypted_pri_key = acct.export_gcm_encrypted_private_key(pwd, salt, Scrypt().get_n())
info.address_u160 = acct.get_address().to_array().hex()
info.salt = salt
return info
def create_account_from_private_key(self, label: str, password: str, private_key: str) -> AccountData or None:
"""
This interface is used to create account by providing an encrypted private key and it's decrypt password.
:param label: a label for account.
:param password: a password which is used to decrypt the encrypted private key.
:param private_key: a private key in the form of string.
:return: if succeed, return an AccountData object.
if failed, return a None object.
"""
salt = get_random_hex_str(16)
info = self.create_account_info(label, password, salt, private_key)
for index in range(len(self.wallet_in_mem.accounts)):
if info.address_base58 == self.wallet_in_mem.accounts[index].address:
return self.wallet_in_mem.accounts[index]
return None
def get_account(self, b58_address_or_ontid: str, password: str) -> Account or None:
"""
:param b58_address_or_ontid: a base58 encode address or ontid
:param password: a password which is used to decrypt the encrypted private key.
:return:
"""
if b58_address_or_ontid.startswith(DID_ONT):
for index in range(len(self.wallet_in_mem.identities)):
if self.wallet_in_mem.identities[index].ont_id == b58_address_or_ontid:
addr = self.wallet_in_mem.identities[index].ont_id.replace(did_ont, "")
key = self.wallet_in_mem.identities[index].controls[0].key
salt = base64.b64decode(self.wallet_in_mem.identities[index].controls[0].salt)
private_key = Account.get_gcm_decoded_private_key(key, password, addr, salt,
self.wallet_in_mem.scrypt.get_n(), self.scheme)
return Account(private_key, self.scheme)
else:
for index in range(len(self.wallet_in_mem.accounts)):
if self.wallet_in_mem.accounts[index].address == b58_address_or_ontid:
key = self.wallet_in_mem.accounts[index].key
addr = self.wallet_in_mem.accounts[index].address
salt = base64.b64decode(self.wallet_in_mem.accounts[index].salt)
private_key = Account.get_gcm_decoded_private_key(key, password, addr, salt,
self.wallet_in_mem.scrypt.get_n(), self.scheme)
return Account(private_key, self.scheme)
return None
def get_default_identity(self) -> Identity:
for identity in self.wallet_in_mem.identities:
if identity.is_default:
return identity
raise SDKException(ErrorCode.param_error)
def get_default_account(self) -> AccountData:
"""
This interface is used to get the default account in WalletManager.
:return: an AccountData object that contain all the information of a default account.
"""
for acct in self.wallet_in_mem.accounts:
if acct.is_default:
return acct
raise SDKException(ErrorCode.get_default_account_err)
class WalletManager(object):
def __init__(self,
scheme: SignatureScheme = SignatureScheme.SHA256withECDSA):
if not isinstance(scheme, SignatureScheme):
raise SDKException(
ErrorCode.other_error('Invalid signature scheme.'))
self.scheme = scheme
self.wallet_file = WalletData()
self.wallet_in_mem = WalletData()
self.__wallet_path = ''
@staticmethod
def __check_ont_id(ont_id: str):
if not isinstance(ont_id, str):
raise SDKException(ErrorCode.require_str_params)
if not ont_id.startswith(DID_ONT):
raise SDKException(ErrorCode.invalid_ont_id_format(ont_id))
def open_wallet(self, wallet_path: str = '', is_create: bool = True):
if not isinstance(wallet_path, str):
raise SDKException(ErrorCode.require_str_params)
if wallet_path != '':
self.__wallet_path = wallet_path
if is_create and not path.isfile(self.__wallet_path):
self.create_wallet_file()
if not path.isfile(self.__wallet_path):
raise SDKException(
ErrorCode.invalid_wallet_path(self.__wallet_path))
self.wallet_file = self.load_file()
self.wallet_in_mem = copy.deepcopy(self.wallet_file)
return self.wallet_file
@property
def wallet_path(self):
return self.__wallet_path
@wallet_path.setter
def wallet_path(self, file_path: str):
if not isinstance(file_path, str):
raise SDKException(ErrorCode.require_str_params)
self.__wallet_path = file_path
def del_wallet_file(self):
if path.isfile(self.__wallet_path):
remove(self.__wallet_path)
return True
return False
def get_account_count(self) -> int:
return len(self.wallet_in_mem.accounts)
def create_wallet_file(self, wallet_path: str = ''):
if not isinstance(wallet_path, str):
raise SDKException(ErrorCode.require_str_params)
if wallet_path != '':
self.__wallet_path = wallet_path
if not path.isfile(self.__wallet_path):
self.wallet_in_mem.create_time = datetime.today().strftime(
"%Y-%m-%d %H:%M:%S")
self.save()
else:
raise SDKException(
ErrorCode.other_error('Wallet file has existed.'))
def load_file(self):
with open(self.__wallet_path, 'rb') as f:
content = f.read()
if content.startswith(codecs.BOM_UTF8):
content = content[len(codecs.BOM_UTF8):]
content = content.decode('utf-8')
wallet_dict = json.loads(content)
create_time = wallet_dict.get('createTime', '')
default_id = wallet_dict.get('defaultOntid', '')
default_address = wallet_dict.get('defaultAccountAddress', '')
identities = wallet_dict.get('identities', list())
try:
scrypt_dict = wallet_dict['scrypt']
scrypt_obj = Scrypt(scrypt_dict.get('n', 16384),
scrypt_dict.get('r', 8),
scrypt_dict.get('p', 8),
scrypt_dict.get('dk_len', 64))
wallet = WalletData(wallet_dict['name'],
wallet_dict['version'], create_time,
default_id, default_address, scrypt_obj,
identities, wallet_dict['accounts'])
except KeyError as e:
raise SDKException(
ErrorCode.param_err(f'wallet file format error: {e}.'))
return wallet
def save(self):
try:
with open(self.__wallet_path, 'w') as f:
json.dump(dict(self.wallet_in_mem), f, indent=4)
except FileNotFoundError as e:
raise SDKException(ErrorCode.other_error(e.args[1])) from None
def get_wallet(self):
return self.wallet_in_mem
def get_acct_data_list(self) -> List[AccountData]:
return self.wallet_in_mem.accounts
def write_wallet(self):
self.save()
self.wallet_file = copy.deepcopy(self.wallet_in_mem)
return self.wallet_file
def restore(self):
self.wallet_in_mem = copy.deepcopy(self.wallet_file)
return self.wallet_in_mem
def reset(self):
self.wallet_in_mem = WalletData()
def get_signature_scheme(self):
return self.scheme
def set_signature_scheme(self, scheme):
self.scheme = scheme
def import_identity(self, label: str, encrypted_pri_key: str, pwd: str,
salt: str, b58_address: str) -> Identity:
"""
This interface is used to import identity by providing encrypted private key, password, salt and
base58 encode address which should be correspond to the encrypted private key provided.
:param label: a label for identity.
:param encrypted_pri_key: an encrypted private key in base64 encoding from.
:param pwd: a password which is used to encrypt and decrypt the private key.
:param salt: a salt value which will be used in the process of encrypt private key.
:param b58_address: a base58 encode address which correspond with the encrypted private key provided.
:return: if succeed, an Identity object will be returned.
"""
scrypt_n = Scrypt().n
pri_key = Account.get_gcm_decoded_private_key(encrypted_pri_key, pwd,
b58_address, salt,
scrypt_n, self.scheme)
info = self.__create_identity(label, pwd, salt, pri_key)
for identity in self.wallet_in_mem.identities:
if identity.ont_id == info.ont_id:
return identity
raise SDKException(ErrorCode.other_error('Import identity failed.'))
def create_identity(self, pwd: str, label: str = '') -> Identity:
"""
:param label: a label for identity.
:param pwd: a password which will be used to encrypt and decrypt the private key.
:return: if succeed, an Identity object will be returned.
"""
pri_key = get_random_hex_str(64)
salt = get_random_hex_str(16)
if len(label) == 0 or label is None:
label = uuid.uuid4().hex[0:8]
return self.__create_identity(label, pwd, salt, pri_key)
def __create_identity(self, label: str, pwd: str, salt: str,
private_key: str):
acct = self.__create_account(label, pwd, salt, private_key, False)
ont_id = DID_ONT + acct.get_address_base58()
return self.wallet_in_mem.get_identity_by_ont_id(ont_id)
def create_identity_from_private_key(self, label: str, pwd: str,
private_key: str) -> Identity:
"""
This interface is used to create identity based on given label, password and private key.
:param label: a label for identity.
:param pwd: a password which will be used to encrypt and decrypt the private key.
:param private_key: a private key in the form of string.
:return: if succeed, an Identity object will be returned.
"""
salt = get_random_hex_str(16)
identity = self.__create_identity(label, pwd, salt, private_key)
return identity
def create_account(self, pwd: str, label: str = '') -> Account:
"""
This interface is used to create account based on given password and label.
:param label: a label for account.
:param pwd: a password which will be used to encrypt and decrypt the private key
:return: if succeed, return an data structure which contain the information of a wallet account.
"""
pri_key = get_random_hex_str(64)
salt = get_random_hex_str(16)
if len(label) == 0 or label is None:
label = uuid.uuid4().hex[0:8]
acct = self.__create_account(label, pwd, salt, pri_key, True)
return self.get_account_by_b58_address(acct.get_address_base58(), pwd)
def del_account_by_b58_address(self, b58_address: str):
for acct in self.wallet_in_mem.accounts:
if acct.b58_address == b58_address:
self.wallet_in_mem.accounts.remove(acct)
return
raise SDKException(ErrorCode.other_error(f'{b58_address} not exist.'))
def __create_account(self, label: str, pwd: str, salt: str,
private_key: str, account_flag: bool) -> Account:
account = Account(private_key, self.scheme)
if self.scheme == SignatureScheme.SHA256withECDSA:
acct_data = AccountData()
else:
raise SDKException(ErrorCode.other_error('Scheme type is error.'))
if pwd is not None:
acct_data.key = account.export_gcm_encrypted_private_key(pwd, salt)
else:
acct_data.key = account.get_private_key_hex()
acct_data.b58_address = account.get_address_base58()
if len(label) == 0 or label is None:
label = uuid.uuid4().hex[0:8]
if account_flag:
for memory_acct in self.wallet_in_mem.accounts:
if memory_acct.b58_address == account.get_address_base58():
raise SDKException(
ErrorCode.other_error('Wallet account exists.'))
if len(self.wallet_in_mem.accounts) == 0:
acct_data.is_default = True
self.wallet_in_mem.default_account_address = acct_data.b58_address
acct_data.label = label
acct_data.salt = base64.b64encode(
salt.encode('latin-1')).decode('ascii')
acct_data.public_key = account.get_public_key_hex()
self.wallet_in_mem.accounts.append(acct_data)
else:
for identity in self.wallet_in_mem.identities:
if identity.ont_id == DID_ONT + acct_data.b58_address:
raise SDKException(
ErrorCode.other_error('Wallet identity exists.'))
idt = Identity()
idt.ont_id = DID_ONT + acct_data.b58_address
idt.label = label
if len(self.wallet_in_mem.identities) == 0:
idt.is_default = True
self.wallet_in_mem.default_ont_id = idt.ont_id
ctl = Control(kid='keys-1',
key=acct_data.key,
salt=base64.b64encode(salt.encode()).decode('ascii'),
address=acct_data.b58_address,
public_key=account.get_public_key_hex())
idt.controls.append(ctl)
self.wallet_in_mem.identities.append(idt)
return account
def add_control(self, ont_id: str, password: str):
WalletManager.__check_ont_id(ont_id)
private_key = get_random_hex_str(64)
salt = get_random_hex_str(16)
b64_salt = base64.b64encode(salt.encode('utf-8')).decode('ascii')
account = Account(private_key, self.scheme)
key = account.export_gcm_encrypted_private_key(password, salt)
b58_address = account.get_address_base58()
public_key = account.get_public_key_hex()
ctrl = Control(kid='',
key=key,
salt=b64_salt,
address=b58_address,
public_key=public_key)
identity = self.get_identity_by_ont_id(ont_id)
identity.add_control(ctrl)
def add_control_by_hex_private_key(self, ont_id: str, password: str,
hex_private_key: str) -> Account:
WalletManager.__check_ont_id(ont_id)
if not isinstance(password, str):
raise SDKException(ErrorCode.require_str_params)
if not isinstance(hex_private_key, str):
raise SDKException(ErrorCode.require_str_params)
salt = get_random_hex_str(16)
b64_salt = base64.b64encode(salt.encode('utf-8')).decode('ascii')
account = Account(hex_private_key, self.scheme)
key = account.export_gcm_encrypted_private_key(password, salt)
b58_address = account.get_address_base58()
public_key = account.get_public_key_hex()
ctrl = Control(kid='',
key=key,
salt=b64_salt,
address=b58_address,
public_key=public_key)
identity = self.get_identity_by_ont_id(ont_id)
identity.add_control(ctrl)
return account
def add_control_by_bytes_private_key(self, ont_id: str, password: str,
bytes_private_key: bytes) -> Account:
WalletManager.__check_ont_id(ont_id)
hex_private_key = bytes_private_key.hex()
return self.add_control_by_hex_private_key(ont_id, password,
hex_private_key)
def import_account(self,
label: str,
encrypted_pri_key: str,
pwd: str,
b58_address: str,
b64_salt: str,
n: int = 16384) -> AccountData:
"""
This interface is used to import account by providing account data.
:param label: str, wallet label
:param encrypted_pri_key: str, an encrypted private key in base64 encoding from
:param pwd: str, a password which is used to encrypt and decrypt the private key
:param b58_address: str, a base58 encode wallet address value
:param b64_salt: str, a base64 encode salt value which is used in the encryption of private key
:param n: int, CPU/Memory cost parameter. It must be a power of 2 and less than :math:`2^{32}`
:return:
if succeed, return an data structure which contain the information of a wallet account.
if failed, return a None object.
"""
salt = base64.b64decode(b64_salt.encode('ascii')).decode('latin-1')
private_key = Account.get_gcm_decoded_private_key(
encrypted_pri_key, pwd, b58_address, salt, n, self.scheme)
acct_info = self.create_account_info(label, pwd, salt, private_key)
for acct in self.wallet_in_mem.accounts:
if not isinstance(acct, AccountData):
raise SDKException(
ErrorCode.other_error('Invalid account data in memory.'))
if acct_info.address_base58 == acct.b58_address:
return acct
raise SDKException(ErrorCode.other_error('Import account failed.'))
def create_account_info(self, label: str, pwd: str, salt: str,
private_key: str) -> AccountInfo:
acct = self.__create_account(label, pwd, salt, private_key, True)
info = AccountInfo()
info.address_base58 = Address.address_from_bytes_pubkey(
acct.get_public_key_bytes()).b58encode()
info.public_key = acct.get_public_key_bytes().hex()
info.encrypted_pri_key = acct.export_gcm_encrypted_private_key(
pwd, salt)
info.address_u160 = acct.get_address().to_bytes().hex()
info.salt = salt
return info
def create_account_from_private_key(self,
password: str,
private_key: str,
label: str = '') -> AccountData:
"""
This interface is used to create account by providing an encrypted private key and it's decrypt password.
:param label: a label for account.
:param password: a password which is used to decrypt the encrypted private key.
:param private_key: a private key in the form of string.
:return: if succeed, return an AccountData object.
if failed, return a None object.
"""
salt = get_random_hex_str(16)
if len(label) == 0 or label is None:
label = uuid.uuid4().hex[0:8]
info = self.create_account_info(label, password, salt, private_key)
for acct in self.wallet_in_mem.accounts:
if info.address_base58 == acct.b58_address:
return acct
raise SDKException(
ErrorCode.other_error(
f'Create account from key {private_key} failed.'))
def create_account_from_wif(self,
wif: str,
password: str,
label: str = '') -> Account:
private_key = Account.get_private_key_from_wif(wif).hex()
salt = get_random_hex_str(16)
if len(label) == 0 or label is None:
label = uuid.uuid4().hex[0:8]
info = self.create_account_info(label, password, salt, private_key)
return self.get_account_by_b58_address(info.address_base58, password)
def get_account_by_ont_id(self, ont_id: str, password: str) -> Account:
"""
:param ont_id: OntId.
:param password: a password which is used to decrypt the encrypted private key.
:return:
"""
WalletManager.__check_ont_id(ont_id)
for identity in self.wallet_in_mem.identities:
if identity.ont_id == ont_id:
addr = identity.ont_id.replace(DID_ONT, "")
key = identity.controls[0].key
salt = base64.b64decode(identity.controls[0].salt)
n = self.wallet_in_mem.scrypt.n
private_key = Account.get_gcm_decoded_private_key(
key, password, addr, salt, n, self.scheme)
return Account(private_key, self.scheme)
raise SDKException(
ErrorCode.other_error(f'Get account {ont_id} failed.'))
def get_identity_by_ont_id(self, ont_id: str) -> Identity:
return self.wallet_in_mem.get_identity_by_ont_id(ont_id)
def get_control_info_by_index(self, ont_id: str, index: int) -> Control:
if not isinstance(ont_id, str):
raise SDKException(ErrorCode.require_str_params)
if not ont_id.startswith(DID_ONT):
raise SDKException(ErrorCode.invalid_ont_id_format(ont_id))
identity = self.get_identity_by_ont_id(ont_id)
try:
ctrl = identity.controls[index]
except IndexError:
raise SDKException(
ErrorCode.other_error(
f'Get {ont_id}\'s control account failed.'))
return ctrl
def get_control_account_by_index(self, ont_id: str, index: int,
password: str) -> Account:
if not isinstance(password, str):
raise SDKException(ErrorCode.require_str_params)
ctrl = self.get_control_info_by_index(ont_id, index)
salt = base64.b64decode(ctrl.salt)
key = ctrl.key
b58_address = ctrl.b58_address
n = self.wallet_in_mem.scrypt.n
private_key = Account.get_gcm_decoded_private_key(
key, password, b58_address, salt, n, self.scheme)
return Account(private_key, self.scheme)
def get_control_info_by_b58_address(self, ont_id: str,
b58_address: str) -> Control:
WalletManager.__check_ont_id(ont_id)
identity = self.get_identity_by_ont_id(ont_id)
for ctrl in identity.controls:
if ctrl.b58_address == b58_address:
return ctrl
raise SDKException(
ErrorCode.other_error(f'Get account {b58_address} failed.'))
def get_control_account_by_b58_address(self, ont_id: str, b58_address: str,
password: str) -> Account:
WalletManager.__check_ont_id(ont_id)
ctrl = self.get_control_info_by_b58_address(ont_id, b58_address)
salt = base64.b64decode(ctrl.salt)
key = ctrl.key
b58_address = ctrl.b58_address
n = self.wallet_in_mem.scrypt.n
private_key = Account.get_gcm_decoded_private_key(
key, password, b58_address, salt, n, self.scheme)
return Account(private_key, self.scheme)
def get_account_data_by_b58_address(self, b58_address: str) -> AccountData:
if not isinstance(b58_address, str):
raise SDKException(ErrorCode.require_str_params)
for acct in self.wallet_in_mem.accounts:
if not isinstance(acct, AccountData):
raise SDKException(
ErrorCode.other_error('Invalid account data in memory.'))
if acct.b58_address == b58_address:
return acct
raise SDKException(
ErrorCode.other_error(f'Get account {b58_address} failed.'))
def get_account_by_b58_address(self, b58_address: str,
password: str) -> Account:
"""
:param b58_address: a base58 encode address.
:param password: a password which is used to decrypt the encrypted private key.
:return:
"""
acct = self.get_account_data_by_b58_address(b58_address)
n = self.wallet_in_mem.scrypt.n
salt = base64.b64decode(acct.salt)
private_key = Account.get_gcm_decoded_private_key(
acct.key, password, b58_address, salt, n, self.scheme)
return Account(private_key, self.scheme)
def get_default_identity(self) -> Identity:
for identity in self.wallet_in_mem.identities:
if identity.is_default:
return identity
raise SDKException(ErrorCode.param_error)
def get_default_account_data(self) -> AccountData:
"""
This interface is used to get the default account in WalletManager.
:return: an AccountData object that contain all the information of a default account.
"""
for acct in self.wallet_in_mem.accounts:
if not isinstance(acct, AccountData):
raise SDKException(
ErrorCode.other_error('Invalid account data in memory.'))
if acct.is_default:
return acct
raise SDKException(ErrorCode.get_default_account_err)
def get_default_account(self, password: str) -> Account:
acct = self.get_default_account_data()
salt = base64.b64decode(acct.salt)
n = self.wallet_in_mem.scrypt.n
private_key = Account.get_gcm_decoded_private_key(
acct.key, password, acct.b58_address, salt, n, self.scheme)
return Account(private_key, self.scheme)
class WalletManager(object):
def __init__(self, scheme=SignatureScheme.SHA256withECDSA):
self.scheme = scheme
self.wallet_file = WalletData()
self.wallet_in_mem = WalletData()
self.wallet_path = ""
def open_wallet(self, wallet_path: str):
self.wallet_path = wallet_path
if is_file_exist(wallet_path) is False:
# create a new wallet file
self.wallet_in_mem.createTime = datetime.today().strftime("%Y-%m-%d %H:%M:%S")
self.save()
# wallet file exists now
self.wallet_file = self.load()
self.wallet_in_mem = self.wallet_file
return self.wallet_file
def load(self):
with open(self.wallet_path, "r") as f:
fstr = f.read()
r = json.loads(fstr.replace("enc-", "enc_"), object_hook=lambda d: namedtuple('X', d.keys())(*d.values()))
# r = json.load(f, object_hook=lambda d: namedtuple('X', d.keys())(*d.values()))
scrypt = Scrypt(r.scrypt.n, r.scrypt.r, r.scrypt.p, r.scrypt.dkLen)
identities = []
try:
for index in range(len(r.identities)):
r_identities = r.identities[index]
control = [Control(id=r_identities.controls[0].id,
algorithm=r_identities.controls[0].algorithm,
param=r_identities.controls[0].parameters,
key=r_identities.controls[0].key,
address=r_identities.controls[0].address,
salt=r_identities.controls[0].salt,
enc_alg=r_identities.controls[0].enc_alg,
hash_value=r_identities.controls[0].hash,
public_key=r_identities.controls[0].publicKey)]
identities.append(Identity(r_identities.ont_id, r_identities.label, r_identities.lock, control))
except AttributeError as e:
pass
accounts = []
try:
for index in range(len(r.accounts)):
temp = AccountData(label=r.accounts[index].label, public_key=r.accounts[index].publicKey,
sign_scheme=r.accounts[index].signatureScheme,
isDefault=r.accounts[index].isDefault,
lock=r.accounts[index].lock, address=r.accounts[index].address,
algorithm=r.accounts[index].algorithm, param=r.accounts[index].parameters,
key=r.accounts[index].key, enc_alg=r.accounts[index].enc_alg,
salt=r.accounts[index].salt)
accounts.append(temp)
except AttributeError as e:
pass
default_ont_id = ""
try:
default_ont_id = r.defaultOntid
except AttributeError as e:
pass
default_account_address = ""
try:
default_account_address = r.defaultAccountAddress
except AttributeError as e:
pass
create_time = datetime.today().strftime("%Y-%m-%d %H:%M:%S")
try:
create_time = r.createTime
except Exception as e:
pass
res = WalletData(r.name, r.version, create_time, default_ont_id, default_account_address, scrypt,
identities, accounts)
return res
def save(self):
fstr = json.dumps(self.wallet_in_mem, default=lambda obj: obj.__dict__, sort_keys=True, indent=4)
temp = fstr.replace("enc_", "enc-")
f = open(self.wallet_path, "w")
f.write(temp)
f.close()
def get_wallet(self):
return self.wallet_in_mem
def write_wallet(self):
self.save()
self.wallet_file = self.wallet_in_mem
return self.wallet_file
def reset_wallet(self):
self.wallet_in_mem = self.wallet_file.clone()
return self.wallet_in_mem
def get_signature_scheme(self):
return self.scheme
def set_signature_scheme(self, scheme):
self.scheme = scheme
def import_identity(self, label: str, encrypted_pri_key: str, pwd: str, salt: str, address: str):
pri_key = Account.get_gcm_decoded_private_key(encrypted_pri_key, pwd, address, salt,
Scrypt().get_n(), self.scheme)
info = self.__create_identity(label, pwd, salt, pri_key)
for index in range(len(self.wallet_in_mem.identities)):
if self.wallet_in_mem.identities[index].ont_id == info.ont_id:
return self.wallet_in_mem.identities[index]
return None
def create_identity(self, label: str, pwd: str):
priv_key = get_random_str(64)
salt = get_random_str(16)
return self.__create_identity(label, pwd, salt, priv_key)
def __create_identity(self, label: str, pwd: str, salt: str, private_key: str):
acct = self.__create_account(label, pwd, salt, private_key, False)
info = IdentityInfo()
info.ont_id = did_ont + acct.get_address_base58()
info.pubic_key = acct.serialize_public_key().hex()
info.private_key = acct.serialize_private_key().hex()
info.pri_key_wif = acct.export_wif()
info.encrypted_pri_key = acct.export_gcm_encrypted_private_key(pwd, salt, Scrypt().get_n())
info.address_u160 = acct.get_address().to_array().hex()
return self.wallet_in_mem.get_identity_by_ont_id(info.ont_id)
def create_identity_from_pri_key(self, label: str, pwd: str, private_key: str):
salt = get_random_str(16)
identity = self.__create_identity(label, pwd, salt, private_key)
return identity
def create_account(self, label: str, pwd: str) -> AccountData:
pri_key = get_random_str(64)
salt = get_random_str(16)
account = self.__create_account(label, pwd, salt, pri_key, True)
return self.wallet_file.get_account_by_address(account.get_address_base58())
def __create_account(self, label: str, pwd: str, salt: str, priv_key: str, account_flag: bool):
print(priv_key)
account = Account(priv_key, self.scheme)
# initialization
if self.scheme == SignatureScheme.SHA256withECDSA:
acct = AccountData()
else:
raise ValueError("scheme type is error")
# set key
if pwd is not None:
acct.key = account.export_gcm_encrypted_private_key(pwd, salt, Scrypt().get_n())
else:
acct.key = account.serialize_private_key().hex()
acct.address = account.get_address_base58()
# set label
if label is None or label == "":
label = str(uuid.uuid4())[0:8]
if account_flag:
for index in range(len(self.wallet_in_mem.accounts)):
if acct.address == self.wallet_in_mem.accounts[index].address:
raise ValueError("wallet account exists")
if len(self.wallet_in_mem.accounts) == 0:
acct.isDefault = True
self.wallet_in_mem.defaultAccountAddress = acct.address
acct.label = label
acct.salt = base64.b64encode(salt.encode()).decode('ascii')
acct.publicKey = account.serialize_public_key().hex()
self.wallet_in_mem.accounts.append(acct)
else:
for index in range(len(self.wallet_in_mem.identities)):
if self.wallet_in_mem.identities[index].ont_id == did_ont + acct.address:
raise ValueError("wallet identity exists")
idt = Identity()
idt.ontid = did_ont + acct.address
idt.label = label
if len(self.wallet_in_mem.identities) == 0:
idt.isDefault = True
self.wallet_in_mem.defaultOntid = idt.ontid
ctl = Control(id="keys-1", key=acct.key, salt=base64.b64encode(salt.encode()).decode('ascii'),
address=acct.address,
public_key=account.serialize_public_key().hex())
idt.controls.append(ctl)
self.wallet_in_mem.identities.append(idt)
return account
def import_account(self, label: str, encrypted_pri_key: str, pwd: str, base58_addr: str, base64_salt: str):
salt = base64.b64decode(base64_salt.encode('ascii')).decode('latin-1')
private_key = Account.get_gcm_decoded_private_key(encrypted_pri_key, pwd, base58_addr, salt, Scrypt().get_n(),
self.scheme)
info = self.create_account_info(label, pwd, salt, private_key)
for index in range(len(self.wallet_in_mem.accounts)):
if info.address_base58 == self.wallet_in_mem.accounts[index].address:
return self.wallet_in_mem.accounts[index]
return None
def create_account_info(self, label: str, pwd: str, salt: str, private_key: str):
acct = self.__create_account(label, pwd, salt, private_key, True)
info = AccountInfo()
info.address_base58 = Address.address_from_bytes_pubkey(acct.serialize_public_key()).b58encode()
info.public_key = acct.serialize_public_key().hex()
info.encrypted_pri_key = acct.export_gcm_encrypted_private_key(pwd, salt, Scrypt().get_n())
info.address_u160 = acct.get_address().to_array().hex()
info.salt = salt
return info
def create_account_from_prikey(self, label: str, pwd: str, private_key: str):
salt = get_random_str(16)
info = self.create_account_info(label, pwd, salt, private_key)
for index in range(len(self.wallet_in_mem.accounts)):
if info.address_base58 == self.wallet_in_mem.accounts[index].address:
return self.wallet_in_mem.accounts[index]
return None
def get_account(self, address: str, pwd: str):
for index in range(len(self.wallet_in_mem.accounts)):
if self.wallet_in_mem.accounts[index].address == address:
key = self.wallet_in_mem.accounts[index].key
addr = self.wallet_in_mem.accounts[index].address
salt = base64.b64decode(self.wallet_in_mem.accounts[index].salt)
private_key = Account.get_gcm_decoded_private_key(key, pwd, addr, salt, Scrypt().get_n(), self.scheme)
return Account(private_key, self.scheme)
for index in range(len(self.wallet_in_mem.identities)):
if self.wallet_in_mem.identities[index].ont_id == did_ont + address:
addr = self.wallet_in_mem.identities[index].ont_id.replace(did_ont, "")
key = self.wallet_in_mem.identities[index].controls[0].key
salt = base64.b64decode(self.wallet_in_mem.identities[index].controls[0].salt)
private_key = Account.get_gcm_decoded_private_key(key, pwd, addr, salt, Scrypt().get_n(), self.scheme)
return Account(private_key, self.scheme)
return None
def get_default_identity(self) -> Identity:
for identity in self.wallet_in_mem.identities:
if identity.isDefault:
return identity
raise SDKException(ErrorCode.param_error)
def get_default_account(self) -> AccountData:
for acct in self.wallet_in_mem.accounts:
if acct.isDefault:
return acct
raise SDKException(ErrorCode.get_default_account_err)