aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/cpuonline.py
blob: 5c5be4e6400d8c68720894d95265d2fe6b63be72 (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
# -*- python -*-
'''
xpybar – xmobar replacement written in python
Copyright © 2014, 2015, 2016, 2017, 2018, 2019  Mattias Andrée (maandree@kth.se)

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 CPUOnline: # TODO make it possible to turn CPU:s on and off
    '''
    Online CPU listing
    
    @variable  offline:list<int>   Offline CPU:s
    @variable  online:list<int>    Online CPU:s
    @variable  possible:list<int>  Possible CPU:s
    @variable  present:list<int>   Present CPU:s
    '''
    
    
    def __init__(self):
        '''
        Constructor
        '''
        def expand(data):
            if '-' not in data:
                return [int(data)]
            range_ = lambda a, b : list(range(a, b + 1))
            return range_(*[int(x) for x in data.split('-')])
        
        data = []
        for filename in ('offline', 'online', 'possible', 'present'):
            with open('/sys/devices/system/cpu/' + filename, 'rb') as file:
                data.append(file.read())
        
        data = [x.decode('utf-8', 'replace').replace('\n', ' ').replace(',', ' ') for x in data]
        data = [map(expand, filter(lambda item : not item == '', x.split(' '))) for x in data]
        data = [reduce(lambda x, y : x + y, list(x)) for x in data]
        data = [x if x is not None else [] for x in data]
        
        (self.offline, self.online, self.possible, self.present) = data