diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/plugins/users.py | 42 | ||||
-rw-r--r-- | src/util.py | 2 |
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 |