Added icons for categories

src/xdgmenumaker

@@ -14,28 +14,40 @@ desktop = False
 
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-applications.directory')
 applications = de.getName().encode('utf-8')
+applications_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-accessories.directory')
 accessories = de.getName().encode('utf-8')
+accessories_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-development.directory')
 development = de.getName().encode('utf-8')
+development_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-education.directory')
 education = de.getName().encode('utf-8')
+education_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-games.directory')
 games = de.getName().encode('utf-8')
+games_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-graphics.directory')
 graphics = de.getName().encode('utf-8')
+graphics_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-multimedia.directory')
 multimedia = de.getName().encode('utf-8')
+multimedia_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-network.directory')
 network = de.getName().encode('utf-8')
+network_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-office.directory')
 office = de.getName().encode('utf-8')
+office_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-settings.directory')
 settings = de.getName().encode('utf-8')
+settings_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-system.directory')
 system = de.getName().encode('utf-8')
+system_icon = de.getIcon()
 de = dentry.DesktopEntry(filename = '/usr/share/desktop-directories/xdgmenumaker-other.directory')
 other = de.getName().encode('utf-8')
+other_icon = de.getIcon()
 
 def main(argv):
 	global desktop
@@ -88,16 +100,8 @@ class MenuEntry:
 	def __repr__(self):
 		return repr((self.category, self.name, self.icon, self.command))
 
-def get_entry_info(desktopfile):
-	global desktop
-	global seticon
-	show = True
-	de = dentry.DesktopEntry(filename = desktopfile)
-	name = de.getName().encode('utf-8')
-
-	if seticon == True:
-		# strip the directory and extension from the icon name
-		icon = os.path.basename(de.getIcon())
+def icon_full_path(icon):
+	icon = os.path.basename(icon)
 	if icon.endswith('.png'):
 		icon = icon.replace('.png', '')
 	elif icon.endswith('.svg'):
@@ -108,6 +112,19 @@ def get_entry_info(desktopfile):
 	icon = icon_theme.lookup_icon(icon, 16, gtk.ICON_LOOKUP_NO_SVG)
 	if icon:
 		icon = icon.get_filename()
+	return icon
+
+def get_entry_info(desktopfile):
+	global desktop
+	global seticon
+	show = True
+	de = dentry.DesktopEntry(filename = desktopfile)
+	name = de.getName().encode('utf-8')
+
+	if seticon == True:
+		# strip the directory and extension from the icon name
+		icon = de.getIcon()
+		icon = icon_full_path(icon)
 	else:
 		icon = None
 
@@ -184,7 +201,6 @@ def desktopfilelist():
 		filelist.append(localdir+'/'+i)
 	return filelist
 
-
 def menu():
 	applist = []
 	for desktopfile in desktopfilelist():
@@ -207,10 +223,49 @@ def menu():
 		menu.append([c, appsincategory])
 	return menu
 
+def category_icon(category):
+	if category == accessories:
+		icon = accessories_icon
+	elif category == development:
+		icon = development_icon
+	elif category == education:
+		icon = education_icon
+	elif category == games:
+		icon = games_icon
+	elif category == graphics:
+		icon = graphics_icon
+	elif category == multimedia:
+		icon = multimedia_icon
+	elif category == network:
+		icon = network_icon
+	elif category == office:
+		icon = office_icon
+	elif category == settings:
+		icon = settings_icon
+	elif category == system:
+		icon = system_icon
+	elif category == other:
+		icon = other_icon
+	else:
+		icon = None
+	return icon
+
 def fluxboxmenu():
+	if seticon == True:
+		app_icon = icon_full_path(applications_icon)
+		print '[submenu] ('+applications+') <'+app_icon+'>'
+	else:
 		print '[submenu] ('+applications+')'
 	for i in menu():
 		category = i[0]
+		if seticon == True:
+			cat_icon = category_icon(category)
+			cat_icon = icon_full_path(cat_icon)
+			if cat_icon:
+				print '  [submenu] ('+category+') <'+cat_icon+'>'
+			else:
+				print '  [submenu] ('+category+')'
+		else:
 			print '  [submenu] ('+category+')'
 		for j in i[1]:
 			name = j[0]