|
@@ -21,15 +21,15 @@ tx.sign: Sign a transaction generated by 'mmgen-txcreate'
|
|
|
"""
|
|
|
|
|
|
from ..cfg import gc
|
|
|
-from ..util import msg,suf,fmt,die,remove_dups,get_extension
|
|
|
+from ..util import msg, suf, fmt, die, remove_dups, get_extension
|
|
|
from ..obj import MMGenList
|
|
|
from ..addr import MMGenAddrType
|
|
|
-from ..addrlist import AddrIdxList,KeyAddrList
|
|
|
-from ..wallet import Wallet,get_wallet_extensions,get_wallet_cls
|
|
|
+from ..addrlist import AddrIdxList, KeyAddrList
|
|
|
+from ..wallet import Wallet, get_wallet_extensions, get_wallet_cls
|
|
|
|
|
|
saved_seeds = {}
|
|
|
|
|
|
-def get_seed_for_seed_id(sid,infiles,saved_seeds):
|
|
|
+def get_seed_for_seed_id(sid, infiles, saved_seeds):
|
|
|
|
|
|
if sid in saved_seeds:
|
|
|
return saved_seeds[sid]
|
|
@@ -39,7 +39,7 @@ def get_seed_for_seed_id(sid,infiles,saved_seeds):
|
|
|
if infiles:
|
|
|
seed = Wallet(cfg, infiles.pop(0), ignore_in_fmt=True, passwd_file=global_passwd_file).seed
|
|
|
elif subseeds_checked is False:
|
|
|
- seed = saved_seeds[list(saved_seeds)[0]].subseed_by_seed_id(sid,print_msg=True)
|
|
|
+ seed = saved_seeds[list(saved_seeds)[0]].subseed_by_seed_id(sid, print_msg=True)
|
|
|
subseeds_checked = True
|
|
|
if not seed:
|
|
|
continue
|
|
@@ -48,23 +48,23 @@ def get_seed_for_seed_id(sid,infiles,saved_seeds):
|
|
|
seed = Wallet(cfg, passwd_file=global_passwd_file).seed
|
|
|
msg(f'User input produced Seed ID {seed.sid}')
|
|
|
if not seed.sid == sid: # TODO: add test
|
|
|
- seed = seed.subseed_by_seed_id(sid,print_msg=True)
|
|
|
+ seed = seed.subseed_by_seed_id(sid, print_msg=True)
|
|
|
|
|
|
if seed:
|
|
|
saved_seeds[seed.sid] = seed
|
|
|
if seed.sid == sid:
|
|
|
return seed
|
|
|
else:
|
|
|
- die(2,f'ERROR: No seed source found for Seed ID: {sid}')
|
|
|
+ die(2, f'ERROR: No seed source found for Seed ID: {sid}')
|
|
|
|
|
|
-def generate_kals_for_mmgen_addrs(need_keys,infiles,saved_seeds,proto):
|
|
|
+def generate_kals_for_mmgen_addrs(need_keys, infiles, saved_seeds, proto):
|
|
|
mmids = [e.mmid for e in need_keys]
|
|
|
- sids = remove_dups((i.sid for i in mmids),quiet=True)
|
|
|
+ sids = remove_dups((i.sid for i in mmids), quiet=True)
|
|
|
cfg._util.vmsg(f"Need seed{suf(sids)}: {' '.join(sids)}")
|
|
|
def gen_kals():
|
|
|
for sid in sids:
|
|
|
# Returns only if seed is found
|
|
|
- seed = get_seed_for_seed_id(sid,infiles,saved_seeds)
|
|
|
+ seed = get_seed_for_seed_id(sid, infiles, saved_seeds)
|
|
|
for id_str in MMGenAddrType.mmtypes:
|
|
|
idx_list = [i.idx for i in mmids if i.sid == sid and i.mmtype == id_str]
|
|
|
if idx_list:
|
|
@@ -73,21 +73,21 @@ def generate_kals_for_mmgen_addrs(need_keys,infiles,saved_seeds,proto):
|
|
|
proto = proto,
|
|
|
seed = seed,
|
|
|
addr_idxs = AddrIdxList(idx_list=idx_list),
|
|
|
- mmtype = MMGenAddrType(proto,id_str),
|
|
|
- skip_chksum = True )
|
|
|
+ mmtype = MMGenAddrType(proto, id_str),
|
|
|
+ skip_chksum = True)
|
|
|
return MMGenList(gen_kals())
|
|
|
|
|
|
-def add_keys(tx,src,infiles=None,saved_seeds=None,keyaddr_list=None):
|
|
|
- need_keys = [e for e in getattr(tx,src) if e.mmid and not e.have_wif]
|
|
|
+def add_keys(tx, src, infiles=None, saved_seeds=None, keyaddr_list=None):
|
|
|
+ need_keys = [e for e in getattr(tx, src) if e.mmid and not e.have_wif]
|
|
|
if not need_keys:
|
|
|
return []
|
|
|
- desc,src_desc = (
|
|
|
- ('key-address file','From key-address file:') if keyaddr_list else
|
|
|
- ('seed(s)','Generated from seed:') )
|
|
|
+ desc, src_desc = (
|
|
|
+ ('key-address file', 'From key-address file:') if keyaddr_list else
|
|
|
+ ('seed(s)', 'Generated from seed:'))
|
|
|
cfg._util.qmsg(f'Checking {gc.proj_name} -> {tx.proto.coin} address mappings for {src} (from {desc})')
|
|
|
d = (
|
|
|
MMGenList([keyaddr_list]) if keyaddr_list else
|
|
|
- generate_kals_for_mmgen_addrs(need_keys,infiles,saved_seeds,tx.proto) )
|
|
|
+ generate_kals_for_mmgen_addrs(need_keys, infiles, saved_seeds, tx.proto))
|
|
|
new_keys = []
|
|
|
for e in need_keys:
|
|
|
for kal in d:
|
|
@@ -99,7 +99,7 @@ def add_keys(tx,src,infiles=None,saved_seeds=None,keyaddr_list=None):
|
|
|
if src == 'inputs':
|
|
|
new_keys.append(f)
|
|
|
else:
|
|
|
- die(3,fmt(f"""
|
|
|
+ die(3, fmt(f"""
|
|
|
{gc.proj_name} -> {tx.proto.coin} address mappings differ!
|
|
|
{src_desc:<23} {mmid} -> {f.addr}
|
|
|
{'tx file:':<23} {e.mmid} -> {e.addr}
|
|
@@ -108,37 +108,37 @@ def add_keys(tx,src,infiles=None,saved_seeds=None,keyaddr_list=None):
|
|
|
cfg._util.vmsg(f'Added {len(new_keys)} wif key{suf(new_keys)} from {desc}')
|
|
|
return new_keys
|
|
|
|
|
|
-def _pop_matching_fns(args,cmplist): # strips found args
|
|
|
+def _pop_matching_fns(args, cmplist): # strips found args
|
|
|
return list(reversed([args.pop(args.index(a)) for a in reversed(args) if get_extension(a) in cmplist]))
|
|
|
|
|
|
def get_tx_files(cfg, args):
|
|
|
from .unsigned import Unsigned, AutomountUnsigned
|
|
|
ret = _pop_matching_fns(args, [(AutomountUnsigned if cfg.autosign else Unsigned).ext])
|
|
|
if not ret:
|
|
|
- die(1,'You must specify a raw transaction file!')
|
|
|
+ die(1, 'You must specify a raw transaction file!')
|
|
|
return ret
|
|
|
|
|
|
-def get_seed_files(cfg,args):
|
|
|
+def get_seed_files(cfg, args):
|
|
|
# favor unencrypted seed sources first, as they don't require passwords
|
|
|
- ret = _pop_matching_fns( args, get_wallet_extensions('unenc') )
|
|
|
+ ret = _pop_matching_fns(args, get_wallet_extensions('unenc'))
|
|
|
from ..filename import find_file_in_dir
|
|
|
- wf = find_file_in_dir(get_wallet_cls('mmgen'),cfg.data_dir) # Make this the first encrypted ss in the list
|
|
|
+ wf = find_file_in_dir(get_wallet_cls('mmgen'), cfg.data_dir) # Make this the first encrypted ss in the list
|
|
|
if wf:
|
|
|
ret.append(wf)
|
|
|
- ret += _pop_matching_fns( args, get_wallet_extensions('enc') )
|
|
|
+ ret += _pop_matching_fns(args, get_wallet_extensions('enc'))
|
|
|
if not (ret or cfg.mmgen_keys_from_file or cfg.keys_from_file): # or cfg.use_wallet_dat
|
|
|
- die(1,'You must specify a seed or key source!')
|
|
|
+ die(1, 'You must specify a seed or key source!')
|
|
|
return ret
|
|
|
|
|
|
-def get_keyaddrlist(cfg,proto):
|
|
|
+def get_keyaddrlist(cfg, proto):
|
|
|
if cfg.mmgen_keys_from_file:
|
|
|
- return KeyAddrList( cfg, proto, cfg.mmgen_keys_from_file )
|
|
|
+ return KeyAddrList(cfg, proto, cfg.mmgen_keys_from_file)
|
|
|
return None
|
|
|
|
|
|
def get_keylist(cfg):
|
|
|
if cfg.keys_from_file:
|
|
|
from ..fileutil import get_lines_from_file
|
|
|
- return get_lines_from_file( cfg, cfg.keys_from_file, 'key-address data', trim_comments=True )
|
|
|
+ return get_lines_from_file(cfg, cfg.keys_from_file, 'key-address data', trim_comments=True)
|
|
|
return None
|
|
|
|
|
|
async def txsign(cfg_parm, tx, seed_files, kl, kal, tx_num_str='', passwd_file=None):
|
|
@@ -151,39 +151,39 @@ async def txsign(cfg_parm, tx, seed_files, kl, kal, tx_num_str='', passwd_file=N
|
|
|
global_passwd_file = passwd_file
|
|
|
|
|
|
if non_mmaddrs:
|
|
|
- tx.check_non_mmgen_inputs(caller='txsign',non_mmaddrs=non_mmaddrs)
|
|
|
+ tx.check_non_mmgen_inputs(caller='txsign', non_mmaddrs=non_mmaddrs)
|
|
|
tmp = KeyAddrList(
|
|
|
cfg = cfg,
|
|
|
proto = tx.proto,
|
|
|
addrlist = non_mmaddrs,
|
|
|
- skip_chksum = True )
|
|
|
+ skip_chksum = True)
|
|
|
if kl:
|
|
|
tmp.add_wifs(kl)
|
|
|
missing = tmp.list_missing('sec')
|
|
|
if missing:
|
|
|
sep = '\n '
|
|
|
- die(2,'ERROR: a key file must be supplied for the following non-{} address{}:{}'.format(
|
|
|
+ die(2, 'ERROR: a key file must be supplied for the following non-{} address{}:{}'.format(
|
|
|
gc.proj_name,
|
|
|
- suf(missing,'es'),
|
|
|
- sep + sep.join(missing) ))
|
|
|
+ suf(missing, 'es'),
|
|
|
+ sep + sep.join(missing)))
|
|
|
keys += tmp.data
|
|
|
|
|
|
if cfg.mmgen_keys_from_file:
|
|
|
- keys += add_keys(tx,'inputs',keyaddr_list=kal)
|
|
|
- add_keys(tx,'outputs',keyaddr_list=kal)
|
|
|
+ keys += add_keys(tx, 'inputs', keyaddr_list=kal)
|
|
|
+ add_keys(tx, 'outputs', keyaddr_list=kal)
|
|
|
|
|
|
- keys += add_keys(tx,'inputs',seed_files,saved_seeds)
|
|
|
- add_keys(tx,'outputs',seed_files,saved_seeds)
|
|
|
+ keys += add_keys(tx, 'inputs', seed_files, saved_seeds)
|
|
|
+ add_keys(tx, 'outputs', seed_files, saved_seeds)
|
|
|
|
|
|
# this (boolean) attr isn't needed in transaction file
|
|
|
- tx.delete_attrs('inputs','have_wif')
|
|
|
- tx.delete_attrs('outputs','have_wif')
|
|
|
+ tx.delete_attrs('inputs', 'have_wif')
|
|
|
+ tx.delete_attrs('outputs', 'have_wif')
|
|
|
|
|
|
extra_sids = remove_dups(
|
|
|
(s for s in saved_seeds if s not in tx.get_sids('inputs') + tx.get_sids('outputs')),
|
|
|
- quiet = True )
|
|
|
+ quiet = True)
|
|
|
|
|
|
if extra_sids:
|
|
|
msg(f"Unused Seed ID{suf(extra_sids)}: {' '.join(extra_sids)}")
|
|
|
|
|
|
- return await tx.sign(tx_num_str,keys) # returns signed TX object or False
|
|
|
+ return await tx.sign(tx_num_str, keys) # returns signed TX object or False
|