summaryrefslogtreecommitdiff
path: root/Mozo
diff options
context:
space:
mode:
Diffstat (limited to 'Mozo')
-rw-r--r--Mozo/MainWindow.py1228
-rw-r--r--Mozo/MenuEditor.py1494
-rw-r--r--Mozo/util.py252
3 files changed, 1487 insertions, 1487 deletions
diff --git a/Mozo/MainWindow.py b/Mozo/MainWindow.py
index 63231df..75533fa 100644
--- a/Mozo/MainWindow.py
+++ b/Mozo/MainWindow.py
@@ -29,623 +29,623 @@ import subprocess
import shutil
import urllib
try:
- from Mozo import config
- gettext.bindtextdomain(config.GETTEXT_PACKAGE,config.localedir)
- gettext.textdomain(config.GETTEXT_PACKAGE)
- GETTEXT_PACKAGE = config.GETTEXT_PACKAGE
+ from Mozo import config
+ gettext.bindtextdomain(config.GETTEXT_PACKAGE,config.localedir)
+ gettext.textdomain(config.GETTEXT_PACKAGE)
+ GETTEXT_PACKAGE = config.GETTEXT_PACKAGE
except:
- GETTEXT_PACKAGE = "mozo"
+ GETTEXT_PACKAGE = "mozo"
_ = gettext.gettext
from Mozo.MenuEditor import MenuEditor
from Mozo import util
class MainWindow:
- timer = None
- #hack to make editing menu properties work
- allow_update = True
- #drag-and-drop stuff
- dnd_items = [('MOZO_ITEM_ROW', Gtk.TargetFlags.SAME_APP, 0), ('text/plain', 0, 1)]
- dnd_menus = [('MOZO_MENU_ROW', Gtk.TargetFlags.SAME_APP, 0)]
- dnd_both = [dnd_items[0],] + dnd_menus
- drag_data = None
- edit_pool = []
-
- def __init__(self, datadir, version, argv):
- self.file_path = datadir
- self.version = version
- self.editor = MenuEditor()
- Gtk.Window.set_default_icon_name('mozo')
- self.tree = Gtk.Builder()
- self.tree.set_translation_domain(GETTEXT_PACKAGE)
- self.tree.add_from_file(os.path.join(self.file_path, 'mozo.ui'))
- self.tree.connect_signals(self)
- self.setupMenuTree()
- self.setupItemTree()
- self.tree.get_object('edit_delete').set_sensitive(False)
- self.tree.get_object('edit_revert_to_original').set_sensitive(False)
- self.tree.get_object('edit_properties').set_sensitive(False)
- self.tree.get_object('move_up_button').set_sensitive(False)
- self.tree.get_object('move_down_button').set_sensitive(False)
- self.tree.get_object('new_separator_button').set_sensitive(False)
- self.tree.get_object('properties_button').set_sensitive(False)
- self.tree.get_object('delete_button').set_sensitive(False)
- accelgroup = Gtk.AccelGroup()
- keyval, modifier = Gtk.accelerator_parse('<Ctrl>Z')
- accelgroup.connect(keyval, modifier, Gtk.AccelFlags.VISIBLE, self.on_mainwindow_undo)
- keyval, modifier = Gtk.accelerator_parse('<Ctrl><Shift>Z')
- accelgroup.connect(keyval, modifier, Gtk.AccelFlags.VISIBLE, self.on_mainwindow_redo)
- keyval, modifier = Gtk.accelerator_parse('F1')
- accelgroup.connect(keyval, modifier, Gtk.AccelFlags.VISIBLE, self.on_help_button_clicked)
- self.tree.get_object('mainwindow').add_accel_group(accelgroup)
-
- def run(self):
- self.loadMenus()
- self.editor.applications.tree.add_monitor(self.menuChanged, None)
- self.editor.settings.tree.add_monitor(self.menuChanged, None)
- self.tree.get_object('mainwindow').show_all()
- Gtk.main()
-
- def menuChanged(self, *a):
- if self.timer:
- GLib.Source.remove(self.timer)
- self.timer = None
- self.timer = GLib.timeout_add(3, self.loadUpdates)
-
- def loadUpdates(self):
- if not self.allow_update:
- self.timer = None
- return False
- menu_tree = self.tree.get_object('menu_tree')
- item_tree = self.tree.get_object('item_tree')
- items, iter = item_tree.get_selection().get_selected()
- update_items = False
- item_id, separator_path = None, None
- if iter:
- update_items = True
- if items[iter][3].get_type() == matemenu.TYPE_DIRECTORY:
- item_id = os.path.split(items[iter][3].get_desktop_file_path())[1]
- update_items = True
- elif items[iter][3].get_type() == matemenu.TYPE_ENTRY:
- item_id = items[iter][3].get_desktop_file_id()
- update_items = True
- elif items[iter][3].get_type() == matemenu.TYPE_SEPARATOR:
- item_id = items.get_path(iter).to_string()
- update_items = True
- menus, iter = menu_tree.get_selection().get_selected()
- update_menus = False
- menu_id = None
- if iter:
- if menus[iter][2].get_desktop_file_path():
- menu_id = os.path.split(menus[iter][2].get_desktop_file_path())[1]
- else:
- menu_id = menus[iter][2].get_menu_id()
- update_menus = True
- self.loadMenus()
- #find current menu in new tree
- if update_menus:
- menu_tree.get_model().foreach(self.findMenu, menu_id)
- menus, iter = menu_tree.get_selection().get_selected()
- if iter:
- self.on_menu_tree_cursor_changed(menu_tree)
- #find current item in new list
- if update_items:
- i = 0
- for item in item_tree.get_model():
- found = False
- if item[3].get_type() == matemenu.TYPE_ENTRY and item[3].get_desktop_file_id() == item_id:
- found = True
- if item[3].get_type() == matemenu.TYPE_DIRECTORY and item[3].get_desktop_file_path():
- if os.path.split(item[3].get_desktop_file_path())[1] == item_id:
- found = True
- if item[3].get_type() == matemenu.TYPE_SEPARATOR:
- if not isinstance(item_id, tuple):
- #we may not skip the increment via "continue"
- i += 1
- continue
- #separators have no id, have to find them manually
- #probably won't work with two separators together
- if (item_id[0] - 1,) == (i,):
- found = True
- elif (item_id[0] + 1,) == (i,):
- found = True
- elif (item_id[0],) == (i,):
- found = True
- if found:
- item_tree.get_selection().select_path((i,))
- self.on_item_tree_cursor_changed(item_tree)
- break
- i += 1
- self.timer = None
- return False
-
- def findMenu(self, menus, path, iter, menu_id):
- if not menus[path][2].get_desktop_file_path():
- if menu_id == menus[path][2].get_menu_id():
- menu_tree = self.tree.get_object('menu_tree')
- menu_tree.expand_to_path(path)
- menu_tree.get_selection().select_path(path)
- return True
- return False
- if os.path.split(menus[path][2].get_desktop_file_path())[1] == menu_id:
- menu_tree = self.tree.get_object('menu_tree')
- menu_tree.expand_to_path(path)
- menu_tree.get_selection().select_path(path)
- return True
-
- def setupMenuTree(self):
- self.menu_store = Gtk.TreeStore(GdkPixbuf.Pixbuf, str, object)
- menus = self.tree.get_object('menu_tree')
- column = Gtk.TreeViewColumn(_('Name'))
- column.set_spacing(4)
- cell = Gtk.CellRendererPixbuf()
- column.pack_start(cell, False)
- column.set_attributes(cell, pixbuf=0)
- cell = Gtk.CellRendererText()
- cell.set_fixed_size(-1, 25)
- column.pack_start(cell, True)
- column.set_attributes(cell, markup=1)
- column.set_sizing(Gtk.TreeViewColumnSizing.FIXED)
- menus.append_column(column)
- menus.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, self.dnd_menus, Gdk.DragAction.COPY)
- menus.enable_model_drag_dest(self.dnd_both, Gdk.DragAction.PRIVATE)
-
- def setupItemTree(self):
- items = self.tree.get_object('item_tree')
- column = Gtk.TreeViewColumn(_('Show'))
- cell = Gtk.CellRendererToggle()
- cell.connect('toggled', self.on_item_tree_show_toggled)
- column.pack_start(cell, True)
- column.set_attributes(cell, active=0)
- #hide toggle for separators
- column.set_cell_data_func(cell, self._cell_data_toggle_func)
- items.append_column(column)
- column = Gtk.TreeViewColumn(_('Item'))
- column.set_spacing(4)
- cell = Gtk.CellRendererPixbuf()
- column.pack_start(cell, False)
- column.set_attributes(cell, pixbuf=1)
- cell = Gtk.CellRendererText()
- cell.set_fixed_size(-1, 25)
- column.pack_start(cell, True)
- column.set_attributes(cell, markup=2)
- items.append_column(column)
- self.item_store = Gtk.ListStore(bool, GdkPixbuf.Pixbuf, str, object)
- items.set_model(self.item_store)
- items.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, self.dnd_items, Gdk.DragAction.COPY)
- items.enable_model_drag_dest(self.dnd_items, Gdk.DragAction.PRIVATE)
-
- def _cell_data_toggle_func(self, tree_column, renderer, model, treeiter, data=None):
- if model[treeiter][3].get_type() == matemenu.TYPE_SEPARATOR:
- renderer.set_property('visible', False)
- else:
- renderer.set_property('visible', True)
-
- def loadMenus(self):
- self.menu_store.clear()
- for menu in self.editor.getMenus():
- iters = [None]*20
- self.loadMenu(iters, menu)
- menu_tree = self.tree.get_object('menu_tree')
- menu_tree.set_model(self.menu_store)
- for menu in self.menu_store:
- #this might not work for some reason
- try:
- menu_tree.expand_to_path(menu.path)
- except:
- pass
- menu_tree.get_selection().select_path((0,))
- self.on_menu_tree_cursor_changed(menu_tree)
-
- def loadMenu(self, iters, parent, depth=0):
- if depth == 0:
- icon = util.getIcon(parent)
- iters[depth] = self.menu_store.append(None, (icon, cgi.escape(parent.get_name()), parent))
- depth += 1
- for menu, show in self.editor.getMenus(parent):
- if show:
- name = cgi.escape(menu.get_name())
- else:
- name = '<small><i>' + cgi.escape(menu.get_name()) + '</i></small>'
- icon = util.getIcon(menu)
- iters[depth] = self.menu_store.append(iters[depth-1], (icon, name, menu))
- self.loadMenu(iters, menu, depth)
- depth -= 1
-
- def loadItems(self, menu, menu_path):
- self.item_store.clear()
- for item, show in self.editor.getItems(menu):
- menu_icon = None
- if item.get_type() == matemenu.TYPE_SEPARATOR:
- name = '---'
- icon = None
- elif item.get_type() == matemenu.TYPE_ENTRY:
- if show:
- name = cgi.escape(item.get_display_name())
- else:
- name = '<small><i>' + cgi.escape(item.get_display_name()) + '</i></small>'
- icon = util.getIcon(item)
- else:
- if show:
- name = cgi.escape(item.get_name())
- else:
- name = '<small><i>' + cgi.escape(item.get_name()) + '</i></small>'
- icon = util.getIcon(item)
- self.item_store.append((show, icon, name, item))
-
- #this is a little timeout callback to insert new items after
- #mate-desktop-item-edit has finished running
- def waitForNewItemProcess(self, process, parent_id, file_path):
- if process.poll() is not None:
- if os.path.isfile(file_path):
- self.editor.insertExternalItem(os.path.split(file_path)[1], parent_id)
- return False
- return True
-
- def waitForNewMenuProcess(self, process, parent_id, file_path):
- if process.poll() is not None:
- #hack for broken mate-desktop-item-edit
- broken_path = os.path.join(os.path.split(file_path)[0], '.directory')
- if os.path.isfile(broken_path):
- os.rename(broken_path, file_path)
- if os.path.isfile(file_path):
- self.editor.insertExternalMenu(os.path.split(file_path)[1], parent_id)
- return False
- return True
-
- #this callback keeps you from editing the same item twice
- def waitForEditProcess(self, process, file_path):
- if process.poll() is not None:
- self.edit_pool.remove(file_path)
- return False
- return True
-
- def on_delete_event(self, widget, event):
- self.quit()
-
- def on_new_menu_button_clicked(self, button):
- menu_tree = self.tree.get_object('menu_tree')
- menus, iter = menu_tree.get_selection().get_selected()
- if not iter:
- parent = menus[(0,)][2]
- menu_tree.expand_to_path((0,))
- menu_tree.get_selection().select_path((0,))
- else:
- parent = menus[iter][2]
- file_path = os.path.join(util.getUserDirectoryPath(), util.getUniqueFileId('mozo-made', '.directory'))
- process = subprocess.Popen(['mate-desktop-item-edit', file_path], env=os.environ)
- GLib.timeout_add(100, self.waitForNewMenuProcess, process, parent.menu_id, file_path)
-
- def on_new_item_button_clicked(self, button):
- menu_tree = self.tree.get_object('menu_tree')
- menus, iter = menu_tree.get_selection().get_selected()
- if not iter:
- parent = menus[(0,)][2]
- menu_tree.expand_to_path((0,))
- menu_tree.get_selection().select_path((0,))
- else:
- parent = menus[iter][2]
- file_path = os.path.join(util.getUserItemPath(), util.getUniqueFileId('mozo-made', '.desktop'))
- process = subprocess.Popen(['mate-desktop-item-edit', file_path], env=os.environ)
- GLib.timeout_add(100, self.waitForNewItemProcess, process, parent.menu_id, file_path)
-
- def on_new_separator_button_clicked(self, button):
- item_tree = self.tree.get_object('item_tree')
- items, iter = item_tree.get_selection().get_selected()
- if not iter:
- return
- else:
- after = items[iter][3]
- menu_tree = self.tree.get_object('menu_tree')
- menus, iter = menu_tree.get_selection().get_selected()
- parent = menus[iter][2]
- self.editor.createSeparator(parent, after=after)
-
- def on_edit_delete_activate(self, menu):
- item_tree = self.tree.get_object('item_tree')
- items, iter = item_tree.get_selection().get_selected()
- if not iter:
- return
- item = items[iter][3]
- if item.get_type() == matemenu.TYPE_ENTRY:
- self.editor.deleteItem(item)
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- self.editor.deleteMenu(item)
- elif item.get_type() == matemenu.TYPE_SEPARATOR:
- self.editor.deleteSeparator(item)
-
- def on_edit_revert_to_original_activate(self, menu):
- item_tree = self.tree.get_object('item_tree')
- items, iter = item_tree.get_selection().get_selected()
- if not iter:
- return
- item = items[iter][3]
- if item.get_type() == matemenu.TYPE_ENTRY:
- self.editor.revertItem(item)
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- self.editor.revertMenu(item)
-
- def on_edit_properties_activate(self, menu):
- item_tree = self.tree.get_object('item_tree')
- items, iter = item_tree.get_selection().get_selected()
- if not iter:
- return
- item = items[iter][3]
- if item.get_type() not in (matemenu.TYPE_ENTRY, matemenu.TYPE_DIRECTORY):
- return
-
- if item.get_type() == matemenu.TYPE_ENTRY:
- file_path = os.path.join(util.getUserItemPath(), item.get_desktop_file_id())
- file_type = 'Item'
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- file_path = os.path.join(util.getUserDirectoryPath(), os.path.split(item.get_desktop_file_path())[1])
- file_type = 'Menu'
-
- if not os.path.isfile(file_path):
- shutil.copy(item.get_desktop_file_path(), file_path)
- self.editor._MenuEditor__addUndo([(file_type, os.path.split(file_path)[1]),])
- else:
- self.editor._MenuEditor__addUndo([item,])
- if file_path not in self.edit_pool:
- self.edit_pool.append(file_path)
- process = subprocess.Popen(['mate-desktop-item-edit', file_path], env=os.environ)
- GLib.timeout_add(100, self.waitForEditProcess, process, file_path)
-
- def on_menu_tree_cursor_changed(self, treeview):
- menus, iter = treeview.get_selection().get_selected()
- if iter is None:
- return
- menu_path = menus.get_path(iter)
- item_tree = self.tree.get_object('item_tree')
- item_tree.get_selection().unselect_all()
- self.loadItems(self.menu_store[menu_path][2], menu_path)
- self.tree.get_object('edit_delete').set_sensitive(False)
- self.tree.get_object('edit_revert_to_original').set_sensitive(False)
- self.tree.get_object('edit_properties').set_sensitive(False)
- self.tree.get_object('move_up_button').set_sensitive(False)
- self.tree.get_object('move_down_button').set_sensitive(False)
- self.tree.get_object('new_separator_button').set_sensitive(False)
-
- def on_menu_tree_drag_data_get(self, treeview, context, selection, target_id, etime):
- menus, iter = treeview.get_selection().get_selected()
- self.drag_data = menus[iter][2]
-
- def on_menu_tree_drag_data_received(self, treeview, context, x, y, selection, info, etime):
- menus = treeview.get_model()
- drop_info = treeview.get_dest_row_at_pos(x, y)
- if drop_info:
- path, position = drop_info
- types = (Gtk.TreeViewDropPosition.INTO_OR_BEFORE, Gtk.TreeViewDropPosition.INTO_OR_AFTER)
- if position not in types:
- context.finish(False, False, etime)
- return False
- if str(selection.get_target()) in ('MOZO_ITEM_ROW', 'MOZO_MENU_ROW'):
- if self.drag_data is None:
- return False
- item = self.drag_data
- new_parent = menus[path][2]
- if item.get_type() == matemenu.TYPE_ENTRY:
- self.editor.copyItem(item, new_parent)
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- if not self.editor.moveMenu(item, new_parent):
- self.loadUpdates()
- elif item.get_type() == matemenu.TYPE_SEPARATOR:
- self.editor.moveSeparator(item, new_parent)
- else:
- context.finish(False, False, etime)
- context.finish(True, True, etime)
- self.drag_data = None
-
- def on_item_tree_show_toggled(self, cell, path):
- item = self.item_store[path][3]
- if item.get_type() == matemenu.TYPE_SEPARATOR:
- return
- if self.item_store[path][0]:
- self.editor.setVisible(item, False)
- else:
- self.editor.setVisible(item, True)
- self.item_store[path][0] = not self.item_store[path][0]
-
- def on_item_tree_cursor_changed(self, treeview):
- items, iter = treeview.get_selection().get_selected()
- if iter is None:
- return
-
- item = items[iter][3]
- self.tree.get_object('edit_delete').set_sensitive(True)
- self.tree.get_object('new_separator_button').set_sensitive(True)
- self.tree.get_object('delete_button').set_sensitive(True)
-
- can_revert = self.editor.canRevert(item)
- self.tree.get_object('edit_revert_to_original').set_sensitive(can_revert)
-
- can_edit = not item.get_type() == matemenu.TYPE_SEPARATOR
- self.tree.get_object('edit_properties').set_sensitive(can_edit)
- self.tree.get_object('properties_button').set_sensitive(can_edit)
-
- index = items.get_path(iter).get_indices()[0]
- can_go_up = index > 0
- can_go_down = index < len(items) - 1
- self.tree.get_object('move_up_button').set_sensitive(can_go_up)
- self.tree.get_object('move_down_button').set_sensitive(can_go_down)
-
- def on_item_tree_row_activated(self, treeview, path, column):
- self.on_edit_properties_activate(None)
-
- def on_item_tree_popup_menu(self, item_tree, event=None):
- model, iter = item_tree.get_selection().get_selected()
- if event:
- #don't show if it's not the right mouse button
- if event.button != 3:
- return
- button = event.button
- event_time = event.time
- info = item_tree.get_path_at_pos(int(event.x), int(event.y))
- if info is not None:
- path, col, cellx, celly = info
- item_tree.grab_focus()
- item_tree.set_cursor(path, col, 0)
- else:
- path = model.get_path(iter)
- button = 0
- event_time = 0
- item_tree.grab_focus()
- item_tree.set_cursor(path, item_tree.get_columns()[0], 0)
- popup = self.tree.get_object('edit_menu')
- popup.popup(None, None, None, None, button, event_time)
- #without this shift-f10 won't work
- return True
-
- def on_item_tree_drag_data_get(self, treeview, context, selection, target_id, etime):
- items, iter = treeview.get_selection().get_selected()
- self.drag_data = items[iter][3]
-
- def on_item_tree_drag_data_received(self, treeview, context, x, y, selection, info, etime):
- items = treeview.get_model()
- types_before = (Gtk.TreeViewDropPosition.BEFORE, Gtk.TreeViewDropPosition.INTO_OR_BEFORE)
- types_into = (Gtk.TreeViewDropPosition.INTO_OR_BEFORE, Gtk.TreeViewDropPosition.INTO_OR_AFTER)
- types_after = (Gtk.TreeViewDropPosition.AFTER, Gtk.TreeViewDropPosition.INTO_OR_AFTER)
- if str(selection.get_target()) == 'MOZO_ITEM_ROW':
- drop_info = treeview.get_dest_row_at_pos(x, y)
- before = None
- after = None
- if self.drag_data is None:
- return False
- item = self.drag_data
- # by default we assume, that the items stays in the same menu
- destination = item.get_parent()
- if drop_info:
- path, position = drop_info
- target = items[path][3]
- # move the item to the directory, if the item was dropped into it
- if (target.get_type() == matemenu.TYPE_DIRECTORY) and (position in types_into):
- # append the selected item to the choosen menu
- destination = target
- elif position in types_before:
- before = target
- elif position in types_after:
- after = target
- else:
- # this does not happen
- pass
- else:
- path = (len(items) - 1,)
- after = items[path][3]
- if item.get_type() == matemenu.TYPE_ENTRY:
- self.editor.moveItem(item, destination, before, after)
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- if not self.editor.moveMenu(item, destination, before, after):
- self.loadUpdates()
- elif item.get_type() == matemenu.TYPE_SEPARATOR:
- self.editor.moveSeparator(item, destination, before, after)
- context.finish(True, True, etime)
- elif str(selection.get_target()) == 'text/plain':
- if selection.data is None:
- return False
- menus, iter = self.tree.get_object('menu_tree').get_selection().get_selected()
- parent = menus[iter][2]
- drop_info = treeview.get_dest_row_at_pos(x, y)
- before = None
- after = None
- if drop_info:
- path, position = drop_info
- if position in types_before:
- before = items[path][3]
- else:
- after = items[path][3]
- else:
- path = (len(items) - 1,)
- after = items[path][3]
- file_path = urllib.unquote(selection.data).strip()
- if not file_path.startswith('file:'):
- return
- myfile = Gio.File(uri=file_path)
- file_info = myfile.query_info(Gio.FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE)
- content_type = file_info.get_content_type()
- if content_type == 'application/x-desktop':
- input_stream = myfile.read()
- keyfile = GLib.KeyFile()
- keyfile.load_from_data(input_stream.read())
- self.editor.createItem(parent, before, after, KeyFile=keyfile)
- elif content_type in ('application/x-shellscript', 'application/x-executable'):
- self.editor.createItem(parent, before, after,
- Name=os.path.split(file_path)[1].strip(),
- Exec=file_path.replace('file://', '').strip(),
- Terminal=False)
- self.drag_data = None
-
- def on_item_tree_key_press_event(self, item_tree, event):
- if event.keyval == Gdk.KEY_Delete:
- self.on_edit_delete_activate(item_tree)
-
- def on_move_up_button_clicked(self, button):
- item_tree = self.tree.get_object('item_tree')
- items, iter = item_tree.get_selection().get_selected()
- if not iter:
- return
- path = items.get_path(iter)
- #at top, can't move up
- if path.get_indices()[0] == 0:
- return
- item = items[path][3]
- before = items[(path[0] - 1,)][3]
- if item.get_type() == matemenu.TYPE_ENTRY:
- self.editor.moveItem(item, item.get_parent(), before=before)
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- self.editor.moveMenu(item, item.get_parent(), before=before)
- elif item.get_type() == matemenu.TYPE_SEPARATOR:
- self.editor.moveSeparator(item, item.get_parent(), before=before)
-
- def on_move_down_button_clicked(self, button):
- item_tree = self.tree.get_object('item_tree')
- items, iter = item_tree.get_selection().get_selected()
- if not iter:
- return
- path = items.get_path(iter)
- #at bottom, can't move down
- if path.get_indices()[0] == (len(items) - 1):
- return
- item = items[path][3]
- after = items[path][3]
- if item.get_type() == matemenu.TYPE_ENTRY:
- self.editor.moveItem(item, item.get_parent(), after=after)
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- self.editor.moveMenu(item, item.get_parent(), after=after)
- elif item.get_type() == matemenu.TYPE_SEPARATOR:
- self.editor.moveSeparator(item, item.get_parent(), after=after)
-
- def on_mainwindow_undo(self, accelgroup, window, keyval, modifier):
- self.editor.undo()
-
- def on_mainwindow_redo(self, accelgroup, window, keyval, modifier):
- self.editor.redo()
-
- def on_help_button_clicked(self, *args):
- Gtk.show_uri(Gdk.Screen.get_default(), "help:mate-user-guide/menu-editor", Gtk.get_current_event_time())
-
- def on_revert_button_clicked(self, button):
- dialog = self.tree.get_object('revertdialog')
- dialog.set_transient_for(self.tree.get_object('mainwindow'))
- dialog.show_all()
- if dialog.run() == Gtk.ResponseType.YES:
- self.editor.revert()
- dialog.hide()
-
- def on_close_button_clicked(self, button):
- try:
- self.tree.get_object('mainwindow').hide()
- except:
- pass
- GLib.timeout_add(10, self.quit)
-
- def on_properties_button_clicked(self, button):
- self.on_edit_properties_activate(None)
-
- def on_delete_button_clicked(self, button):
- self.on_edit_delete_activate(None)
-
- def on_style_updated(self, *args):
- self.loadUpdates()
-
- def quit(self):
- self.editor.quit()
- Gtk.main_quit()
+ timer = None
+ #hack to make editing menu properties work
+ allow_update = True
+ #drag-and-drop stuff
+ dnd_items = [('MOZO_ITEM_ROW', Gtk.TargetFlags.SAME_APP, 0), ('text/plain', 0, 1)]
+ dnd_menus = [('MOZO_MENU_ROW', Gtk.TargetFlags.SAME_APP, 0)]
+ dnd_both = [dnd_items[0],] + dnd_menus
+ drag_data = None
+ edit_pool = []
+
+ def __init__(self, datadir, version, argv):
+ self.file_path = datadir
+ self.version = version
+ self.editor = MenuEditor()
+ Gtk.Window.set_default_icon_name('mozo')
+ self.tree = Gtk.Builder()
+ self.tree.set_translation_domain(GETTEXT_PACKAGE)
+ self.tree.add_from_file(os.path.join(self.file_path, 'mozo.ui'))
+ self.tree.connect_signals(self)
+ self.setupMenuTree()
+ self.setupItemTree()
+ self.tree.get_object('edit_delete').set_sensitive(False)
+ self.tree.get_object('edit_revert_to_original').set_sensitive(False)
+ self.tree.get_object('edit_properties').set_sensitive(False)
+ self.tree.get_object('move_up_button').set_sensitive(False)
+ self.tree.get_object('move_down_button').set_sensitive(False)
+ self.tree.get_object('new_separator_button').set_sensitive(False)
+ self.tree.get_object('properties_button').set_sensitive(False)
+ self.tree.get_object('delete_button').set_sensitive(False)
+ accelgroup = Gtk.AccelGroup()
+ keyval, modifier = Gtk.accelerator_parse('<Ctrl>Z')
+ accelgroup.connect(keyval, modifier, Gtk.AccelFlags.VISIBLE, self.on_mainwindow_undo)
+ keyval, modifier = Gtk.accelerator_parse('<Ctrl><Shift>Z')
+ accelgroup.connect(keyval, modifier, Gtk.AccelFlags.VISIBLE, self.on_mainwindow_redo)
+ keyval, modifier = Gtk.accelerator_parse('F1')
+ accelgroup.connect(keyval, modifier, Gtk.AccelFlags.VISIBLE, self.on_help_button_clicked)
+ self.tree.get_object('mainwindow').add_accel_group(accelgroup)
+
+ def run(self):
+ self.loadMenus()
+ self.editor.applications.tree.add_monitor(self.menuChanged, None)
+ self.editor.settings.tree.add_monitor(self.menuChanged, None)
+ self.tree.get_object('mainwindow').show_all()
+ Gtk.main()
+
+ def menuChanged(self, *a):
+ if self.timer:
+ GLib.Source.remove(self.timer)
+ self.timer = None
+ self.timer = GLib.timeout_add(3, self.loadUpdates)
+
+ def loadUpdates(self):
+ if not self.allow_update:
+ self.timer = None
+ return False
+ menu_tree = self.tree.get_object('menu_tree')
+ item_tree = self.tree.get_object('item_tree')
+ items, iter = item_tree.get_selection().get_selected()
+ update_items = False
+ item_id, separator_path = None, None
+ if iter:
+ update_items = True
+ if items[iter][3].get_type() == matemenu.TYPE_DIRECTORY:
+ item_id = os.path.split(items[iter][3].get_desktop_file_path())[1]
+ update_items = True
+ elif items[iter][3].get_type() == matemenu.TYPE_ENTRY:
+ item_id = items[iter][3].get_desktop_file_id()
+ update_items = True
+ elif items[iter][3].get_type() == matemenu.TYPE_SEPARATOR:
+ item_id = items.get_path(iter).to_string()
+ update_items = True
+ menus, iter = menu_tree.get_selection().get_selected()
+ update_menus = False
+ menu_id = None
+ if iter:
+ if menus[iter][2].get_desktop_file_path():
+ menu_id = os.path.split(menus[iter][2].get_desktop_file_path())[1]
+ else:
+ menu_id = menus[iter][2].get_menu_id()
+ update_menus = True
+ self.loadMenus()
+ #find current menu in new tree
+ if update_menus:
+ menu_tree.get_model().foreach(self.findMenu, menu_id)
+ menus, iter = menu_tree.get_selection().get_selected()
+ if iter:
+ self.on_menu_tree_cursor_changed(menu_tree)
+ #find current item in new list
+ if update_items:
+ i = 0
+ for item in item_tree.get_model():
+ found = False
+ if item[3].get_type() == matemenu.TYPE_ENTRY and item[3].get_desktop_file_id() == item_id:
+ found = True
+ if item[3].get_type() == matemenu.TYPE_DIRECTORY and item[3].get_desktop_file_path():
+ if os.path.split(item[3].get_desktop_file_path())[1] == item_id:
+ found = True
+ if item[3].get_type() == matemenu.TYPE_SEPARATOR:
+ if not isinstance(item_id, tuple):
+ #we may not skip the increment via "continue"
+ i += 1
+ continue
+ #separators have no id, have to find them manually
+ #probably won't work with two separators together
+ if (item_id[0] - 1,) == (i,):
+ found = True
+ elif (item_id[0] + 1,) == (i,):
+ found = True
+ elif (item_id[0],) == (i,):
+ found = True
+ if found:
+ item_tree.get_selection().select_path((i,))
+ self.on_item_tree_cursor_changed(item_tree)
+ break
+ i += 1
+ self.timer = None
+ return False
+
+ def findMenu(self, menus, path, iter, menu_id):
+ if not menus[path][2].get_desktop_file_path():
+ if menu_id == menus[path][2].get_menu_id():
+ menu_tree = self.tree.get_object('menu_tree')
+ menu_tree.expand_to_path(path)
+ menu_tree.get_selection().select_path(path)
+ return True
+ return False
+ if os.path.split(menus[path][2].get_desktop_file_path())[1] == menu_id:
+ menu_tree = self.tree.get_object('menu_tree')
+ menu_tree.expand_to_path(path)
+ menu_tree.get_selection().select_path(path)
+ return True
+
+ def setupMenuTree(self):
+ self.menu_store = Gtk.TreeStore(GdkPixbuf.Pixbuf, str, object)
+ menus = self.tree.get_object('menu_tree')
+ column = Gtk.TreeViewColumn(_('Name'))
+ column.set_spacing(4)
+ cell = Gtk.CellRendererPixbuf()
+ column.pack_start(cell, False)
+ column.set_attributes(cell, pixbuf=0)
+ cell = Gtk.CellRendererText()
+ cell.set_fixed_size(-1, 25)
+ column.pack_start(cell, True)
+ column.set_attributes(cell, markup=1)
+ column.set_sizing(Gtk.TreeViewColumnSizing.FIXED)
+ menus.append_column(column)
+ menus.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, self.dnd_menus, Gdk.DragAction.COPY)
+ menus.enable_model_drag_dest(self.dnd_both, Gdk.DragAction.PRIVATE)
+
+ def setupItemTree(self):
+ items = self.tree.get_object('item_tree')
+ column = Gtk.TreeViewColumn(_('Show'))
+ cell = Gtk.CellRendererToggle()
+ cell.connect('toggled', self.on_item_tree_show_toggled)
+ column.pack_start(cell, True)
+ column.set_attributes(cell, active=0)
+ #hide toggle for separators
+ column.set_cell_data_func(cell, self._cell_data_toggle_func)
+ items.append_column(column)
+ column = Gtk.TreeViewColumn(_('Item'))
+ column.set_spacing(4)
+ cell = Gtk.CellRendererPixbuf()
+ column.pack_start(cell, False)
+ column.set_attributes(cell, pixbuf=1)
+ cell = Gtk.CellRendererText()
+ cell.set_fixed_size(-1, 25)
+ column.pack_start(cell, True)
+ column.set_attributes(cell, markup=2)
+ items.append_column(column)
+ self.item_store = Gtk.ListStore(bool, GdkPixbuf.Pixbuf, str, object)
+ items.set_model(self.item_store)
+ items.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, self.dnd_items, Gdk.DragAction.COPY)
+ items.enable_model_drag_dest(self.dnd_items, Gdk.DragAction.PRIVATE)
+
+ def _cell_data_toggle_func(self, tree_column, renderer, model, treeiter, data=None):
+ if model[treeiter][3].get_type() == matemenu.TYPE_SEPARATOR:
+ renderer.set_property('visible', False)
+ else:
+ renderer.set_property('visible', True)
+
+ def loadMenus(self):
+ self.menu_store.clear()
+ for menu in self.editor.getMenus():
+ iters = [None]*20
+ self.loadMenu(iters, menu)
+ menu_tree = self.tree.get_object('menu_tree')
+ menu_tree.set_model(self.menu_store)
+ for menu in self.menu_store:
+ #this might not work for some reason
+ try:
+ menu_tree.expand_to_path(menu.path)
+ except:
+ pass
+ menu_tree.get_selection().select_path((0,))
+ self.on_menu_tree_cursor_changed(menu_tree)
+
+ def loadMenu(self, iters, parent, depth=0):
+ if depth == 0:
+ icon = util.getIcon(parent)
+ iters[depth] = self.menu_store.append(None, (icon, cgi.escape(parent.get_name()), parent))
+ depth += 1
+ for menu, show in self.editor.getMenus(parent):
+ if show:
+ name = cgi.escape(menu.get_name())
+ else:
+ name = '<small><i>' + cgi.escape(menu.get_name()) + '</i></small>'
+ icon = util.getIcon(menu)
+ iters[depth] = self.menu_store.append(iters[depth-1], (icon, name, menu))
+ self.loadMenu(iters, menu, depth)
+ depth -= 1
+
+ def loadItems(self, menu, menu_path):
+ self.item_store.clear()
+ for item, show in self.editor.getItems(menu):
+ menu_icon = None
+ if item.get_type() == matemenu.TYPE_SEPARATOR:
+ name = '---'
+ icon = None
+ elif item.get_type() == matemenu.TYPE_ENTRY:
+ if show:
+ name = cgi.escape(item.get_display_name())
+ else:
+ name = '<small><i>' + cgi.escape(item.get_display_name()) + '</i></small>'
+ icon = util.getIcon(item)
+ else:
+ if show:
+ name = cgi.escape(item.get_name())
+ else:
+ name = '<small><i>' + cgi.escape(item.get_name()) + '</i></small>'
+ icon = util.getIcon(item)
+ self.item_store.append((show, icon, name, item))
+
+ #this is a little timeout callback to insert new items after
+ #mate-desktop-item-edit has finished running
+ def waitForNewItemProcess(self, process, parent_id, file_path):
+ if process.poll() is not None:
+ if os.path.isfile(file_path):
+ self.editor.insertExternalItem(os.path.split(file_path)[1], parent_id)
+ return False
+ return True
+
+ def waitForNewMenuProcess(self, process, parent_id, file_path):
+ if process.poll() is not None:
+ #hack for broken mate-desktop-item-edit
+ broken_path = os.path.join(os.path.split(file_path)[0], '.directory')
+ if os.path.isfile(broken_path):
+ os.rename(broken_path, file_path)
+ if os.path.isfile(file_path):
+ self.editor.insertExternalMenu(os.path.split(file_path)[1], parent_id)
+ return False
+ return True
+
+ #this callback keeps you from editing the same item twice
+ def waitForEditProcess(self, process, file_path):
+ if process.poll() is not None:
+ self.edit_pool.remove(file_path)
+ return False
+ return True
+
+ def on_delete_event(self, widget, event):
+ self.quit()
+
+ def on_new_menu_button_clicked(self, button):
+ menu_tree = self.tree.get_object('menu_tree')
+ menus, iter = menu_tree.get_selection().get_selected()
+ if not iter:
+ parent = menus[(0,)][2]
+ menu_tree.expand_to_path((0,))
+ menu_tree.get_selection().select_path((0,))
+ else:
+ parent = menus[iter][2]
+ file_path = os.path.join(util.getUserDirectoryPath(), util.getUniqueFileId('mozo-made', '.directory'))
+ process = subprocess.Popen(['mate-desktop-item-edit', file_path], env=os.environ)
+ GLib.timeout_add(100, self.waitForNewMenuProcess, process, parent.menu_id, file_path)
+
+ def on_new_item_button_clicked(self, button):
+ menu_tree = self.tree.get_object('menu_tree')
+ menus, iter = menu_tree.get_selection().get_selected()
+ if not iter:
+ parent = menus[(0,)][2]
+ menu_tree.expand_to_path((0,))
+ menu_tree.get_selection().select_path((0,))
+ else:
+ parent = menus[iter][2]
+ file_path = os.path.join(util.getUserItemPath(), util.getUniqueFileId('mozo-made', '.desktop'))
+ process = subprocess.Popen(['mate-desktop-item-edit', file_path], env=os.environ)
+ GLib.timeout_add(100, self.waitForNewItemProcess, process, parent.menu_id, file_path)
+
+ def on_new_separator_button_clicked(self, button):
+ item_tree = self.tree.get_object('item_tree')
+ items, iter = item_tree.get_selection().get_selected()
+ if not iter:
+ return
+ else:
+ after = items[iter][3]
+ menu_tree = self.tree.get_object('menu_tree')
+ menus, iter = menu_tree.get_selection().get_selected()
+ parent = menus[iter][2]
+ self.editor.createSeparator(parent, after=after)
+
+ def on_edit_delete_activate(self, menu):
+ item_tree = self.tree.get_object('item_tree')
+ items, iter = item_tree.get_selection().get_selected()
+ if not iter:
+ return
+ item = items[iter][3]
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ self.editor.deleteItem(item)
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ self.editor.deleteMenu(item)
+ elif item.get_type() == matemenu.TYPE_SEPARATOR:
+ self.editor.deleteSeparator(item)
+
+ def on_edit_revert_to_original_activate(self, menu):
+ item_tree = self.tree.get_object('item_tree')
+ items, iter = item_tree.get_selection().get_selected()
+ if not iter:
+ return
+ item = items[iter][3]
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ self.editor.revertItem(item)
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ self.editor.revertMenu(item)
+
+ def on_edit_properties_activate(self, menu):
+ item_tree = self.tree.get_object('item_tree')
+ items, iter = item_tree.get_selection().get_selected()
+ if not iter:
+ return
+ item = items[iter][3]
+ if item.get_type() not in (matemenu.TYPE_ENTRY, matemenu.TYPE_DIRECTORY):
+ return
+
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ file_path = os.path.join(util.getUserItemPath(), item.get_desktop_file_id())
+ file_type = 'Item'
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ file_path = os.path.join(util.getUserDirectoryPath(), os.path.split(item.get_desktop_file_path())[1])
+ file_type = 'Menu'
+
+ if not os.path.isfile(file_path):
+ shutil.copy(item.get_desktop_file_path(), file_path)
+ self.editor._MenuEditor__addUndo([(file_type, os.path.split(file_path)[1]),])
+ else:
+ self.editor._MenuEditor__addUndo([item,])
+ if file_path not in self.edit_pool:
+ self.edit_pool.append(file_path)
+ process = subprocess.Popen(['mate-desktop-item-edit', file_path], env=os.environ)
+ GLib.timeout_add(100, self.waitForEditProcess, process, file_path)
+
+ def on_menu_tree_cursor_changed(self, treeview):
+ menus, iter = treeview.get_selection().get_selected()
+ if iter is None:
+ return
+ menu_path = menus.get_path(iter)
+ item_tree = self.tree.get_object('item_tree')
+ item_tree.get_selection().unselect_all()
+ self.loadItems(self.menu_store[menu_path][2], menu_path)
+ self.tree.get_object('edit_delete').set_sensitive(False)
+ self.tree.get_object('edit_revert_to_original').set_sensitive(False)
+ self.tree.get_object('edit_properties').set_sensitive(False)
+ self.tree.get_object('move_up_button').set_sensitive(False)
+ self.tree.get_object('move_down_button').set_sensitive(False)
+ self.tree.get_object('new_separator_button').set_sensitive(False)
+
+ def on_menu_tree_drag_data_get(self, treeview, context, selection, target_id, etime):
+ menus, iter = treeview.get_selection().get_selected()
+ self.drag_data = menus[iter][2]
+
+ def on_menu_tree_drag_data_received(self, treeview, context, x, y, selection, info, etime):
+ menus = treeview.get_model()
+ drop_info = treeview.get_dest_row_at_pos(x, y)
+ if drop_info:
+ path, position = drop_info
+ types = (Gtk.TreeViewDropPosition.INTO_OR_BEFORE, Gtk.TreeViewDropPosition.INTO_OR_AFTER)
+ if position not in types:
+ context.finish(False, False, etime)
+ return False
+ if str(selection.get_target()) in ('MOZO_ITEM_ROW', 'MOZO_MENU_ROW'):
+ if self.drag_data is None:
+ return False
+ item = self.drag_data
+ new_parent = menus[path][2]
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ self.editor.copyItem(item, new_parent)
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ if not self.editor.moveMenu(item, new_parent):
+ self.loadUpdates()
+ elif item.get_type() == matemenu.TYPE_SEPARATOR:
+ self.editor.moveSeparator(item, new_parent)
+ else:
+ context.finish(False, False, etime)
+ context.finish(True, True, etime)
+ self.drag_data = None
+
+ def on_item_tree_show_toggled(self, cell, path):
+ item = self.item_store[path][3]
+ if item.get_type() == matemenu.TYPE_SEPARATOR:
+ return
+ if self.item_store[path][0]:
+ self.editor.setVisible(item, False)
+ else:
+ self.editor.setVisible(item, True)
+ self.item_store[path][0] = not self.item_store[path][0]
+
+ def on_item_tree_cursor_changed(self, treeview):
+ items, iter = treeview.get_selection().get_selected()
+ if iter is None:
+ return
+
+ item = items[iter][3]
+ self.tree.get_object('edit_delete').set_sensitive(True)
+ self.tree.get_object('new_separator_button').set_sensitive(True)
+ self.tree.get_object('delete_button').set_sensitive(True)
+
+ can_revert = self.editor.canRevert(item)
+ self.tree.get_object('edit_revert_to_original').set_sensitive(can_revert)
+
+ can_edit = not item.get_type() == matemenu.TYPE_SEPARATOR
+ self.tree.get_object('edit_properties').set_sensitive(can_edit)
+ self.tree.get_object('properties_button').set_sensitive(can_edit)
+
+ index = items.get_path(iter).get_indices()[0]
+ can_go_up = index > 0
+ can_go_down = index < len(items) - 1
+ self.tree.get_object('move_up_button').set_sensitive(can_go_up)
+ self.tree.get_object('move_down_button').set_sensitive(can_go_down)
+
+ def on_item_tree_row_activated(self, treeview, path, column):
+ self.on_edit_properties_activate(None)
+
+ def on_item_tree_popup_menu(self, item_tree, event=None):
+ model, iter = item_tree.get_selection().get_selected()
+ if event:
+ #don't show if it's not the right mouse button
+ if event.button != 3:
+ return
+ button = event.button
+ event_time = event.time
+ info = item_tree.get_path_at_pos(int(event.x), int(event.y))
+ if info is not None:
+ path, col, cellx, celly = info
+ item_tree.grab_focus()
+ item_tree.set_cursor(path, col, 0)
+ else:
+ path = model.get_path(iter)
+ button = 0
+ event_time = 0
+ item_tree.grab_focus()
+ item_tree.set_cursor(path, item_tree.get_columns()[0], 0)
+ popup = self.tree.get_object('edit_menu')
+ popup.popup(None, None, None, None, button, event_time)
+ #without this shift-f10 won't work
+ return True
+
+ def on_item_tree_drag_data_get(self, treeview, context, selection, target_id, etime):
+ items, iter = treeview.get_selection().get_selected()
+ self.drag_data = items[iter][3]
+
+ def on_item_tree_drag_data_received(self, treeview, context, x, y, selection, info, etime):
+ items = treeview.get_model()
+ types_before = (Gtk.TreeViewDropPosition.BEFORE, Gtk.TreeViewDropPosition.INTO_OR_BEFORE)
+ types_into = (Gtk.TreeViewDropPosition.INTO_OR_BEFORE, Gtk.TreeViewDropPosition.INTO_OR_AFTER)
+ types_after = (Gtk.TreeViewDropPosition.AFTER, Gtk.TreeViewDropPosition.INTO_OR_AFTER)
+ if str(selection.get_target()) == 'MOZO_ITEM_ROW':
+ drop_info = treeview.get_dest_row_at_pos(x, y)
+ before = None
+ after = None
+ if self.drag_data is None:
+ return False
+ item = self.drag_data
+ # by default we assume, that the items stays in the same menu
+ destination = item.get_parent()
+ if drop_info:
+ path, position = drop_info
+ target = items[path][3]
+ # move the item to the directory, if the item was dropped into it
+ if (target.get_type() == matemenu.TYPE_DIRECTORY) and (position in types_into):
+ # append the selected item to the choosen menu
+ destination = target
+ elif position in types_before:
+ before = target
+ elif position in types_after:
+ after = target
+ else:
+ # this does not happen
+ pass
+ else:
+ path = (len(items) - 1,)
+ after = items[path][3]
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ self.editor.moveItem(item, destination, before, after)
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ if not self.editor.moveMenu(item, destination, before, after):
+ self.loadUpdates()
+ elif item.get_type() == matemenu.TYPE_SEPARATOR:
+ self.editor.moveSeparator(item, destination, before, after)
+ context.finish(True, True, etime)
+ elif str(selection.get_target()) == 'text/plain':
+ if selection.data is None:
+ return False
+ menus, iter = self.tree.get_object('menu_tree').get_selection().get_selected()
+ parent = menus[iter][2]
+ drop_info = treeview.get_dest_row_at_pos(x, y)
+ before = None
+ after = None
+ if drop_info:
+ path, position = drop_info
+ if position in types_before:
+ before = items[path][3]
+ else:
+ after = items[path][3]
+ else:
+ path = (len(items) - 1,)
+ after = items[path][3]
+ file_path = urllib.unquote(selection.data).strip()
+ if not file_path.startswith('file:'):
+ return
+ myfile = Gio.File(uri=file_path)
+ file_info = myfile.query_info(Gio.FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE)
+ content_type = file_info.get_content_type()
+ if content_type == 'application/x-desktop':
+ input_stream = myfile.read()
+ keyfile = GLib.KeyFile()
+ keyfile.load_from_data(input_stream.read())
+ self.editor.createItem(parent, before, after, KeyFile=keyfile)
+ elif content_type in ('application/x-shellscript', 'application/x-executable'):
+ self.editor.createItem(parent, before, after,
+ Name=os.path.split(file_path)[1].strip(),
+ Exec=file_path.replace('file://', '').strip(),
+ Terminal=False)
+ self.drag_data = None
+
+ def on_item_tree_key_press_event(self, item_tree, event):
+ if event.keyval == Gdk.KEY_Delete:
+ self.on_edit_delete_activate(item_tree)
+
+ def on_move_up_button_clicked(self, button):
+ item_tree = self.tree.get_object('item_tree')
+ items, iter = item_tree.get_selection().get_selected()
+ if not iter:
+ return
+ path = items.get_path(iter)
+ #at top, can't move up
+ if path.get_indices()[0] == 0:
+ return
+ item = items[path][3]
+ before = items[(path[0] - 1,)][3]
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ self.editor.moveItem(item, item.get_parent(), before=before)
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ self.editor.moveMenu(item, item.get_parent(), before=before)
+ elif item.get_type() == matemenu.TYPE_SEPARATOR:
+ self.editor.moveSeparator(item, item.get_parent(), before=before)
+
+ def on_move_down_button_clicked(self, button):
+ item_tree = self.tree.get_object('item_tree')
+ items, iter = item_tree.get_selection().get_selected()
+ if not iter:
+ return
+ path = items.get_path(iter)
+ #at bottom, can't move down
+ if path.get_indices()[0] == (len(items) - 1):
+ return
+ item = items[path][3]
+ after = items[path][3]
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ self.editor.moveItem(item, item.get_parent(), after=after)
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ self.editor.moveMenu(item, item.get_parent(), after=after)
+ elif item.get_type() == matemenu.TYPE_SEPARATOR:
+ self.editor.moveSeparator(item, item.get_parent(), after=after)
+
+ def on_mainwindow_undo(self, accelgroup, window, keyval, modifier):
+ self.editor.undo()
+
+ def on_mainwindow_redo(self, accelgroup, window, keyval, modifier):
+ self.editor.redo()
+
+ def on_help_button_clicked(self, *args):
+ Gtk.show_uri(Gdk.Screen.get_default(), "help:mate-user-guide/menu-editor", Gtk.get_current_event_time())
+
+ def on_revert_button_clicked(self, button):
+ dialog = self.tree.get_object('revertdialog')
+ dialog.set_transient_for(self.tree.get_object('mainwindow'))
+ dialog.show_all()
+ if dialog.run() == Gtk.ResponseType.YES:
+ self.editor.revert()
+ dialog.hide()
+
+ def on_close_button_clicked(self, button):
+ try:
+ self.tree.get_object('mainwindow').hide()
+ except:
+ pass
+ GLib.timeout_add(10, self.quit)
+
+ def on_properties_button_clicked(self, button):
+ self.on_edit_properties_activate(None)
+
+ def on_delete_button_clicked(self, button):
+ self.on_edit_delete_activate(None)
+
+ def on_style_updated(self, *args):
+ self.loadUpdates()
+
+ def quit(self):
+ self.editor.quit()
+ Gtk.main_quit()
diff --git a/Mozo/MenuEditor.py b/Mozo/MenuEditor.py
index 0bc7cc4..62832e6 100644
--- a/Mozo/MenuEditor.py
+++ b/Mozo/MenuEditor.py
@@ -27,758 +27,758 @@ from gi.repository import GLib
from Mozo import util
class Menu:
- tree = None
- visible_tree = None
- path = None
- dom = None
+ tree = None
+ visible_tree = None
+ path = None
+ dom = None
class MenuEditor:
- def __init__(self):
- self.locale = locale.getdefaultlocale()[0]
- self.__loadMenus()
- self.__undo = []
- self.__redo = []
-
-
- def __loadMenus(self):
- self.applications = Menu()
- self.applications.tree = matemenu.lookup_tree('mate-applications.menu', matemenu.FLAGS_SHOW_EMPTY|matemenu.FLAGS_INCLUDE_EXCLUDED|matemenu.FLAGS_INCLUDE_NODISPLAY|matemenu.FLAGS_SHOW_ALL_SEPARATORS)
- self.applications.visible_tree = matemenu.lookup_tree('mate-applications.menu')
- self.applications.tree.sort_key = matemenu.SORT_DISPLAY_NAME
- self.applications.visible_tree.sort_key = matemenu.SORT_DISPLAY_NAME
- self.applications.path = os.path.join(util.getUserMenuPath(), self.applications.tree.get_menu_file())
- try:
- self.applications.dom = xml.dom.minidom.parse(self.applications.path)
- except (IOError, xml.parsers.expat.ExpatError):
- self.applications.dom = xml.dom.minidom.parseString(util.getUserMenuXml(self.applications.tree))
- util.removeWhitespaceNodes(self.applications.dom)
-
- self.settings = Menu()
- self.settings.tree = matemenu.lookup_tree('mate-settings.menu', matemenu.FLAGS_SHOW_EMPTY|matemenu.FLAGS_INCLUDE_EXCLUDED|matemenu.FLAGS_INCLUDE_NODISPLAY|matemenu.FLAGS_SHOW_ALL_SEPARATORS)
- self.settings.visible_tree = matemenu.lookup_tree('mate-settings.menu')
- self.settings.tree.sort_key = matemenu.SORT_DISPLAY_NAME
- self.settings.visible_tree.sort_key = matemenu.SORT_DISPLAY_NAME
- self.settings.path = os.path.join(util.getUserMenuPath(), self.settings.tree.get_menu_file())
- try:
- self.settings.dom = xml.dom.minidom.parse(self.settings.path)
- except (IOError, xml.parsers.expat.ExpatError):
- self.settings.dom = xml.dom.minidom.parseString(util.getUserMenuXml(self.settings.tree))
- util.removeWhitespaceNodes(self.settings.dom)
-
- self.save(True)
-
- def save(self, from_loading=False):
- for menu in ('applications', 'settings'):
- with codecs.open(getattr(self, menu).path, 'w', 'utf-8') as f:
- f.write(getattr(self, menu).dom.toprettyxml())
- if not from_loading:
- self.__loadMenus()
-
- def quit(self):
- for file_name in os.listdir(util.getUserItemPath()):
- if file_name[-6:-2] in ('redo', 'undo'):
- file_path = os.path.join(util.getUserItemPath(), file_name)
- os.unlink(file_path)
- for file_name in os.listdir(util.getUserDirectoryPath()):
- if file_name[-6:-2] in ('redo', 'undo'):
- file_path = os.path.join(util.getUserDirectoryPath(), file_name)
- os.unlink(file_path)
- for file_name in os.listdir(util.getUserMenuPath()):
- if file_name[-6:-2] in ('redo', 'undo'):
- file_path = os.path.join(util.getUserMenuPath(), file_name)
- os.unlink(file_path)
-
- def revert(self):
- for name in ('applications', 'settings'):
- menu = getattr(self, name)
- self.revertTree(menu.tree.root)
- path = os.path.join(util.getUserMenuPath(), menu.tree.get_menu_file())
- try:
- os.unlink(path)
- except OSError:
- pass
- #reload DOM for each menu
- try:
- menu.dom = xml.dom.minidom.parse(menu.path)
- except (IOError, xml.parsers.expat.ExpatError):
- menu.dom = xml.dom.minidom.parseString(util.getUserMenuXml(menu.tree))
- util.removeWhitespaceNodes(menu.dom)
- #reset undo/redo, no way to recover from this
- self.__undo, self.__redo = [], []
- self.save()
-
- def revertTree(self, menu):
- for child in menu.get_contents():
- if child.get_type() == matemenu.TYPE_DIRECTORY:
- self.revertTree(child)
- elif child.get_type() == matemenu.TYPE_ENTRY:
- self.revertItem(child)
- self.revertMenu(menu)
-
- def revertItem(self, item):
- if not self.canRevert(item):
- return
- self.__addUndo([item,])
- try:
- os.remove(item.get_desktop_file_path())
- except OSError:
- pass
- self.save()
-
- def revertMenu(self, menu):
- if not self.canRevert(menu):
- return
- #wtf happened here? oh well, just bail
- if not menu.get_desktop_file_path():
- return
- self.__addUndo([menu,])
- file_id = os.path.split(menu.get_desktop_file_path())[1]
- path = os.path.join(util.getUserDirectoryPath(), file_id)
- try:
- os.remove(path)
- except OSError:
- pass
- self.save()
-
- def undo(self):
- if len(self.__undo) == 0:
- return
- files = self.__undo.pop()
- redo = []
- for file_path in files:
- new_path = file_path.rsplit('.', 1)[0]
- redo_path = util.getUniqueRedoFile(new_path)
-
- f_file_path = codecs.open(new_path, 'r', 'utf-8')
- f_new_path = codecs.open(new_path, 'rw', 'utf-8')
- f_redo_path = codecs.open(redo_path, 'rw', 'utf-8')
-
- data = f_new_path.read()
- f_redo_path.write(data)
- data = f_file_path.read()
- f_new_path.write(data)
-
- f_file_path.close()
- f_new_path.close()
- f_redo_path.close()
-
- os.unlink(file_path)
- redo.append(redo_path)
- # reload DOM to make changes stick
- for name in ('applications', 'settings'):
- menu = getattr(self, name)
- try:
- menu.dom = xml.dom.minidom.parse(menu.path)
- except (IOError, xml.parsers.expat.ExpatError):
- menu.dom = xml.dom.minidom.parseString(util.getUserMenuXml(menu.tree))
- util.removeWhitespaceNodes(menu.dom)
- self.__redo.append(redo)
-
- def redo(self):
- if len(self.__redo) == 0:
- return
- files = self.__redo.pop()
- undo = []
- for file_path in files:
- new_path = file_path.rsplit('.', 1)[0]
- undo_path = util.getUniqueUndoFile(new_path)
- f_file_path = codecs.open(new_path, 'r', 'utf-8')
- f_new_path = codecs.open(new_path, 'rw', 'utf-8')
- f_undo_path = codecs.open(undo_path, 'rw', 'utf-8')
- data = f_new_path.read()
- f_undo_path.write(data)
- data = f_file_path.read()
- f_new_path.write(data)
- os.unlink(file_path)
- undo.append(undo_path)
- f_file_path.close()
- f_new_path.close()
- f_undo_path.close()
- #reload DOM to make changes stick
- for name in ('applications', 'settings'):
- menu = getattr(self, name)
- try:
- menu.dom = xml.dom.minidom.parse(menu.path)
- except (IOError, xml.parsers.expat.ExpatError):
- menu.dom = xml.dom.minidom.parseString(util.getUserMenuXml(menu.tree))
- util.removeWhitespaceNodes(menu.dom)
- self.__undo.append(undo)
-
- def getMenus(self, parent=None):
- if parent is None:
- yield self.applications.tree.root
- yield self.settings.tree.root
- else:
- for menu in parent.get_contents():
- if menu.get_type() == matemenu.TYPE_DIRECTORY:
- yield (menu, self.__isVisible(menu))
-
- def getItems(self, menu):
- for item in menu.get_contents():
- if item.get_type() == matemenu.TYPE_SEPARATOR:
- yield (item, True)
- else:
- if item.get_type() == matemenu.TYPE_ENTRY and item.get_desktop_file_id()[-19:] == '-usercustom.desktop':
- continue
- yield (item, self.__isVisible(item))
-
- def canRevert(self, item):
- if item.get_type() == matemenu.TYPE_ENTRY:
- if util.getItemPath(item.get_desktop_file_id()) is not None:
- path = util.getUserItemPath()
- if os.path.isfile(os.path.join(path, item.get_desktop_file_id())):
- return True
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- if item.get_desktop_file_path():
- file_id = os.path.split(item.get_desktop_file_path())[1]
- else:
- file_id = item.get_menu_id() + '.directory'
- if util.getDirectoryPath(file_id) is not None:
- path = util.getUserDirectoryPath()
- if os.path.isfile(os.path.join(path, file_id)):
- return True
- return False
-
- def setVisible(self, item, visible):
- dom = self.__getMenu(item).dom
- if item.get_type() == matemenu.TYPE_ENTRY:
- self.__addUndo([self.__getMenu(item), item])
- menu_xml = self.__getXmlMenu(self.__getPath(item.get_parent()), dom.documentElement, dom)
- if visible:
- self.__addXmlFilename(menu_xml, dom, item.get_desktop_file_id(), 'Include')
- self.__writeItem(item, NoDisplay=False)
- else:
- self.__addXmlFilename(menu_xml, dom, item.get_desktop_file_id(), 'Exclude')
- self.__addXmlTextElement(menu_xml, 'AppDir', util.getUserItemPath(), dom)
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- self.__addUndo([self.__getMenu(item), item])
- #don't mess with it if it's empty
- if len(item.get_contents()) == 0:
- return
- menu_xml = self.__getXmlMenu(self.__getPath(item), dom.documentElement, dom)
- for node in self.__getXmlNodesByName(['Deleted', 'NotDeleted'], menu_xml):
- node.parentNode.removeChild(node)
- self.__writeMenu(item, NoDisplay=not visible)
- self.__addXmlTextElement(menu_xml, 'DirectoryDir', util.getUserDirectoryPath(), dom)
- self.save()
-
- def createItem(self, parent, before, after, **kwargs):
- file_id = self.__writeItem(None, **kwargs)
- self.insertExternalItem(file_id, parent.menu_id, before, after)
-
- def insertExternalItem(self, file_id, parent_id, before=None, after=None):
- parent = self.__findMenu(parent_id)
- dom = self.__getMenu(parent).dom
- self.__addItem(parent, file_id, dom)
- self.__positionItem(parent, ('Item', file_id), before, after)
- self.__addUndo([self.__getMenu(parent), ('Item', file_id)])
- self.save()
-
- def insertExternalMenu(self, file_id, parent_id, before=None, after=None):
- menu_id = file_id.rsplit('.', 1)[0]
- parent = self.__findMenu(parent_id)
- dom = self.__getMenu(parent).dom
- self.__addXmlDefaultLayout(self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom) , dom)
- menu_xml = self.__getXmlMenu(self.__getPath(parent) + [menu_id], dom.documentElement, dom)
- self.__addXmlTextElement(menu_xml, 'Directory', file_id, dom)
- self.__positionItem(parent, ('Menu', menu_id), before, after)
- self.__addUndo([self.__getMenu(parent), ('Menu', file_id)])
- self.save()
-
- def createSeparator(self, parent, before=None, after=None):
- self.__positionItem(parent, ('Separator',), before, after)
- self.__addUndo([self.__getMenu(parent), ('Separator',)])
- self.save()
-
- def editItem(self, item, icon, name, comment, command, use_term, parent=None, final=True):
- #if nothing changed don't make a user copy
- if icon == item.get_icon() and name == item.get_display_name() and comment == item.get_comment() and command == item.get_exec() and use_term == item.get_launch_in_terminal():
- return
- #hack, item.get_parent() seems to fail a lot
- if not parent:
- parent = item.get_parent()
- if final:
- self.__addUndo([self.__getMenu(parent), item])
- self.__writeItem(item, Icon=icon, Name=name, Comment=comment, Exec=command, Terminal=use_term)
- if final:
- dom = self.__getMenu(parent).dom
- menu_xml = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
- self.__addXmlTextElement(menu_xml, 'AppDir', util.getUserItemPath(), dom)
- self.save()
-
- def editMenu(self, menu, icon, name, comment, final=True):
- #if nothing changed don't make a user copy
- if icon == menu.get_icon() and name == menu.get_name() and comment == menu.get_comment():
- return
- #we don't use this, we just need to make sure the <Menu> exists
- #otherwise changes won't show up
- dom = self.__getMenu(menu).dom
- menu_xml = self.__getXmlMenu(self.__getPath(menu), dom.documentElement, dom)
- self.__writeMenu(menu, Icon=icon, Name=name, Comment=comment)
- if final:
- self.__addXmlTextElement(menu_xml, 'DirectoryDir', util.getUserDirectoryPath(), dom)
- self.__addUndo([self.__getMenu(menu), menu])
- self.save()
-
- def copyItem(self, item, new_parent, before=None, after=None):
- dom = self.__getMenu(new_parent).dom
- file_path = item.get_desktop_file_path()
- keyfile = GLib.KeyFile()
- keyfile.load_from_file(file_path, util.KEY_FILE_FLAGS)
-
- util.fillKeyFile(keyfile, dict(Categories=[], Hidden=False))
-
- file_id = util.getUniqueFileId(item.get_name().replace(os.sep, '-'), '.desktop')
- out_path = os.path.join(util.getUserItemPath(), file_id)
-
- contents, length = keyfile.to_data()
-
- with open(out_path, 'w') as f:
- f.write(contents)
-
- self.__addItem(new_parent, file_id, dom)
- self.__positionItem(new_parent, ('Item', file_id), before, after)
- self.__addUndo([self.__getMenu(new_parent), ('Item', file_id)])
- self.save()
- return file_id
-
- def moveItem(self, item, new_parent, before=None, after=None):
- undo = []
- if item.get_parent() != new_parent:
- #hide old item
- self.deleteItem(item)
- undo.append(item)
- file_id = self.copyItem(item, new_parent)
- item = ('Item', file_id)
- undo.append(item)
- self.__positionItem(new_parent, item, before, after)
- undo.append(self.__getMenu(new_parent))
- self.__addUndo(undo)
- self.save()
-
- def moveMenu(self, menu, new_parent, before=None, after=None):
- parent = new_parent
- #don't move a menu into it's child
- while parent.get_parent():
- parent = parent.get_parent()
- if parent == menu:
- return False
-
- #don't move a menu into itself
- if new_parent == menu:
- return False
-
- #can't move between top-level menus
- if self.__getMenu(menu) != self.__getMenu(new_parent):
- return False
- if menu.get_parent() != new_parent:
- dom = self.__getMenu(menu).dom
- path = self.__getPath(menu)
- root_path = path[0]
- xml_root = self.__getXmlMenu(root_path, dom.documentElement, dom)
- old_path = path[1:]
- new_path = self.__getPath(new_parent)[1:] + [menu.get_menu_id()]
- self.__addXmlMove(xml_root, '/'.join(old_path), '/'.join(new_path), dom)
- self.__positionItem(new_parent, menu, before, after)
- self.__addUndo([self.__getMenu(new_parent),])
- self.save()
-
- def moveSeparator(self, separator, new_parent, before=None, after=None):
- undo = []
- # remove the original separator if its parent is not the new destination
- if separator.get_parent() != new_parent:
- self.deleteSeparator(separator)
- undo.append(separator)
- # this adds the new separator to the specified position
- self.__positionItem(new_parent, separator, before, after)
- undo.append(self.__getMenu(new_parent))
- self.__addUndo(undo)
- self.save()
-
- def deleteItem(self, item):
- self.__addUndo([item,])
- self.__writeItem(item, Hidden=True)
- self.save()
-
- def deleteMenu(self, menu):
- dom = self.__getMenu(menu).dom
- menu_xml = self.__getXmlMenu(self.__getPath(menu), dom.documentElement, dom)
- self.__addDeleted(menu_xml, dom)
- self.__addUndo([self.__getMenu(menu),])
- self.save()
-
- def deleteSeparator(self, item):
- parent = item.get_parent()
- contents = parent.get_contents()
- contents.remove(item)
- layout = self.__createLayout(contents)
- dom = self.__getMenu(parent).dom
- menu_xml = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
- self.__addXmlLayout(menu_xml, layout, dom)
- self.__addUndo([self.__getMenu(item.get_parent()),])
- self.save()
-
- #private stuff
- def __addUndo(self, items):
- self.__undo.append([])
- for item in items:
- if isinstance(item, Menu):
- file_path = item.path
- elif isinstance(item, tuple):
- if item[0] == 'Item':
- file_path = os.path.join(util.getUserItemPath(), item[1])
- if not os.path.isfile(file_path):
- file_path = util.getItemPath(item[1])
- elif item[0] == 'Menu':
- file_path = os.path.join(util.getUserDirectoryPath(), item[1])
- if not os.path.isfile(file_path):
- file_path = util.getDirectoryPath(item[1])
- else:
- continue
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- if item.get_desktop_file_path() is None:
- continue
- file_path = os.path.join(util.getUserDirectoryPath(), os.path.split(item.get_desktop_file_path())[1])
- if not os.path.isfile(file_path):
- file_path = item.get_desktop_file_path()
- elif item.get_type() == matemenu.TYPE_ENTRY:
- file_path = os.path.join(util.getUserItemPath(), item.get_desktop_file_id())
- if not os.path.isfile(file_path):
- file_path = item.get_desktop_file_path()
- else:
- continue
- with codecs.open(file_path, 'r', 'utf-8') as f:
- data = f.read()
- undo_path = util.getUniqueUndoFile(file_path)
- with codecs.open(undo_path, 'w', 'utf-8') as f:
- f.write(data)
- self.__undo[-1].append(undo_path)
-
- def __getMenu(self, item):
- root = item.get_parent()
- if not root:
- #already at the top
- root = item
- else:
- while True:
- if root.get_parent():
- root = root.get_parent()
- else:
- break
- if root.menu_id == self.applications.tree.root.menu_id:
- return self.applications
- return self.settings
-
- def __findMenu(self, menu_id, parent=None):
- if parent is None:
- menu = self.__findMenu(menu_id, self.applications.tree.root)
- if menu is not None:
- return menu
- else:
- return self.__findMenu(menu_id, self.settings.tree.root)
- if menu_id == self.applications.tree.root.menu_id:
- return self.applications.tree.root
- if menu_id == self.settings.tree.root.menu_id:
- return self.settings.tree.root
- for item in parent.get_contents():
- if item.get_type() == matemenu.TYPE_DIRECTORY:
- if item.menu_id == menu_id:
- return item
- menu = self.__findMenu(menu_id, item)
- if menu is not None:
- return menu
-
- def __isVisible(self, item):
- if item.get_type() == matemenu.TYPE_ENTRY:
- return not (item.get_is_excluded() or item.get_is_nodisplay())
- menu = self.__getMenu(item)
- if menu == self.applications:
- root = self.applications.visible_tree.root
- elif menu == self.settings:
- root = self.settings.visible_tree.root
- if item.get_type() == matemenu.TYPE_DIRECTORY:
- if self.__findMenu(item.menu_id, root) is None:
- return False
- return True
-
- def __getPath(self, menu):
- names = []
- current = menu
- while current is not None:
- names.append(current.get_menu_id())
- current = current.get_parent()
-
- # XXX - don't append root menu name, mozo doesn't
- # expect it. look into this more.
- names.pop(-1)
- return names[::-1]
-
- def __getXmlMenuPart(self, element, name):
- for node in self.__getXmlNodesByName('Menu', element):
- for child in self.__getXmlNodesByName('Name', node):
- if child.childNodes[0].nodeValue == name:
- return node
- return None
-
- def __getXmlMenu(self, path, element, dom):
- for name in path:
- found = self.__getXmlMenuPart(element, name)
- if found is not None:
- element = found
- else:
- element = self.__addXmlMenuElement(element, name, dom)
- return element
-
- def __addXmlMenuElement(self, element, name, dom):
- if isinstance(name, bytes):
- name = name.decode('utf-8')
-
- node = dom.createElement('Menu')
- self.__addXmlTextElement(node, 'Name', name, dom)
- return element.appendChild(node)
-
- def __addXmlTextElement(self, element, name, text, dom):
- if isinstance(name, bytes):
- name = name.decode('utf-8')
- if isinstance(text, bytes):
- text = text.decode('utf-8')
-
- for temp in element.childNodes:
- if temp.nodeName == name:
- if temp.childNodes[0].nodeValue == text:
- return
- node = dom.createElement(name)
- text = dom.createTextNode(text)
- node.appendChild(text)
- return element.appendChild(node)
-
- def __addXmlFilename(self, element, dom, filename, type = 'Include'):
- if isinstance(filename, bytes):
- filename = filename.decode('utf-8')
-
- # remove old filenames
- for node in self.__getXmlNodesByName(['Include', 'Exclude'], element):
- if node.childNodes[0].nodeName == 'Filename' and node.childNodes[0].childNodes[0].nodeValue == filename:
- element.removeChild(node)
-
- # add new filename
- node = dom.createElement(type)
- node.appendChild(self.__addXmlTextElement(node, 'Filename', filename, dom))
- return element.appendChild(node)
-
- def __addDeleted(self, element, dom):
- node = dom.createElement('Deleted')
- return element.appendChild(node)
-
- def __makeKeyFile(self, file_path, kwargs):
- if 'KeyFile' in kwargs:
- return kwargs['KeyFile']
-
- keyfile = GLib.KeyFile()
-
- if file_path is not None:
- keyfile.load_from_file(file_path, util.KEY_FILE_FLAGS)
-
- util.fillKeyFile(keyfile, kwargs)
- return keyfile
-
- def __writeItem(self, item, **kwargs):
- if item is not None:
- file_path = item.get_desktop_file_path()
- else:
- file_path = None
-
- keyfile = self.__makeKeyFile(file_path, kwargs)
-
- if item is not None:
- file_id = item.get_desktop_file_id()
- else:
- file_id = util.getUniqueFileId(keyfile.get_string(GLib.KEY_FILE_DESKTOP_GROUP, 'Name'), '.desktop')
-
- contents, length = keyfile.to_data()
-
- with open(os.path.join(util.getUserItemPath(), file_id), 'w') as f:
- f.write(contents)
- return file_id
-
- def __writeMenu(self, menu, **kwargs):
- if menu is not None:
- file_id = os.path.split(menu.get_desktop_file_path())[1]
- file_path = menu.get_desktop_file_path()
- keyfile = GLib.KeyFile()
- keyfile.load_from_file(file_path, util.KEY_FILE_FLAGS)
- elif menu is None and 'Name' not in kwargs:
- raise Exception('New menus need a name')
- else:
- file_id = util.getUniqueFileId(kwargs['Name'], '.directory')
- keyfile = GLib.KeyFile()
-
- util.fillKeyFile(keyfile, kwargs)
-
- contents, length = keyfile.to_data()
-
- with open(os.path.join(util.getUserDirectoryPath(), file_id), 'w') as f:
- f.write(contents)
- return file_id
-
- def __getXmlNodesByName(self, name, element):
- for child in element.childNodes:
- if child.nodeType == xml.dom.Node.ELEMENT_NODE:
- if isinstance(name, str) and child.nodeName == name:
- yield child
- elif isinstance(name, list) or isinstance(name, tuple):
- if child.nodeName in name:
- yield child
-
- def __addXmlMove(self, element, old, new, dom):
- if not self.__undoMoves(element, old, new, dom):
- node = dom.createElement('Move')
- node.appendChild(self.__addXmlTextElement(node, 'Old', old, dom))
- node.appendChild(self.__addXmlTextElement(node, 'New', new, dom))
- #are parsed in reverse order, need to put at the beginning
- return element.insertBefore(node, element.firstChild)
-
- def __addXmlLayout(self, element, layout, dom):
- # remove old layout
- for node in self.__getXmlNodesByName('Layout', element):
- element.removeChild(node)
-
- # add new layout
- node = dom.createElement('Layout')
- for order in layout.order:
- if order[0] == 'Separator':
- child = dom.createElement('Separator')
- node.appendChild(child)
- elif order[0] == 'Filename':
- child = self.__addXmlTextElement(node, 'Filename', order[1], dom)
- elif order[0] == 'Menuname':
- child = self.__addXmlTextElement(node, 'Menuname', order[1], dom)
- elif order[0] == 'Merge':
- child = dom.createElement('Merge')
- child.setAttribute('type', order[1])
- node.appendChild(child)
- return element.appendChild(node)
-
- def __addXmlDefaultLayout(self, element, dom):
- # remove old default layout
- for node in self.__getXmlNodesByName('DefaultLayout', element):
- element.removeChild(node)
-
- # add new layout
- node = dom.createElement('DefaultLayout')
- node.setAttribute('inline', 'false')
- return element.appendChild(node)
-
- def __createLayout(self, items):
- layout = Layout()
- layout.order = []
-
- layout.order.append(['Merge', 'menus'])
- for item in items:
- if isinstance(item, tuple):
- if item[0] == 'Separator':
- layout.parseSeparator()
- elif item[0] == 'Menu':
- layout.parseMenuname(item[1])
- elif item[0] == 'Item':
- layout.parseFilename(item[1])
- elif item.get_type() == matemenu.TYPE_DIRECTORY:
- layout.parseMenuname(item.get_menu_id())
- elif item.get_type() == matemenu.TYPE_ENTRY:
- layout.parseFilename(item.get_desktop_file_id())
- elif item.get_type() == matemenu.TYPE_SEPARATOR:
- layout.parseSeparator()
- layout.order.append(['Merge', 'files'])
- return layout
-
- def __addItem(self, parent, file_id, dom):
- xml_parent = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
- self.__addXmlFilename(xml_parent, dom, file_id, 'Include')
-
- def __positionItem(self, parent, item, before=None, after=None):
- if after:
- index = parent.contents.index(after) + 1
- elif before:
- index = parent.contents.index(before)
- else:
- # append the item to the list
- index = len(parent.contents)
- contents = parent.contents
- #if this is a move to a new parent you can't remove the item
- if item in contents:
- # decrease the destination index, if we shorten the list
- if (before and (contents.index(item) < index)) \
- or (after and (contents.index(item) < index - 1)):
- index -= 1
- contents.remove(item)
- contents.insert(index, item)
- layout = self.__createLayout(contents)
- dom = self.__getMenu(parent).dom
- menu_xml = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
- self.__addXmlLayout(menu_xml, layout, dom)
-
- def __undoMoves(self, element, old, new, dom):
- nodes = []
- matches = []
- original_old = old
- final_old = old
- #get all <Move> elements
- for node in self.__getXmlNodesByName(['Move'], element):
- nodes.insert(0, node)
- #if the <New> matches our old parent we've found a stage to undo
- for node in nodes:
- xml_old = node.getElementsByTagName('Old')[0]
- xml_new = node.getElementsByTagName('New')[0]
- if xml_new.childNodes[0].nodeValue == old:
- matches.append(node)
- #we should end up with this path when completed
- final_old = xml_old.childNodes[0].nodeValue
- #undoing <Move>s
- for node in matches:
- element.removeChild(node)
- if len(matches) > 0:
- for node in nodes:
- xml_old = node.getElementsByTagName('Old')[0]
- xml_new = node.getElementsByTagName('New')[0]
- path = os.path.split(xml_new.childNodes[0].nodeValue)
- if path[0] == original_old:
- element.removeChild(node)
- for node in dom.getElementsByTagName('Menu'):
- name_node = node.getElementsByTagName('Name')[0]
- name = name_node.childNodes[0].nodeValue
- if name == os.path.split(new)[1]:
- #copy app and dir directory info from old <Menu>
- root_path = dom.getElementsByTagName('Menu')[0].getElementsByTagName('Name')[0].childNodes[0].nodeValue
- xml_menu = self.__getXmlMenu(root_path + '/' + new, dom.documentElement, dom)
- for app_dir in node.getElementsByTagName('AppDir'):
- xml_menu.appendChild(app_dir)
- for dir_dir in node.getElementsByTagName('DirectoryDir'):
- xml_menu.appendChild(dir_dir)
- parent = node.parentNode
- parent.removeChild(node)
- node = dom.createElement('Move')
- node.appendChild(self.__addXmlTextElement(node, 'Old', xml_old.childNodes[0].nodeValue, dom))
- node.appendChild(self.__addXmlTextElement(node, 'New', os.path.join(new, path[1]), dom))
- element.appendChild(node)
- if final_old == new:
- return True
- node = dom.createElement('Move')
- node.appendChild(self.__addXmlTextElement(node, 'Old', final_old, dom))
- node.appendChild(self.__addXmlTextElement(node, 'New', new, dom))
- return element.appendChild(node)
+ def __init__(self):
+ self.locale = locale.getdefaultlocale()[0]
+ self.__loadMenus()
+ self.__undo = []
+ self.__redo = []
+
+
+ def __loadMenus(self):
+ self.applications = Menu()
+ self.applications.tree = matemenu.lookup_tree('mate-applications.menu', matemenu.FLAGS_SHOW_EMPTY|matemenu.FLAGS_INCLUDE_EXCLUDED|matemenu.FLAGS_INCLUDE_NODISPLAY|matemenu.FLAGS_SHOW_ALL_SEPARATORS)
+ self.applications.visible_tree = matemenu.lookup_tree('mate-applications.menu')
+ self.applications.tree.sort_key = matemenu.SORT_DISPLAY_NAME
+ self.applications.visible_tree.sort_key = matemenu.SORT_DISPLAY_NAME
+ self.applications.path = os.path.join(util.getUserMenuPath(), self.applications.tree.get_menu_file())
+ try:
+ self.applications.dom = xml.dom.minidom.parse(self.applications.path)
+ except (IOError, xml.parsers.expat.ExpatError):
+ self.applications.dom = xml.dom.minidom.parseString(util.getUserMenuXml(self.applications.tree))
+ util.removeWhitespaceNodes(self.applications.dom)
+
+ self.settings = Menu()
+ self.settings.tree = matemenu.lookup_tree('mate-settings.menu', matemenu.FLAGS_SHOW_EMPTY|matemenu.FLAGS_INCLUDE_EXCLUDED|matemenu.FLAGS_INCLUDE_NODISPLAY|matemenu.FLAGS_SHOW_ALL_SEPARATORS)
+ self.settings.visible_tree = matemenu.lookup_tree('mate-settings.menu')
+ self.settings.tree.sort_key = matemenu.SORT_DISPLAY_NAME
+ self.settings.visible_tree.sort_key = matemenu.SORT_DISPLAY_NAME
+ self.settings.path = os.path.join(util.getUserMenuPath(), self.settings.tree.get_menu_file())
+ try:
+ self.settings.dom = xml.dom.minidom.parse(self.settings.path)
+ except (IOError, xml.parsers.expat.ExpatError):
+ self.settings.dom = xml.dom.minidom.parseString(util.getUserMenuXml(self.settings.tree))
+ util.removeWhitespaceNodes(self.settings.dom)
+
+ self.save(True)
+
+ def save(self, from_loading=False):
+ for menu in ('applications', 'settings'):
+ with codecs.open(getattr(self, menu).path, 'w', 'utf-8') as f:
+ f.write(getattr(self, menu).dom.toprettyxml())
+ if not from_loading:
+ self.__loadMenus()
+
+ def quit(self):
+ for file_name in os.listdir(util.getUserItemPath()):
+ if file_name[-6:-2] in ('redo', 'undo'):
+ file_path = os.path.join(util.getUserItemPath(), file_name)
+ os.unlink(file_path)
+ for file_name in os.listdir(util.getUserDirectoryPath()):
+ if file_name[-6:-2] in ('redo', 'undo'):
+ file_path = os.path.join(util.getUserDirectoryPath(), file_name)
+ os.unlink(file_path)
+ for file_name in os.listdir(util.getUserMenuPath()):
+ if file_name[-6:-2] in ('redo', 'undo'):
+ file_path = os.path.join(util.getUserMenuPath(), file_name)
+ os.unlink(file_path)
+
+ def revert(self):
+ for name in ('applications', 'settings'):
+ menu = getattr(self, name)
+ self.revertTree(menu.tree.root)
+ path = os.path.join(util.getUserMenuPath(), menu.tree.get_menu_file())
+ try:
+ os.unlink(path)
+ except OSError:
+ pass
+ #reload DOM for each menu
+ try:
+ menu.dom = xml.dom.minidom.parse(menu.path)
+ except (IOError, xml.parsers.expat.ExpatError):
+ menu.dom = xml.dom.minidom.parseString(util.getUserMenuXml(menu.tree))
+ util.removeWhitespaceNodes(menu.dom)
+ #reset undo/redo, no way to recover from this
+ self.__undo, self.__redo = [], []
+ self.save()
+
+ def revertTree(self, menu):
+ for child in menu.get_contents():
+ if child.get_type() == matemenu.TYPE_DIRECTORY:
+ self.revertTree(child)
+ elif child.get_type() == matemenu.TYPE_ENTRY:
+ self.revertItem(child)
+ self.revertMenu(menu)
+
+ def revertItem(self, item):
+ if not self.canRevert(item):
+ return
+ self.__addUndo([item,])
+ try:
+ os.remove(item.get_desktop_file_path())
+ except OSError:
+ pass
+ self.save()
+
+ def revertMenu(self, menu):
+ if not self.canRevert(menu):
+ return
+ #wtf happened here? oh well, just bail
+ if not menu.get_desktop_file_path():
+ return
+ self.__addUndo([menu,])
+ file_id = os.path.split(menu.get_desktop_file_path())[1]
+ path = os.path.join(util.getUserDirectoryPath(), file_id)
+ try:
+ os.remove(path)
+ except OSError:
+ pass
+ self.save()
+
+ def undo(self):
+ if len(self.__undo) == 0:
+ return
+ files = self.__undo.pop()
+ redo = []
+ for file_path in files:
+ new_path = file_path.rsplit('.', 1)[0]
+ redo_path = util.getUniqueRedoFile(new_path)
+
+ f_file_path = codecs.open(new_path, 'r', 'utf-8')
+ f_new_path = codecs.open(new_path, 'rw', 'utf-8')
+ f_redo_path = codecs.open(redo_path, 'rw', 'utf-8')
+
+ data = f_new_path.read()
+ f_redo_path.write(data)
+ data = f_file_path.read()
+ f_new_path.write(data)
+
+ f_file_path.close()
+ f_new_path.close()
+ f_redo_path.close()
+
+ os.unlink(file_path)
+ redo.append(redo_path)
+ # reload DOM to make changes stick
+ for name in ('applications', 'settings'):
+ menu = getattr(self, name)
+ try:
+ menu.dom = xml.dom.minidom.parse(menu.path)
+ except (IOError, xml.parsers.expat.ExpatError):
+ menu.dom = xml.dom.minidom.parseString(util.getUserMenuXml(menu.tree))
+ util.removeWhitespaceNodes(menu.dom)
+ self.__redo.append(redo)
+
+ def redo(self):
+ if len(self.__redo) == 0:
+ return
+ files = self.__redo.pop()
+ undo = []
+ for file_path in files:
+ new_path = file_path.rsplit('.', 1)[0]
+ undo_path = util.getUniqueUndoFile(new_path)
+ f_file_path = codecs.open(new_path, 'r', 'utf-8')
+ f_new_path = codecs.open(new_path, 'rw', 'utf-8')
+ f_undo_path = codecs.open(undo_path, 'rw', 'utf-8')
+ data = f_new_path.read()
+ f_undo_path.write(data)
+ data = f_file_path.read()
+ f_new_path.write(data)
+ os.unlink(file_path)
+ undo.append(undo_path)
+ f_file_path.close()
+ f_new_path.close()
+ f_undo_path.close()
+ #reload DOM to make changes stick
+ for name in ('applications', 'settings'):
+ menu = getattr(self, name)
+ try:
+ menu.dom = xml.dom.minidom.parse(menu.path)
+ except (IOError, xml.parsers.expat.ExpatError):
+ menu.dom = xml.dom.minidom.parseString(util.getUserMenuXml(menu.tree))
+ util.removeWhitespaceNodes(menu.dom)
+ self.__undo.append(undo)
+
+ def getMenus(self, parent=None):
+ if parent is None:
+ yield self.applications.tree.root
+ yield self.settings.tree.root
+ else:
+ for menu in parent.get_contents():
+ if menu.get_type() == matemenu.TYPE_DIRECTORY:
+ yield (menu, self.__isVisible(menu))
+
+ def getItems(self, menu):
+ for item in menu.get_contents():
+ if item.get_type() == matemenu.TYPE_SEPARATOR:
+ yield (item, True)
+ else:
+ if item.get_type() == matemenu.TYPE_ENTRY and item.get_desktop_file_id()[-19:] == '-usercustom.desktop':
+ continue
+ yield (item, self.__isVisible(item))
+
+ def canRevert(self, item):
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ if util.getItemPath(item.get_desktop_file_id()) is not None:
+ path = util.getUserItemPath()
+ if os.path.isfile(os.path.join(path, item.get_desktop_file_id())):
+ return True
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ if item.get_desktop_file_path():
+ file_id = os.path.split(item.get_desktop_file_path())[1]
+ else:
+ file_id = item.get_menu_id() + '.directory'
+ if util.getDirectoryPath(file_id) is not None:
+ path = util.getUserDirectoryPath()
+ if os.path.isfile(os.path.join(path, file_id)):
+ return True
+ return False
+
+ def setVisible(self, item, visible):
+ dom = self.__getMenu(item).dom
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ self.__addUndo([self.__getMenu(item), item])
+ menu_xml = self.__getXmlMenu(self.__getPath(item.get_parent()), dom.documentElement, dom)
+ if visible:
+ self.__addXmlFilename(menu_xml, dom, item.get_desktop_file_id(), 'Include')
+ self.__writeItem(item, NoDisplay=False)
+ else:
+ self.__addXmlFilename(menu_xml, dom, item.get_desktop_file_id(), 'Exclude')
+ self.__addXmlTextElement(menu_xml, 'AppDir', util.getUserItemPath(), dom)
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ self.__addUndo([self.__getMenu(item), item])
+ #don't mess with it if it's empty
+ if len(item.get_contents()) == 0:
+ return
+ menu_xml = self.__getXmlMenu(self.__getPath(item), dom.documentElement, dom)
+ for node in self.__getXmlNodesByName(['Deleted', 'NotDeleted'], menu_xml):
+ node.parentNode.removeChild(node)
+ self.__writeMenu(item, NoDisplay=not visible)
+ self.__addXmlTextElement(menu_xml, 'DirectoryDir', util.getUserDirectoryPath(), dom)
+ self.save()
+
+ def createItem(self, parent, before, after, **kwargs):
+ file_id = self.__writeItem(None, **kwargs)
+ self.insertExternalItem(file_id, parent.menu_id, before, after)
+
+ def insertExternalItem(self, file_id, parent_id, before=None, after=None):
+ parent = self.__findMenu(parent_id)
+ dom = self.__getMenu(parent).dom
+ self.__addItem(parent, file_id, dom)
+ self.__positionItem(parent, ('Item', file_id), before, after)
+ self.__addUndo([self.__getMenu(parent), ('Item', file_id)])
+ self.save()
+
+ def insertExternalMenu(self, file_id, parent_id, before=None, after=None):
+ menu_id = file_id.rsplit('.', 1)[0]
+ parent = self.__findMenu(parent_id)
+ dom = self.__getMenu(parent).dom
+ self.__addXmlDefaultLayout(self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom) , dom)
+ menu_xml = self.__getXmlMenu(self.__getPath(parent) + [menu_id], dom.documentElement, dom)
+ self.__addXmlTextElement(menu_xml, 'Directory', file_id, dom)
+ self.__positionItem(parent, ('Menu', menu_id), before, after)
+ self.__addUndo([self.__getMenu(parent), ('Menu', file_id)])
+ self.save()
+
+ def createSeparator(self, parent, before=None, after=None):
+ self.__positionItem(parent, ('Separator',), before, after)
+ self.__addUndo([self.__getMenu(parent), ('Separator',)])
+ self.save()
+
+ def editItem(self, item, icon, name, comment, command, use_term, parent=None, final=True):
+ #if nothing changed don't make a user copy
+ if icon == item.get_icon() and name == item.get_display_name() and comment == item.get_comment() and command == item.get_exec() and use_term == item.get_launch_in_terminal():
+ return
+ #hack, item.get_parent() seems to fail a lot
+ if not parent:
+ parent = item.get_parent()
+ if final:
+ self.__addUndo([self.__getMenu(parent), item])
+ self.__writeItem(item, Icon=icon, Name=name, Comment=comment, Exec=command, Terminal=use_term)
+ if final:
+ dom = self.__getMenu(parent).dom
+ menu_xml = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
+ self.__addXmlTextElement(menu_xml, 'AppDir', util.getUserItemPath(), dom)
+ self.save()
+
+ def editMenu(self, menu, icon, name, comment, final=True):
+ #if nothing changed don't make a user copy
+ if icon == menu.get_icon() and name == menu.get_name() and comment == menu.get_comment():
+ return
+ #we don't use this, we just need to make sure the <Menu> exists
+ #otherwise changes won't show up
+ dom = self.__getMenu(menu).dom
+ menu_xml = self.__getXmlMenu(self.__getPath(menu), dom.documentElement, dom)
+ self.__writeMenu(menu, Icon=icon, Name=name, Comment=comment)
+ if final:
+ self.__addXmlTextElement(menu_xml, 'DirectoryDir', util.getUserDirectoryPath(), dom)
+ self.__addUndo([self.__getMenu(menu), menu])
+ self.save()
+
+ def copyItem(self, item, new_parent, before=None, after=None):
+ dom = self.__getMenu(new_parent).dom
+ file_path = item.get_desktop_file_path()
+ keyfile = GLib.KeyFile()
+ keyfile.load_from_file(file_path, util.KEY_FILE_FLAGS)
+
+ util.fillKeyFile(keyfile, dict(Categories=[], Hidden=False))
+
+ file_id = util.getUniqueFileId(item.get_name().replace(os.sep, '-'), '.desktop')
+ out_path = os.path.join(util.getUserItemPath(), file_id)
+
+ contents, length = keyfile.to_data()
+
+ with open(out_path, 'w') as f:
+ f.write(contents)
+
+ self.__addItem(new_parent, file_id, dom)
+ self.__positionItem(new_parent, ('Item', file_id), before, after)
+ self.__addUndo([self.__getMenu(new_parent), ('Item', file_id)])
+ self.save()
+ return file_id
+
+ def moveItem(self, item, new_parent, before=None, after=None):
+ undo = []
+ if item.get_parent() != new_parent:
+ #hide old item
+ self.deleteItem(item)
+ undo.append(item)
+ file_id = self.copyItem(item, new_parent)
+ item = ('Item', file_id)
+ undo.append(item)
+ self.__positionItem(new_parent, item, before, after)
+ undo.append(self.__getMenu(new_parent))
+ self.__addUndo(undo)
+ self.save()
+
+ def moveMenu(self, menu, new_parent, before=None, after=None):
+ parent = new_parent
+ #don't move a menu into it's child
+ while parent.get_parent():
+ parent = parent.get_parent()
+ if parent == menu:
+ return False
+
+ #don't move a menu into itself
+ if new_parent == menu:
+ return False
+
+ #can't move between top-level menus
+ if self.__getMenu(menu) != self.__getMenu(new_parent):
+ return False
+ if menu.get_parent() != new_parent:
+ dom = self.__getMenu(menu).dom
+ path = self.__getPath(menu)
+ root_path = path[0]
+ xml_root = self.__getXmlMenu(root_path, dom.documentElement, dom)
+ old_path = path[1:]
+ new_path = self.__getPath(new_parent)[1:] + [menu.get_menu_id()]
+ self.__addXmlMove(xml_root, '/'.join(old_path), '/'.join(new_path), dom)
+ self.__positionItem(new_parent, menu, before, after)
+ self.__addUndo([self.__getMenu(new_parent),])
+ self.save()
+
+ def moveSeparator(self, separator, new_parent, before=None, after=None):
+ undo = []
+ # remove the original separator if its parent is not the new destination
+ if separator.get_parent() != new_parent:
+ self.deleteSeparator(separator)
+ undo.append(separator)
+ # this adds the new separator to the specified position
+ self.__positionItem(new_parent, separator, before, after)
+ undo.append(self.__getMenu(new_parent))
+ self.__addUndo(undo)
+ self.save()
+
+ def deleteItem(self, item):
+ self.__addUndo([item,])
+ self.__writeItem(item, Hidden=True)
+ self.save()
+
+ def deleteMenu(self, menu):
+ dom = self.__getMenu(menu).dom
+ menu_xml = self.__getXmlMenu(self.__getPath(menu), dom.documentElement, dom)
+ self.__addDeleted(menu_xml, dom)
+ self.__addUndo([self.__getMenu(menu),])
+ self.save()
+
+ def deleteSeparator(self, item):
+ parent = item.get_parent()
+ contents = parent.get_contents()
+ contents.remove(item)
+ layout = self.__createLayout(contents)
+ dom = self.__getMenu(parent).dom
+ menu_xml = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
+ self.__addXmlLayout(menu_xml, layout, dom)
+ self.__addUndo([self.__getMenu(item.get_parent()),])
+ self.save()
+
+ #private stuff
+ def __addUndo(self, items):
+ self.__undo.append([])
+ for item in items:
+ if isinstance(item, Menu):
+ file_path = item.path
+ elif isinstance(item, tuple):
+ if item[0] == 'Item':
+ file_path = os.path.join(util.getUserItemPath(), item[1])
+ if not os.path.isfile(file_path):
+ file_path = util.getItemPath(item[1])
+ elif item[0] == 'Menu':
+ file_path = os.path.join(util.getUserDirectoryPath(), item[1])
+ if not os.path.isfile(file_path):
+ file_path = util.getDirectoryPath(item[1])
+ else:
+ continue
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ if item.get_desktop_file_path() is None:
+ continue
+ file_path = os.path.join(util.getUserDirectoryPath(), os.path.split(item.get_desktop_file_path())[1])
+ if not os.path.isfile(file_path):
+ file_path = item.get_desktop_file_path()
+ elif item.get_type() == matemenu.TYPE_ENTRY:
+ file_path = os.path.join(util.getUserItemPath(), item.get_desktop_file_id())
+ if not os.path.isfile(file_path):
+ file_path = item.get_desktop_file_path()
+ else:
+ continue
+ with codecs.open(file_path, 'r', 'utf-8') as f:
+ data = f.read()
+ undo_path = util.getUniqueUndoFile(file_path)
+ with codecs.open(undo_path, 'w', 'utf-8') as f:
+ f.write(data)
+ self.__undo[-1].append(undo_path)
+
+ def __getMenu(self, item):
+ root = item.get_parent()
+ if not root:
+ #already at the top
+ root = item
+ else:
+ while True:
+ if root.get_parent():
+ root = root.get_parent()
+ else:
+ break
+ if root.menu_id == self.applications.tree.root.menu_id:
+ return self.applications
+ return self.settings
+
+ def __findMenu(self, menu_id, parent=None):
+ if parent is None:
+ menu = self.__findMenu(menu_id, self.applications.tree.root)
+ if menu is not None:
+ return menu
+ else:
+ return self.__findMenu(menu_id, self.settings.tree.root)
+ if menu_id == self.applications.tree.root.menu_id:
+ return self.applications.tree.root
+ if menu_id == self.settings.tree.root.menu_id:
+ return self.settings.tree.root
+ for item in parent.get_contents():
+ if item.get_type() == matemenu.TYPE_DIRECTORY:
+ if item.menu_id == menu_id:
+ return item
+ menu = self.__findMenu(menu_id, item)
+ if menu is not None:
+ return menu
+
+ def __isVisible(self, item):
+ if item.get_type() == matemenu.TYPE_ENTRY:
+ return not (item.get_is_excluded() or item.get_is_nodisplay())
+ menu = self.__getMenu(item)
+ if menu == self.applications:
+ root = self.applications.visible_tree.root
+ elif menu == self.settings:
+ root = self.settings.visible_tree.root
+ if item.get_type() == matemenu.TYPE_DIRECTORY:
+ if self.__findMenu(item.menu_id, root) is None:
+ return False
+ return True
+
+ def __getPath(self, menu):
+ names = []
+ current = menu
+ while current is not None:
+ names.append(current.get_menu_id())
+ current = current.get_parent()
+
+ # XXX - don't append root menu name, mozo doesn't
+ # expect it. look into this more.
+ names.pop(-1)
+ return names[::-1]
+
+ def __getXmlMenuPart(self, element, name):
+ for node in self.__getXmlNodesByName('Menu', element):
+ for child in self.__getXmlNodesByName('Name', node):
+ if child.childNodes[0].nodeValue == name:
+ return node
+ return None
+
+ def __getXmlMenu(self, path, element, dom):
+ for name in path:
+ found = self.__getXmlMenuPart(element, name)
+ if found is not None:
+ element = found
+ else:
+ element = self.__addXmlMenuElement(element, name, dom)
+ return element
+
+ def __addXmlMenuElement(self, element, name, dom):
+ if isinstance(name, bytes):
+ name = name.decode('utf-8')
+
+ node = dom.createElement('Menu')
+ self.__addXmlTextElement(node, 'Name', name, dom)
+ return element.appendChild(node)
+
+ def __addXmlTextElement(self, element, name, text, dom):
+ if isinstance(name, bytes):
+ name = name.decode('utf-8')
+ if isinstance(text, bytes):
+ text = text.decode('utf-8')
+
+ for temp in element.childNodes:
+ if temp.nodeName == name:
+ if temp.childNodes[0].nodeValue == text:
+ return
+ node = dom.createElement(name)
+ text = dom.createTextNode(text)
+ node.appendChild(text)
+ return element.appendChild(node)
+
+ def __addXmlFilename(self, element, dom, filename, type = 'Include'):
+ if isinstance(filename, bytes):
+ filename = filename.decode('utf-8')
+
+ # remove old filenames
+ for node in self.__getXmlNodesByName(['Include', 'Exclude'], element):
+ if node.childNodes[0].nodeName == 'Filename' and node.childNodes[0].childNodes[0].nodeValue == filename:
+ element.removeChild(node)
+
+ # add new filename
+ node = dom.createElement(type)
+ node.appendChild(self.__addXmlTextElement(node, 'Filename', filename, dom))
+ return element.appendChild(node)
+
+ def __addDeleted(self, element, dom):
+ node = dom.createElement('Deleted')
+ return element.appendChild(node)
+
+ def __makeKeyFile(self, file_path, kwargs):
+ if 'KeyFile' in kwargs:
+ return kwargs['KeyFile']
+
+ keyfile = GLib.KeyFile()
+
+ if file_path is not None:
+ keyfile.load_from_file(file_path, util.KEY_FILE_FLAGS)
+
+ util.fillKeyFile(keyfile, kwargs)
+ return keyfile
+
+ def __writeItem(self, item, **kwargs):
+ if item is not None:
+ file_path = item.get_desktop_file_path()
+ else:
+ file_path = None
+
+ keyfile = self.__makeKeyFile(file_path, kwargs)
+
+ if item is not None:
+ file_id = item.get_desktop_file_id()
+ else:
+ file_id = util.getUniqueFileId(keyfile.get_string(GLib.KEY_FILE_DESKTOP_GROUP, 'Name'), '.desktop')
+
+ contents, length = keyfile.to_data()
+
+ with open(os.path.join(util.getUserItemPath(), file_id), 'w') as f:
+ f.write(contents)
+ return file_id
+
+ def __writeMenu(self, menu, **kwargs):
+ if menu is not None:
+ file_id = os.path.split(menu.get_desktop_file_path())[1]
+ file_path = menu.get_desktop_file_path()
+ keyfile = GLib.KeyFile()
+ keyfile.load_from_file(file_path, util.KEY_FILE_FLAGS)
+ elif menu is None and 'Name' not in kwargs:
+ raise Exception('New menus need a name')
+ else:
+ file_id = util.getUniqueFileId(kwargs['Name'], '.directory')
+ keyfile = GLib.KeyFile()
+
+ util.fillKeyFile(keyfile, kwargs)
+
+ contents, length = keyfile.to_data()
+
+ with open(os.path.join(util.getUserDirectoryPath(), file_id), 'w') as f:
+ f.write(contents)
+ return file_id
+
+ def __getXmlNodesByName(self, name, element):
+ for child in element.childNodes:
+ if child.nodeType == xml.dom.Node.ELEMENT_NODE:
+ if isinstance(name, str) and child.nodeName == name:
+ yield child
+ elif isinstance(name, list) or isinstance(name, tuple):
+ if child.nodeName in name:
+ yield child
+
+ def __addXmlMove(self, element, old, new, dom):
+ if not self.__undoMoves(element, old, new, dom):
+ node = dom.createElement('Move')
+ node.appendChild(self.__addXmlTextElement(node, 'Old', old, dom))
+ node.appendChild(self.__addXmlTextElement(node, 'New', new, dom))
+ #are parsed in reverse order, need to put at the beginning
+ return element.insertBefore(node, element.firstChild)
+
+ def __addXmlLayout(self, element, layout, dom):
+ # remove old layout
+ for node in self.__getXmlNodesByName('Layout', element):
+ element.removeChild(node)
+
+ # add new layout
+ node = dom.createElement('Layout')
+ for order in layout.order:
+ if order[0] == 'Separator':
+ child = dom.createElement('Separator')
+ node.appendChild(child)
+ elif order[0] == 'Filename':
+ child = self.__addXmlTextElement(node, 'Filename', order[1], dom)
+ elif order[0] == 'Menuname':
+ child = self.__addXmlTextElement(node, 'Menuname', order[1], dom)
+ elif order[0] == 'Merge':
+ child = dom.createElement('Merge')
+ child.setAttribute('type', order[1])
+ node.appendChild(child)
+ return element.appendChild(node)
+
+ def __addXmlDefaultLayout(self, element, dom):
+ # remove old default layout
+ for node in self.__getXmlNodesByName('DefaultLayout', element):
+ element.removeChild(node)
+
+ # add new layout
+ node = dom.createElement('DefaultLayout')
+ node.setAttribute('inline', 'false')
+ return element.appendChild(node)
+
+ def __createLayout(self, items):
+ layout = Layout()
+ layout.order = []
+
+ layout.order.append(['Merge', 'menus'])
+ for item in items:
+ if isinstance(item, tuple):
+ if item[0] == 'Separator':
+ layout.parseSeparator()
+ elif item[0] == 'Menu':
+ layout.parseMenuname(item[1])
+ elif item[0] == 'Item':
+ layout.parseFilename(item[1])
+ elif item.get_type() == matemenu.TYPE_DIRECTORY:
+ layout.parseMenuname(item.get_menu_id())
+ elif item.get_type() == matemenu.TYPE_ENTRY:
+ layout.parseFilename(item.get_desktop_file_id())
+ elif item.get_type() == matemenu.TYPE_SEPARATOR:
+ layout.parseSeparator()
+ layout.order.append(['Merge', 'files'])
+ return layout
+
+ def __addItem(self, parent, file_id, dom):
+ xml_parent = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
+ self.__addXmlFilename(xml_parent, dom, file_id, 'Include')
+
+ def __positionItem(self, parent, item, before=None, after=None):
+ if after:
+ index = parent.contents.index(after) + 1
+ elif before:
+ index = parent.contents.index(before)
+ else:
+ # append the item to the list
+ index = len(parent.contents)
+ contents = parent.contents
+ #if this is a move to a new parent you can't remove the item
+ if item in contents:
+ # decrease the destination index, if we shorten the list
+ if (before and (contents.index(item) < index)) \
+ or (after and (contents.index(item) < index - 1)):
+ index -= 1
+ contents.remove(item)
+ contents.insert(index, item)
+ layout = self.__createLayout(contents)
+ dom = self.__getMenu(parent).dom
+ menu_xml = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
+ self.__addXmlLayout(menu_xml, layout, dom)
+
+ def __undoMoves(self, element, old, new, dom):
+ nodes = []
+ matches = []
+ original_old = old
+ final_old = old
+ #get all <Move> elements
+ for node in self.__getXmlNodesByName(['Move'], element):
+ nodes.insert(0, node)
+ #if the <New> matches our old parent we've found a stage to undo
+ for node in nodes:
+ xml_old = node.getElementsByTagName('Old')[0]
+ xml_new = node.getElementsByTagName('New')[0]
+ if xml_new.childNodes[0].nodeValue == old:
+ matches.append(node)
+ #we should end up with this path when completed
+ final_old = xml_old.childNodes[0].nodeValue
+ #undoing <Move>s
+ for node in matches:
+ element.removeChild(node)
+ if len(matches) > 0:
+ for node in nodes:
+ xml_old = node.getElementsByTagName('Old')[0]
+ xml_new = node.getElementsByTagName('New')[0]
+ path = os.path.split(xml_new.childNodes[0].nodeValue)
+ if path[0] == original_old:
+ element.removeChild(node)
+ for node in dom.getElementsByTagName('Menu'):
+ name_node = node.getElementsByTagName('Name')[0]
+ name = name_node.childNodes[0].nodeValue
+ if name == os.path.split(new)[1]:
+ #copy app and dir directory info from old <Menu>
+ root_path = dom.getElementsByTagName('Menu')[0].getElementsByTagName('Name')[0].childNodes[0].nodeValue
+ xml_menu = self.__getXmlMenu(root_path + '/' + new, dom.documentElement, dom)
+ for app_dir in node.getElementsByTagName('AppDir'):
+ xml_menu.appendChild(app_dir)
+ for dir_dir in node.getElementsByTagName('DirectoryDir'):
+ xml_menu.appendChild(dir_dir)
+ parent = node.parentNode
+ parent.removeChild(node)
+ node = dom.createElement('Move')
+ node.appendChild(self.__addXmlTextElement(node, 'Old', xml_old.childNodes[0].nodeValue, dom))
+ node.appendChild(self.__addXmlTextElement(node, 'New', os.path.join(new, path[1]), dom))
+ element.appendChild(node)
+ if final_old == new:
+ return True
+ node = dom.createElement('Move')
+ node.appendChild(self.__addXmlTextElement(node, 'Old', final_old, dom))
+ node.appendChild(self.__addXmlTextElement(node, 'New', new, dom))
+ return element.appendChild(node)
class Layout:
- def __init__(self, node=None):
- self.order = []
+ def __init__(self, node=None):
+ self.order = []
- def parseMenuname(self, value):
- self.order.append(['Menuname', value])
+ def parseMenuname(self, value):
+ self.order.append(['Menuname', value])
- def parseSeparator(self):
- self.order.append(['Separator'])
+ def parseSeparator(self):
+ self.order.append(['Separator'])
- def parseFilename(self, value):
- self.order.append(['Filename', value])
+ def parseFilename(self, value):
+ self.order.append(['Filename', value])
- def parseMerge(self, merge_type='all'):
- self.order.append(['Merge', merge_type])
+ def parseMerge(self, merge_type='all'):
+ self.order.append(['Merge', merge_type])
diff --git a/Mozo/util.py b/Mozo/util.py
index 54dcf1f..bb9e1c6 100644
--- a/Mozo/util.py
+++ b/Mozo/util.py
@@ -27,154 +27,154 @@ DESKTOP_GROUP = GLib.KEY_FILE_DESKTOP_GROUP
KEY_FILE_FLAGS = GLib.KeyFileFlags.KEEP_COMMENTS | GLib.KeyFileFlags.KEEP_TRANSLATIONS
def fillKeyFile(keyfile, items):
- for key, item in items.iteritems():
- if item is None:
- continue
+ for key, item in items.iteritems():
+ if item is None:
+ continue
- if isinstance(item, bool):
- keyfile.set_boolean(DESKTOP_GROUP, key, item)
- elif isinstance(item, Sequence):
- keyfile.set_string_list(DESKTOP_GROUP, key, item)
- elif isinstance(item, basestring):
- keyfile.set_string(DESKTOP_GROUP, key, item)
+ if isinstance(item, bool):
+ keyfile.set_boolean(DESKTOP_GROUP, key, item)
+ elif isinstance(item, Sequence):
+ keyfile.set_string_list(DESKTOP_GROUP, key, item)
+ elif isinstance(item, basestring):
+ keyfile.set_string(DESKTOP_GROUP, key, item)
def getUniqueFileId(name, extension):
- append = 0
- while 1:
- if append == 0:
- filename = name + extension
- else:
- filename = name + '-' + str(append) + extension
- if extension == '.desktop':
- path = getUserItemPath()
- if not os.path.isfile(os.path.join(path, filename)) and not getItemPath(filename):
- break
- elif extension == '.directory':
- path = getUserDirectoryPath()
- if not os.path.isfile(os.path.join(path, filename)) and not getDirectoryPath(filename):
- break
- append += 1
- return filename
+ append = 0
+ while 1:
+ if append == 0:
+ filename = name + extension
+ else:
+ filename = name + '-' + str(append) + extension
+ if extension == '.desktop':
+ path = getUserItemPath()
+ if not os.path.isfile(os.path.join(path, filename)) and not getItemPath(filename):
+ break
+ elif extension == '.directory':
+ path = getUserDirectoryPath()
+ if not os.path.isfile(os.path.join(path, filename)) and not getDirectoryPath(filename):
+ break
+ append += 1
+ return filename
def getUniqueRedoFile(filepath):
- append = 0
- while 1:
- new_filepath = filepath + '.redo-' + str(append)
- if not os.path.isfile(new_filepath):
- break
- else:
- append += 1
- return new_filepath
+ append = 0
+ while 1:
+ new_filepath = filepath + '.redo-' + str(append)
+ if not os.path.isfile(new_filepath):
+ break
+ else:
+ append += 1
+ return new_filepath
def getUniqueUndoFile(filepath):
- filename, extension = os.path.split(filepath)[1].rsplit('.', 1)
- append = 0
- while 1:
- if extension == 'desktop':
- path = getUserItemPath()
- elif extension == 'directory':
- path = getUserDirectoryPath()
- elif extension == 'menu':
- path = getUserMenuPath()
- new_filepath = os.path.join(path, filename + '.' + extension + '.undo-' + str(append))
- if not os.path.isfile(new_filepath):
- break
- else:
- append += 1
- return new_filepath
+ filename, extension = os.path.split(filepath)[1].rsplit('.', 1)
+ append = 0
+ while 1:
+ if extension == 'desktop':
+ path = getUserItemPath()
+ elif extension == 'directory':
+ path = getUserDirectoryPath()
+ elif extension == 'menu':
+ path = getUserMenuPath()
+ new_filepath = os.path.join(path, filename + '.' + extension + '.undo-' + str(append))
+ if not os.path.isfile(new_filepath):
+ break
+ else:
+ append += 1
+ return new_filepath
def getItemPath(file_id):
- for path in GLib.get_system_data_dirs():
- file_path = os.path.join(path, 'applications', file_id)
- if os.path.isfile(file_path):
- return file_path
- return None
+ for path in GLib.get_system_data_dirs():
+ file_path = os.path.join(path, 'applications', file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ return None
def getUserItemPath():
- item_dir = os.path.join(GLib.get_user_data_dir(), 'applications')
- if not os.path.isdir(item_dir):
- os.makedirs(item_dir)
- return item_dir
+ item_dir = os.path.join(GLib.get_user_data_dir(), 'applications')
+ if not os.path.isdir(item_dir):
+ os.makedirs(item_dir)
+ return item_dir
def getDirectoryPath(file_id):
- for path in GLib.get_system_data_dirs():
- file_path = os.path.join(path, 'desktop-directories', file_id)
- if os.path.isfile(file_path):
- return file_path
- return None
+ for path in GLib.get_system_data_dirs():
+ file_path = os.path.join(path, 'desktop-directories', file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ return None
def getUserDirectoryPath():
- menu_dir = os.path.join(GLib.get_user_data_dir(), 'desktop-directories')
- if not os.path.isdir(menu_dir):
- os.makedirs(menu_dir)
- return menu_dir
+ menu_dir = os.path.join(GLib.get_user_data_dir(), 'desktop-directories')
+ if not os.path.isdir(menu_dir):
+ os.makedirs(menu_dir)
+ return menu_dir
def getUserMenuPath():
- menu_dir = os.path.join(GLib.get_user_config_dir(), 'menus')
- if not os.path.isdir(menu_dir):
- os.makedirs(menu_dir)
- return menu_dir
+ menu_dir = os.path.join(GLib.get_user_config_dir(), 'menus')
+ if not os.path.isdir(menu_dir):
+ os.makedirs(menu_dir)
+ return menu_dir
def getSystemMenuPath(file_id):
- for path in GLib.get_system_config_dirs():
- file_path = os.path.join(path, 'menus', file_id)
- if os.path.isfile(file_path):
- return file_path
- return None
+ for path in GLib.get_system_config_dirs():
+ file_path = os.path.join(path, 'menus', file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ return None
def getUserMenuXml(tree):
- system_file = getSystemMenuPath(tree.get_menu_file())
- name = tree.root.get_menu_id()
- menu_xml = "<!DOCTYPE Menu PUBLIC '-//freedesktop//DTD Menu 1.0//EN' 'http://standards.freedesktop.org/menu-spec/menu-1.0.dtd'>\n"
- menu_xml += "<Menu>\n <Name>" + name + "</Name>\n "
- menu_xml += "<MergeFile type=\"parent\">" + system_file + "</MergeFile>\n</Menu>\n"
- return menu_xml
+ system_file = getSystemMenuPath(tree.get_menu_file())
+ name = tree.root.get_menu_id()
+ menu_xml = "<!DOCTYPE Menu PUBLIC '-//freedesktop//DTD Menu 1.0//EN' 'http://standards.freedesktop.org/menu-spec/menu-1.0.dtd'>\n"
+ menu_xml += "<Menu>\n <Name>" + name + "</Name>\n "
+ menu_xml += "<MergeFile type=\"parent\">" + system_file + "</MergeFile>\n</Menu>\n"
+ return menu_xml
def getIcon(item):
- pixbuf, path = None, None
- if item is None:
- return None
- if isinstance(item, str):
- iconName = item
- else:
- iconName = item.get_icon()
- if iconName and not '/' in iconName and iconName[-3:] in ('png', 'svg', 'xpm'):
- iconName = iconName[:-4]
- icon_theme = Gtk.IconTheme.get_default()
- try:
- pixbuf = icon_theme.load_icon(iconName, 24, 0)
- path = icon_theme.lookup_icon(iconName, 24, 0).get_filename()
- except:
- if iconName and '/' in iconName:
- try:
- pixbuf = GdkPixbuf.Pixbuf.new_from_file_at_size(iconName, 24, 24)
- path = iconName
- except:
- pass
- if pixbuf is None:
- if item.get_type() == matemenu.TYPE_DIRECTORY:
- iconName = 'mate-fs-directory'
- elif item.get_type() == matemenu.TYPE_ENTRY:
- iconName = 'application-default-icon'
- try:
- pixbuf = icon_theme.load_icon(iconName, 24, 0)
- path = icon_theme.lookup_icon(iconName, 24, 0).get_filename()
- except:
- return None
- if pixbuf is None:
- return None
- if pixbuf.get_width() != 24 or pixbuf.get_height() != 24:
- pixbuf = pixbuf.scale_simple(24, 24, GdkPixbuf.InterpType.HYPER)
- return pixbuf
+ pixbuf, path = None, None
+ if item is None:
+ return None
+ if isinstance(item, str):
+ iconName = item
+ else:
+ iconName = item.get_icon()
+ if iconName and not '/' in iconName and iconName[-3:] in ('png', 'svg', 'xpm'):
+ iconName = iconName[:-4]
+ icon_theme = Gtk.IconTheme.get_default()
+ try:
+ pixbuf = icon_theme.load_icon(iconName, 24, 0)
+ path = icon_theme.lookup_icon(iconName, 24, 0).get_filename()
+ except:
+ if iconName and '/' in iconName:
+ try:
+ pixbuf = GdkPixbuf.Pixbuf.new_from_file_at_size(iconName, 24, 24)
+ path = iconName
+ except:
+ pass
+ if pixbuf is None:
+ if item.get_type() == matemenu.TYPE_DIRECTORY:
+ iconName = 'mate-fs-directory'
+ elif item.get_type() == matemenu.TYPE_ENTRY:
+ iconName = 'application-default-icon'
+ try:
+ pixbuf = icon_theme.load_icon(iconName, 24, 0)
+ path = icon_theme.lookup_icon(iconName, 24, 0).get_filename()
+ except:
+ return None
+ if pixbuf is None:
+ return None
+ if pixbuf.get_width() != 24 or pixbuf.get_height() != 24:
+ pixbuf = pixbuf.scale_simple(24, 24, GdkPixbuf.InterpType.HYPER)
+ return pixbuf
def removeWhitespaceNodes(node):
- remove_list = []
- for child in node.childNodes:
- if child.nodeType == xml.dom.minidom.Node.TEXT_NODE:
- child.data = child.data.strip()
- if not child.data.strip():
- remove_list.append(child)
- elif child.hasChildNodes():
- removeWhitespaceNodes(child)
- for node in remove_list:
- node.parentNode.removeChild(node)
+ remove_list = []
+ for child in node.childNodes:
+ if child.nodeType == xml.dom.minidom.Node.TEXT_NODE:
+ child.data = child.data.strip()
+ if not child.data.strip():
+ remove_list.append(child)
+ elif child.hasChildNodes():
+ removeWhitespaceNodes(child)
+ for node in remove_list:
+ node.parentNode.removeChild(node)