Browse Source

minor cleanups (8 files)

The MMGen Project 3 months ago
parent
commit
220dc5a31d

+ 1 - 1
mmgen/addrlist.py

@@ -34,7 +34,7 @@ class AddrIdxList(tuple, InitErrors, MMGenObject):
 		try:
 		try:
 			if fmt_str:
 			if fmt_str:
 				def gen():
 				def gen():
-					for i in (fmt_str.split(sep)):
+					for i in fmt_str.split(sep):
 						match [int(x) for x in i.split('-')]:
 						match [int(x) for x in i.split('-')]:
 							case [a]:
 							case [a]:
 								yield a
 								yield a

+ 1 - 2
mmgen/cfgfile.py

@@ -228,8 +228,7 @@ class CfgFileSampleUsr(cfg_file_sample):
 
 
 	def parse_metadata(self):
 	def parse_metadata(self):
 		if self.data:
 		if self.data:
-			m = re.match(r'# Version (\d+) ([a-f0-9]{40})$', self.data[-1])
-			if m:
+			if m := re.match(r'# Version (\d+) ([a-f0-9]{40})$', self.data[-1]):
 				self.ver = m[1]
 				self.ver = m[1]
 				self.chksum = m[2]
 				self.chksum = m[2]
 				self.data = self.data[:-1] # remove metadata line
 				self.data = self.data[:-1] # remove metadata line

+ 3 - 4
mmgen/proto/bch/cashaddr.py

@@ -51,8 +51,8 @@ def PolyMod(v):
 
 
 def parse_ver_byte(ver):
 def parse_ver_byte(ver):
 	assert not (ver >> 7), 'invalid version byte: most-significant bit must be zero'
 	assert not (ver >> 7), 'invalid version byte: most-significant bit must be zero'
-	t = namedtuple('parsed_version_byte', ['addr_type', 'bitlen'])
-	return t(addr_types_rev[ver >> 3], data_sizes[ver & 7])
+	return namedtuple('parsed_version_byte', ['addr_type', 'bitlen'])(
+		addr_types_rev[ver >> 3], data_sizes[ver & 7])
 
 
 def make_ver_byte(addr_type, bitlen):
 def make_ver_byte(addr_type, bitlen):
 	assert addr_type in addr_types_rev, f'{addr_type}: invalid addr type'
 	assert addr_type in addr_types_rev, f'{addr_type}: invalid addr type'
@@ -66,8 +66,7 @@ def make_polymod_vec(pfx, payload_vec):
 	return ([ord(c) & 31  for c in pfx] + [0] + payload_vec)
 	return ([ord(c) & 31  for c in pfx] + [0] + payload_vec)
 
 
 def cashaddr_parse_addr(addr):
 def cashaddr_parse_addr(addr):
-	t = namedtuple('parsed_cashaddr', ['pfx', 'payload'])
-	return t(*addr.split(':', 1))
+	return namedtuple('parsed_cashaddr', ['pfx', 'payload'])(*addr.split(':', 1))
 
 
 def cashaddr_encode_addr(addr_type, size, pfx, data):
 def cashaddr_encode_addr(addr_type, size, pfx, data):
 	t = namedtuple('encoded_cashaddr', ['addr', 'pfx', 'payload'])
 	t = namedtuple('encoded_cashaddr', ['addr', 'pfx', 'payload'])

+ 7 - 7
mmgen/proto/btc/tw/txhistory.py

@@ -297,15 +297,15 @@ class BitcoinTwTxHistory(BitcoinTwView, TwTxHistory, BitcoinTwRPC):
 		def gen_parsed_data():
 		def gen_parsed_data():
 			for o in rpc_data:
 			for o in rpc_data:
 				if lbl_id in o:
 				if lbl_id in o:
-					l = get_tw_label(self.proto, o[lbl_id])
+					lbl = get_tw_label(self.proto, o[lbl_id])
+					yield o | {
+						'twmmid': lbl.mmid,
+						'comment': lbl.comment or ''}
 				else:
 				else:
 					assert o['category'] == 'send', f"{o['address']}: {o['category']} != 'send'"
 					assert o['category'] == 'send', f"{o['address']}: {o['category']} != 'send'"
