Improved remove behaviour

This commit is contained in:
Philip Trauner 2016-11-16 18:53:15 +01:00
parent 003bf3aa84
commit 203d62db0d

View file

@ -109,8 +109,11 @@ class Menu:
def remove(self, entry):
if entry.__class__ == Entry:
if entry in self.entries.values():
found_entry_id = None
for entry_id in self.entries:
if self.entries[entry_id] == entry:
del self.entries[entry_id]
found_entry_id = entry_id
if found_entry_id != None:
del self.entries[entry_id]
else:
raise TypeError("invalid type supplied")