add users

Signed-off-by: Mattias Andrée <maandree@operamail.com>

2 files changed, 43 insertions, 1 deletions

diff --git a/src/plugins/users.py b/src/plugins/users.py
new file mode 100644
index 0000000..a73939e
--- /dev/null
+++ b/src/plugins/users.py
@@ -0,0 +1,42 @@
+# -*- python -*-
+'''
+xpybar – xmobar replacement written in python
+Copyright © 2014  Mattias Andrée (maandree@member.fsf.org)
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program 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 General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+'''
+
+from util import *
+
+
+class Users:
+    '''
+    Gets the logged in users and what TTY:s they are using
+    
+    @variable  users:dict<str, list<str>>  Map from logged in users to acquired TTY:s
+    '''
+    
+    
+    def __init__(self):
+        '''
+        Constructor
+        '''
+        users = spawn('w').read().decode('utf-8', 'replace').split('\n')[2 : -1]
+        users = [list(filter(lambda cell : not cell == '', line.split(' ')))[:2] for line in users]
+        self.users = {}
+        for (user, tty) in users:
+            if user not in self.users:
+                self.users[user] = []
+            self.users[user].append(tty)
+
diff --git a/src/util.py b/src/util.py
index d12d3b2..25aac1f 100644
--- a/src/util.py
+++ b/src/util.py
@@ -55,6 +55,6 @@ def spawn(*command):
     @param   command:*str  The command line
     @return  :istream      The process's stdout
     '''
-    proc = process.Popen(list(command), stderr = sys.stderr, stdout = PIPE)
+    proc = subprocess.Popen(list(command), stderr = sys.stderr, stdout = subprocess.PIPE)
     return proc.stdout