-					l = None
-				o.update({
-					'twmmid': l.mmid if l else None,
-					'comment': (l.comment or '') if l else None,
-				})
-				yield o
+					yield o | {
+						'twmmid': None,
+						'comment': None}
 
 
 		data = list(gen_parsed_data())
 		data = list(gen_parsed_data())
 
 

+ 1 - 2
mmgen/tw/addresses.py

@@ -12,7 +12,7 @@
 tw.addresses: Tracking wallet listaddresses class for the MMGen suite
 tw.addresses: Tracking wallet listaddresses class for the MMGen suite
 """
 """
 
 
-from ..util import msg, is_int
+from ..util import msg, is_int, die
 from ..obj import MMGenListItem, ImmutableAttr, ListItemAttr, TwComment, NonNegativeInt
 from ..obj import MMGenListItem, ImmutableAttr, ListItemAttr, TwComment, NonNegativeInt
 from ..addr import CoinAddr, MMGenID, MMGenAddrType
 from ..addr import CoinAddr, MMGenID, MMGenAddrType
 from ..amt import CoinAmtChk
 from ..amt import CoinAmtChk
@@ -113,7 +113,6 @@ class TwAddresses(TwView):
 		if mmgen_addrs:
 		if mmgen_addrs:
 			a = mmgen_addrs.rsplit(':', 1)
 			a = mmgen_addrs.rsplit(':', 1)
 			if len(a) != 2:
 			if len(a) != 2:
-				from ..util import die
 				die(1,
 				die(1,
 					f'{mmgen_addrs}: invalid address list argument ' +
 					f'{mmgen_addrs}: invalid address list argument ' +
 					'(must be in form <seed ID>:[<type>:]<idx list>)')
 					'(must be in form <seed ID>:[<type>:]<idx list>)')

+ 1 - 1
mmgen/wallet/__init__.py

@@ -148,7 +148,7 @@ def Wallet(
 				die(1, f'{in_fmt}: --in-fmt parameter does not match extension of input file')
 				die(1, f'{in_fmt}: --in-fmt parameter does not match extension of input file')
 			me = _get_me(wd.type)
 			me = _get_me(wd.type)
 		else:
 		else:
-			fn = ','.join(cfg.hidden_incog_input_params.split(',')[:-1]) # permit comma in filename
+			fn = cfg.hidden_incog_input_params.rsplit(',', 1)[0] # permit comma in filename
 			me = _get_me('incog_hidden')
 			me = _get_me('incog_hidden')
 		from ..filename import MMGenFile
 		from ..filename import MMGenFile
 		me.infile = MMGenFile(fn, subclass=type(me))
 		me.infile = MMGenFile(fn, subclass=type(me))

+ 2 - 2
mmgen/wallet/brain.py

@@ -25,8 +25,8 @@ class wallet(wallet):
 
 
 	def get_bw_params(self):
 	def get_bw_params(self):
 		# already checked
 		# already checked
-		a = self.cfg.brain_params.split(',')
-		return int(a[0]), a[1]
+		a, b = self.cfg.brain_params.split(',', 1)
+		return int(a), b
 
 
 	def _deformat(self):
 	def _deformat(self):
 		self.brainpasswd = ' '.join(self.fmt_data.split())
 		self.brainpasswd = ' '.join(self.fmt_data.split())

+ 3 - 2
mmgen/wallet/incog_hidden.py

@@ -47,8 +47,9 @@ class wallet(wallet):
 	}
 	}
 
 
 	def _get_hincog_params(self, wtype):
 	def _get_hincog_params(self, wtype):
-		a = getattr(self.cfg, 'hidden_incog_'+ wtype +'_params').split(',')
-		return ','.join(a[:-1]), int(a[-1]) # permit comma in filename
+		# permit comma in filename:
+		fn, offset = getattr(self.cfg, f'hidden_incog_{wtype}_params').rsplit(',', 1)
+		return fn, int(offset)
 
 
 	def _check_valid_offset(self, fn, action):
 	def _check_valid_offset(self, fn, action):
 		d = self.ssdata
 		d = self.ssdata