now the wm.py can detect console

lazyscripts/wm.py

@@ -34,13 +34,17 @@ def __init__(self, dist=None):
         if not dist:
             dist = distro.Distribution().name
         self.distro = dist
-        self.name = self.get_wminfo()
-        if self.distro == 'ubuntu' and \
-           distro.Distribution().version == '11.04' and \
-           self.name == 'gnome':
-            self.unity_check()
-
-        self.version = self.get_version()
+        if self.is_under_X():
+            self.name = self.get_wminfo()
+            if self.distro == 'ubuntu' and \
+               distro.Distribution().version == '11.04' and \
+               self.name == 'gnome':
+                self.unity_check()
+
+            self.version = self.get_version()
+        else:
+            self.name = 'console'
+            self.version = None
 
     def __str__(self):
         return self.name
@@ -157,5 +161,12 @@ def unity_check(self):
         elif session in ('kde-plasma'):
             self.name = 'kde'
 
+    def is_under_X(self):
+        display = getenv('DISPLAY')
+        if display:
+            return True
+        else:
+            return False
+
 
 #END