pySim-shell: Use iteration when unregistering command sets

This makes it more easy to extend in the future

Change-Id: I53ecf03d330fe1eb80ee920e6f9c8185f9be48fd
This commit is contained in:
Harald Welte 2021-10-20 14:24:41 +02:00
parent ec95053249
commit 0bc5326b76
1 changed files with 4 additions and 6 deletions

View File

@ -138,10 +138,8 @@ class PysimApp(cmd2.Cmd):
# Unequip everything from pySim-shell that would not work in unequipped state # Unequip everything from pySim-shell that would not work in unequipped state
if self.rs: if self.rs:
self.rs.unregister_cmds(self) self.rs.unregister_cmds(self)
cmd_set = self.find_commandsets(Iso7816Commands) for cmds in [Iso7816Commands, PySimCommands]:
if cmd_set: cmd_set = self.find_commandsets(cmds)
self.unregister_command_set(cmd_set[0])
cmd_set = self.find_commandsets(PySimCommands)
if cmd_set: if cmd_set:
self.unregister_command_set(cmd_set[0]) self.unregister_command_set(cmd_set[0])