From 1824bef1dec2cc271fba4f4047dd2f02d168ed1f Mon Sep 17 00:00:00 2001 From: mbkma Date: Thu, 19 Aug 2021 09:35:23 +0200 Subject: make the collection menu item removeable --- Mozo/MenuEditor.py | 29 ++++++++++++----------------- 1 file changed, 12 insertions(+), 17 deletions(-) diff --git a/Mozo/MenuEditor.py b/Mozo/MenuEditor.py index 947a2b7..892fab1 100644 --- a/Mozo/MenuEditor.py +++ b/Mozo/MenuEditor.py @@ -268,7 +268,6 @@ class MenuEditor(object): item_type = item_iter.next() def getContents(self, item): - class CollectionDirectoryException(Exception): pass contents = [] item_iter = item.iter() @@ -276,22 +275,18 @@ class MenuEditor(object): while item_type != MateMenu.TreeItemType.INVALID: item = None - try: - if item_type == MateMenu.TreeItemType.DIRECTORY: - item = item_iter.get_directory() - if item.get_menu_id() == 'Collection': raise CollectionDirectoryException() - elif item_type == MateMenu.TreeItemType.ENTRY: - item = item_iter.get_entry() - elif item_type == MateMenu.TreeItemType.HEADER: - item = item_iter.get_header() - elif item_type == MateMenu.TreeItemType.ALIAS: - item = item_iter.get_alias() - elif item_type == MateMenu.TreeItemType.SEPARATOR: - item = item_iter.get_separator() - if item: - contents.append(item) - except CollectionDirectoryException: - pass + if item_type == MateMenu.TreeItemType.DIRECTORY: + item = item_iter.get_directory() + elif item_type == MateMenu.TreeItemType.ENTRY: + item = item_iter.get_entry() + elif item_type == MateMenu.TreeItemType.HEADER: + item = item_iter.get_header() + elif item_type == MateMenu.TreeItemType.ALIAS: + item = item_iter.get_alias() + elif item_type == MateMenu.TreeItemType.SEPARATOR: + item = item_iter.get_separator() + if item: + contents.append(item) item_type = item_iter.next() return contents -- cgit v1.2.1