From cc7b8ad78f600c67157a2af236301a63e32dc4f5 Mon Sep 17 00:00:00 2001 From: The MMGen Project Date: Fri, 28 Apr 2023 11:23:24 +0000 Subject: [PATCH] move `get_autosign_obj()` to autosign.py --- mmgen/autosign.py | 9 +++++++++ mmgen/main_xmrwallet.py | 4 ++-- mmgen/xmrwallet.py | 21 ++++++--------------- 3 files changed, 17 insertions(+), 17 deletions(-) diff --git a/mmgen/autosign.py b/mmgen/autosign.py index d7bb4d24..5c0b94d9 100755 --- a/mmgen/autosign.py +++ b/mmgen/autosign.py @@ -558,3 +558,12 @@ class Autosign: enabled = self.cfg.led, simulate = os.getenv('MMGEN_TEST_SUITE_AUTOSIGN_LED_SIMULATE') ) self.led.set('off') + +def get_autosign_obj(cfg,coins=None): + return Autosign( + AutosignConfig({ + 'mountpoint': cfg.autosign_mountpoint or cfg.mountpoint, + 'test_suite': cfg.test_suite, + 'coins': coins if isinstance(coins,str) else ','.join(coins) if coins else 'btc', + }) + ) diff --git a/mmgen/main_xmrwallet.py b/mmgen/main_xmrwallet.py index 323797c1..7ba0f365 100755 --- a/mmgen/main_xmrwallet.py +++ b/mmgen/main_xmrwallet.py @@ -30,7 +30,6 @@ from .xmrwallet import ( MoneroWalletOps, xmrwallet_uarg_info, xmrwallet_uargs, - get_autosign_obj, ) opts_data = { @@ -148,7 +147,8 @@ elif op in ('export-outputs','import-key-images'): wallets = cmd_args.pop(0) if cmd_args else None if cfg.autosign and not cfg.test_suite: - asi = get_autosign_obj(cfg) + from .autosign import get_autosign_obj + asi = get_autosign_obj(cfg,'xmr') if not asi.get_insert_status(): die(1,'Removable device not present!') asi.do_mount() diff --git a/mmgen/xmrwallet.py b/mmgen/xmrwallet.py index 36f3d003..90e6d32a 100755 --- a/mmgen/xmrwallet.py +++ b/mmgen/xmrwallet.py @@ -54,6 +54,7 @@ from .rpc import json_encoder from .proto.xmr.rpc import MoneroRPCClient,MoneroWalletRPCClient from .proto.xmr.daemon import MoneroWalletDaemon from .ui import keypress_confirm +from .autosign import get_autosign_obj xmrwallet_uargs = namedtuple('xmrwallet_uargs',[ 'infile', @@ -74,16 +75,6 @@ xmrwallet_uarg_info = ( r'(?:[^:]+):(?:\d+)' ) -def get_autosign_obj(cfg): - from .autosign import Autosign,AutosignConfig - return Autosign( - AutosignConfig({ - 'mountpoint': cfg.autosign_mountpoint, - 'test_suite': cfg.test_suite, - 'coins': 'XMR', - }) - ) - class XMRWalletAddrSpec(str,Hilite,InitErrors,MMGenObject): color = 'cyan' width = 0 @@ -267,7 +258,7 @@ class MoneroMMGenTX: ) if self.cfg.autosign: - fn = get_autosign_obj(self.cfg).xmr_tx_dir / fn + fn = get_autosign_obj(self.cfg,'xmr').xmr_tx_dir / fn from .fileutil import write_data_to_file write_data_to_file( @@ -450,7 +441,7 @@ class MoneroWalletOutputsFile: def get_outfile(self,cfg,wallet_fn): return ( - get_autosign_obj(cfg).xmr_outputs_dir if cfg.autosign else + get_autosign_obj(cfg,'xmr').xmr_outputs_dir if cfg.autosign else wallet_fn.parent ) / self.fn_fs.format( a = wallet_fn.name, b = self.base_chksum, @@ -506,7 +497,7 @@ class MoneroWalletOutputsFile: @classmethod def find_fn_from_wallet_fn(cls,cfg,wallet_fn,ret_on_no_match=False): - path = get_autosign_obj(cfg).xmr_outputs_dir or Path() + path = get_autosign_obj(cfg,'xmr').xmr_outputs_dir or Path() pat = cls.fn_fs.format( a = wallet_fn.name, b = f'[0-9a-f]{{{cls.chksum_nchars}}}\\', @@ -805,7 +796,7 @@ class MoneroWalletOps: @property def autosign_viewkey_addr_file(self): from .addrfile import ViewKeyAddrFile - mpdir = get_autosign_obj(self.cfg).xmr_dir + mpdir = get_autosign_obj(self.cfg,'xmr').xmr_dir flist = [f for f in mpdir.iterdir() if f.name.endswith(ViewKeyAddrFile.ext)] if len(flist) != 1: die(2, @@ -1618,7 +1609,7 @@ class MoneroWalletOps: @property def unsubmitted_tx_path(self): from .autosign import Signable - t = Signable.xmr_transaction( get_autosign_obj(self.cfg) ) + t = Signable.xmr_transaction( get_autosign_obj(self.cfg,'xmr') ) if len(t.unsubmitted) != 1: die('AutosignTXError', "{a} unsubmitted transaction{b} in '{c}'!".format( a = 'More than one' if t.unsubmitted else 'No',