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