summaryrefslogtreecommitdiff
path: root/Mozo/util.py
diff options
context:
space:
mode:
Diffstat (limited to 'Mozo/util.py')
-rw-r--r--Mozo/util.py244
1 files changed, 244 insertions, 0 deletions
diff --git a/Mozo/util.py b/Mozo/util.py
new file mode 100644
index 0000000..2fb123c
--- /dev/null
+++ b/Mozo/util.py
@@ -0,0 +1,244 @@
+# -*- coding: utf-8 -*-
+# Mozo Menu Editor - Simple fd.o Compliant Menu Editor
+# Copyright (C) 2006 Travis Watkins
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Library General Public
+# License as published by the Free Software Foundation; either
+# version 2 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Library General Public License for more details.
+#
+# You should have received a copy of the GNU Library General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+import os
+import gtk, matemenu
+from ConfigParser import ConfigParser
+
+class DesktopParser(ConfigParser):
+ def __init__(self, filename=None, file_type='Application'):
+ ConfigParser.__init__(self)
+ self.filename = filename
+ self.file_type = file_type
+ if filename:
+ if len(self.read(filename)) == 0:
+ #file doesn't exist
+ self.add_section('Desktop Entry')
+ else:
+ self.add_section('Desktop Entry')
+ self._list_separator = ';'
+
+ def optionxform(self, option):
+ #makes keys not be lowercase
+ return option
+
+ def get(self, option, locale=None):
+ locale_option = option + '[%s]' % locale
+ try:
+ value = ConfigParser.get(self, 'Desktop Entry', locale_option)
+ except:
+ try:
+ value = ConfigParser.get(self, 'Desktop Entry', option)
+ except:
+ return None
+ if self._list_separator in value:
+ value = value.split(self._list_separator)
+ if value == 'true':
+ value = True
+ if value == 'false':
+ value = False
+ return value
+
+ def set(self, option, value, locale=None):
+ if locale:
+ option = option + '[%s]' % locale
+ if value == True:
+ value = 'true'
+ if value == False:
+ value = 'false'
+ if isinstance(value, tuple) or isinstance(value, list):
+ value = self._list_separator.join(value) + ';'
+ ConfigParser.set(self, 'Desktop Entry', option, value)
+
+ def write(self, file_object):
+ file_object.write('[Desktop Entry]\n')
+ items = []
+ if not self.filename:
+ file_object.write('Encoding=UTF-8\n')
+ file_object.write('Type=' + str(self.file_type) + '\n')
+ for item in self.items('Desktop Entry'):
+ items.append(item)
+ items.sort()
+ for item in items:
+ file_object.write(item[0] + '=' + item[1] + '\n')
+
+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
+
+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
+
+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
+
+def getUserMenuPath():
+ menu_dir = None
+ if os.environ.has_key('XDG_CONFIG_HOME'):
+ menu_dir = os.path.join(os.environ['XDG_CONFIG_HOME'], 'menus')
+ else:
+ menu_dir = os.path.join(os.environ['HOME'], '.config', 'menus')
+ #move .config out of the way if it's not a dir, it shouldn't be there
+ if os.path.isfile(os.path.split(menu_dir)[0]):
+ os.rename(os.path.split(menu_dir)[0], os.path.split(menu_dir)[0] + '.old')
+ if not os.path.isdir(menu_dir):
+ os.makedirs(menu_dir)
+ return menu_dir
+
+def getItemPath(file_id):
+ if os.environ.has_key('XDG_DATA_DIRS'):
+ for system_path in os.environ['XDG_DATA_DIRS'].split(':'):
+ file_path = os.path.join(system_path, 'applications', file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ file_path = os.path.join('/', 'usr', 'share', 'applications', file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ return False
+
+def getUserItemPath():
+ item_dir = None
+ if os.environ.has_key('XDG_DATA_HOME'):
+ item_dir = os.path.join(os.environ['XDG_DATA_HOME'], 'applications')
+ else:
+ item_dir = os.path.join(os.environ['HOME'], '.local', 'share', 'applications')
+ if not os.path.isdir(item_dir):
+ os.makedirs(item_dir)
+ return item_dir
+
+def getDirectoryPath(file_id):
+ home = getUserDirectoryPath()
+ file_path = os.path.join(home, file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ if os.environ.has_key('XDG_DATA_DIRS'):
+ for system_path in os.environ['XDG_DATA_DIRS'].split(':'):
+ file_path = os.path.join(system_path, 'desktop-directories', file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ file_path = os.path.join('/', 'usr', 'share', 'desktop-directories', file_id)
+ if os.path.isfile(file_path):
+ return file_path
+ return False
+
+def getUserDirectoryPath():
+ menu_dir = None
+ if os.environ.has_key('XDG_DATA_HOME'):
+ menu_dir = os.path.join(os.environ['XDG_DATA_HOME'], 'desktop-directories')
+ else:
+ menu_dir = os.path.join(os.environ['HOME'], '.local', 'share', 'desktop-directories')
+ if not os.path.isdir(menu_dir):
+ os.makedirs(menu_dir)
+ return menu_dir
+
+def getSystemMenuPath(file_name):
+ if os.environ.has_key('XDG_CONFIG_DIRS'):
+ for system_path in os.environ['XDG_CONFIG_DIRS'].split(':'):
+ file_path = os.path.join(system_path, 'menus', file_name)
+ if os.path.isfile(file_path):
+ return file_path
+ file_path = os.path.join('/', 'etc', 'xdg', 'menus', file_name)
+ if os.path.isfile(file_path):
+ return file_path
+ return False
+
+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
+
+def getIcon(item, for_properties=False):
+ pixbuf, path = None, None
+ if item == None:
+ if for_properties:
+ return None, 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.icon_theme_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 = gtk.gdk.pixbuf_new_from_file_at_size(iconName, 24, 24)
+ path = iconName
+ except:
+ pass
+ if pixbuf == None:
+ if for_properties:
+ return None, 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 == None:
+ return None
+ if pixbuf.get_width() != 24 or pixbuf.get_height() != 24:
+ pixbuf = pixbuf.scale_simple(24, 24, gtk.gdk.INTERP_HYPER)
+ if for_properties:
+ return pixbuf, path
+ return pixbuf