1 files changed, 16 insertions, 0 deletions
diff --git a/usr/share/mageiawelcome/helpers.py b/usr/share/mageiawelcome/helpers.py
index cac5578..1ad236b 100644
--- a/usr/share/mageiawelcome/helpers.py
+++ b/usr/share/mageiawelcome/helpers.py
@@ -4,6 +4,22 @@
 import csv
 import os
 
+def get_desktop_name(x):
+  return {
+    '01KDE4':'KDE',
+    '02GNOME':'Gnome',
+    'LXDE':'LXDE',
+    '06xfce4':'XFCE',
+    '10MATE':'Mate',
+    '10Cinnamon':'Cinnamon',
+    '05RazorDesktop':'RazorQt',
+    '23E17':'Enlightenment',
+    '07IceWM':'IceWM',
+    '26Openbox':'Openbox',
+    '03WindowMaker':'WindowMaker',
+    '09Fvwm2':'Fvwm2',
+  }.get(x,'Other')
+
 def is_installed(name):
   return os.WEXITSTATUS(os.system('rpm -q --quiet ' + name))