summaryrefslogtreecommitdiff
path: root/Mozo
diff options
context:
space:
mode:
Diffstat (limited to 'Mozo')
-rw-r--r--Mozo/MainWindow.py8
-rw-r--r--Mozo/MenuEditor.py80
2 files changed, 41 insertions, 47 deletions
diff --git a/Mozo/MainWindow.py b/Mozo/MainWindow.py
index 6b92227..93a9e4c 100644
--- a/Mozo/MainWindow.py
+++ b/Mozo/MainWindow.py
@@ -98,7 +98,7 @@ class MainWindow:
update_items = False
update_type = None
item_id = None
- if iter:
+ if iter and items[iter][3].get_desktop_file_path():
update_items = True
if isinstance(items[iter][3], MateMenu.TreeDirectory):
item_id = os.path.split(items[iter][3].get_desktop_file_path())[1]
@@ -425,7 +425,7 @@ class MainWindow:
item = self.drag_data
new_parent = menus[path][2]
if isinstance(item, MateMenu.TreeEntry):
- self.editor.copyItem(item, new_parent)
+ self.editor.moveItem(item, new_parent)
elif isinstance(item, MateMenu.TreeDirectory):
if not self.editor.moveMenu(item, new_parent):
self.loadUpdates()
@@ -593,7 +593,7 @@ class MainWindow:
item = items[path][3]
before = items[(path.get_indices()[0] - 1,)][3]
if isinstance(item, MateMenu.TreeEntry):
- self.editor.moveItem(item.get_parent(), item, before=before)
+ self.editor.moveItem(item, item.get_parent(), before=before)
elif isinstance(item, MateMenu.TreeDirectory):
self.editor.moveMenu(item, item.get_parent(), before=before)
elif isinstance(item, MateMenu.TreeSeparator):
@@ -611,7 +611,7 @@ class MainWindow:
item = items[path][3]
after = items[path][3]
if isinstance(item, MateMenu.TreeEntry):
- self.editor.moveItem(item.get_parent(), item, after=after)
+ self.editor.moveItem(item, item.get_parent(), after=after)
elif isinstance(item, MateMenu.TreeDirectory):
self.editor.moveMenu(item, item.get_parent(), after=after)
elif isinstance(item, MateMenu.TreeSeparator):
diff --git a/Mozo/MenuEditor.py b/Mozo/MenuEditor.py
index f94d591..c9b83a5 100644
--- a/Mozo/MenuEditor.py
+++ b/Mozo/MenuEditor.py
@@ -18,7 +18,6 @@
import codecs
import os
-import re
import xml.dom.minidom
import xml.parsers.expat
import locale
@@ -167,25 +166,22 @@ class MenuEditor(object):
return
files = self.__undo.pop()
redo = []
- for file_path in files:
- new_path = file_path.rsplit('.', 1)[0]
+ for undo_path in files[::-1]:
+ new_path = undo_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)
+ # create redo file
+ with codecs.open(new_path, 'r', 'utf-8') as f_new:
+ with codecs.open(redo_path, 'w', 'utf-8') as f_redo:
+ f_redo.write(f_new.read())
+ redo.append(redo_path)
- f_file_path.close()
- f_new_path.close()
- f_redo_path.close()
+ # restore undo file
+ with codecs.open(undo_path, 'r', 'utf-8') as f_undo:
+ with codecs.open(new_path, 'w', 'utf-8') as f_new:
+ f_new.write(f_undo.read())
+ os.unlink(undo_path)
- os.unlink(file_path)
- redo.append(redo_path)
# reload DOM to make changes stick
for name in ('applications', 'settings'):
menu = getattr(self, name)
@@ -201,21 +197,22 @@ class MenuEditor(object):
return
files = self.__redo.pop()
undo = []
- for file_path in files:
- new_path = file_path.rsplit('.', 1)[0]
+ for redo_path in files[::-1]:
+ new_path = redo_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)
+
+ # create undo file
+ with codecs.open(new_path, 'r', 'utf-8') as f_new:
+ with codecs.open(undo_path, 'w', 'utf-8') as f_undo:
+ f_undo.write(f_new.read())
undo.append(undo_path)
- f_file_path.close()
- f_new_path.close()
- f_undo_path.close()
+
+ # restore redo file
+ with codecs.open(redo_path, 'r', 'utf-8') as f_redo:
+ with codecs.open(new_path, 'w', 'utf-8') as f_new:
+ f_new.write(f_redo.read())
+ os.unlink(redo_path)
+
#reload DOM to make changes stick
for name in ('applications', 'settings'):
menu = getattr(self, name)
@@ -376,14 +373,12 @@ class MenuEditor(object):
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))
- app_info = item.get_app_info()
- file_id = util.getUniqueFileId(app_info.get_name().replace(os.sep, '-'), '.desktop')
+ file_id = util.getUniqueFileId(os.path.basename(file_path).split(".desktop", 1)[0], ".desktop")
out_path = os.path.join(util.getUserItemPath(), file_id)
- contents, length = keyfile.to_data()
+ contents = keyfile.to_data()[0]
with open(out_path, 'w') as f:
f.write(contents)
@@ -397,15 +392,18 @@ class MenuEditor(object):
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)
+ # create new item
file_id = self.copyItem(item, new_parent)
+ # hide old item
+ self.deleteItem(item)
item = ('Item', file_id)
- undo.append(item)
self.__positionItem(new_parent, item, before, after)
undo.append(self.__getMenu(new_parent))
self.__addUndo(undo)
+ # merge the undo entries created by the previous operations in the
+ # correct order so we can undo in one go
+ items = self.__undo.pop()[::-1] + self.__undo.pop()[::-1] + self.__undo.pop()[::-1]
+ self.__undo.append(items[::-1])
self.save()
def moveMenu(self, menu, new_parent, before=None, after=None):
@@ -653,7 +651,7 @@ class MenuEditor(object):
else:
file_id = util.getUniqueFileId(keyfile.get_string(GLib.KEY_FILE_DESKTOP_GROUP, 'Name'), '.desktop')
- contents, length = keyfile.to_data()
+ contents = keyfile.to_data()[0]
with open(os.path.join(util.getUserItemPath(), file_id), 'w') as f:
f.write(contents)
@@ -673,7 +671,7 @@ class MenuEditor(object):
util.fillKeyFile(keyfile, kwargs)
- contents, length = keyfile.to_data()
+ contents = keyfile.to_data()[0]
with open(os.path.join(util.getUserDirectoryPath(), file_id), 'w') as f:
f.write(contents)
@@ -753,10 +751,6 @@ class MenuEditor(object):
xml_parent = self.__getXmlMenu(self.__getPath(parent), dom.documentElement, dom)
self.__addXmlFilename(xml_parent, dom, file_id, 'Include')
- def moveItem(self, parent, item, before=None, after=None):
- self.__positionItem(parent, item, before=before, after=after)
- self.save()
-
def __positionItem(self, parent, item, before=None, after=None):
contents = self.getContents(parent)
if after: