aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/users.py
blob: e5124f0b0cbe1fc4f5a899cf5694281ad02d2753 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
# -*- 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/>.
'''

import os

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 (all might not be listed)
    '''
    
    
    def __init__(self, method = 'devfs'):
        '''
        Constructor
        
        @param  method:str|()→itr<str>  The user listing method
        '''
        users = Users.who
        if not isinstance(method, str):
            users = method
        elif method == 'who':    users = Users.who
        elif method == 'w':      users = Users.w
        elif method == 'devfs':  users = Users.devfs
        
        users = users()
        self.users = {}
        for (user, tty) in users:
            if user not in self.users:
                self.users[user] = []
            self.users[user].append(tty)
    
    
    @staticmethod
    def who():
        '''
        Use Coreutil's `who` command to fetch users and ttys
        
        @return  :list<(user:str, tty:str>)  List of user–tty pairs
        '''
        users = spawn('who').read().decode('utf-8', 'replace').split('\n')[:-1]
        users = [list(filter(lambda cell : not cell == '', line.split(' ')))[:2] for line in users]
        return users
    
    
    @staticmethod
    def w():
        '''
        Use procps-ng's `w` command to fetch users and ttys
        
        @return  :list<(user:str, tty:str>)  List of user–tty pairs
        '''
        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]
        return users
    
    
    @staticmethod
    def devfs(try_to_find_root = False):
        '''
        Walk /dev to find TTY acquisitions
        
        @param   try_to_find_root:bool       Whether to list root for /dev/tty[[:numeric:]]+ if the permission bits looks good
        @return  :list<(user:str, tty:str>)  List of user–tty pairs
        '''
        import pwd
        
        def num(s):
            return (not s == '') and (0 == len(list(filter(lambda c : not ('0' <= c <= '9'), s))))
        
        ttys = [f for f in os.listdir('/dev') if f.startswith('tty') and num(f[3:])]
        ttys += ['pts/' + f for f in os.listdir('/dev/pts') if num(f)]
        
        users, rc = {}, []
        for tty in ttys:
            attr = os.stat('/dev/' + tty)
            if ('/' not in tty) and (attr.st_uid == 0):
                if not (try_to_find_root and (attr.st_mode == 0o20600)):
                    continue
            user = attr.st_uid
            if user in users:
                user = users[user]
            else:
                try:
                    users[user] = user = pwd.getpwuid(user).pw_name
                except:
                    users[user] = user = str(user)
            rc.append((user, tty))
        return rc