mmgen/mmgen-wallet
1
Fork 0
Code Releases 28 Wiki Activity

TwView.item_action: add acct_methods attribute

Browse source
This commit is contained in:
The MMGen Project 2026-01-21 09:06:50 +00:00
commit caade42c16
Signed by: mmgen
GPG key ID: 3F8B1861E32B7DA2
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
mmgen/proto/xmr/tw/view.py
View file

@ -210,8 +210,10 @@ class MoneroTwView:
color_override = None if d.amt == d.unlocked_amt else 'orange',
prec = self.disp_prec))
async def get_idx_from_user(self):
async def get_idx_from_user(self, method_name):
if res := await self.get_idx(f'{self.item_desc} number', self.accts_data):
if method_name in self.item_action.acct_methods:
return res
return await self.get_idx(
'address index',
self.accts_data[res.idx - 1].data,

5
mmgen/tw/view.py
View file

@ -657,7 +657,7 @@ class TwView(MMGenObject, metaclass=AsyncInit):
msg_r('\r'+''.ljust(self.term_width)+'\r'+yellow('Canceling! '))
return False
async def get_idx_from_user(self):
async def get_idx_from_user(self, method_name):
return await self.get_idx(f'{self.item_desc} number', self.disp_data)
async def get_idx(self, desc, data, *, is_addr_idx=False):
@ -748,6 +748,7 @@ class TwView(MMGenObject, metaclass=AsyncInit):
do_pager(await parent.format('detail', color=True))
class item_action:
acct_methods = ()
@enable_echo
async def run(self, parent, action_method):
@ -762,7 +763,7 @@ class TwView(MMGenObject, metaclass=AsyncInit):
# None: action aborted by user or no action performed
# 'redo': user will be re-prompted for item number
# 'redraw': action successfully performed, screen will be redrawn
if usr_ret := await parent.get_idx_from_user():
if usr_ret := await parent.get_idx_from_user(action_method.__name__):
ret = await action_method(parent, usr_ret.idx, usr_ret.acct_addr_idx)
else:
ret = None