aboutsummaryrefslogtreecommitdiffstats
path: root/examples/plugin-test
blob: a012542d7e83df4c54995e2aa0137fc7d59c5d0f (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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
# -*- python -*-

# A simple xpybar configuration example that tests the plugins

import os
import time

from plugins.uptime import Uptime
from plugins.loadavg import AverageLoad
from plugins.users import Users
from plugins.pacman import Pacman
from plugins.uname import Uname
from plugins.df import Discs
from plugins.mem import Memory
from plugins.discstats import DiscStats
from plugins.xdisplay import XDisplay
from plugins.moc import MOC
from plugins.cpu import CPU
from plugins.clock import Clock
from plugins.network import Network


OUTPUT, HEIGHT, YPOS, TOP = 0, 3 * 12, 24, True


clock = Clock(format = '%Y-(%m)%b-%d %T, %a w%V, %Z')


def redraw():
    date = clock.read()
    
    uptime_ = Uptime()
    uptime = '%id %02i:%02i:%05.2f' % uptime_.uptime
    prc_idle = uptime_.average_idle_seconds / uptime_.uptime_seconds
    tot_idle = '%id %02i:%02i:%05.2f' % uptime_.total_idle
    avg_idle = '%id %02i:%02i:%05.2f' % uptime_.average_idle
    uptime = 'Uptime: %s' % (uptime)
    idle = 'Idle time: %s : %s : %4.2f%%' % (avg_idle, tot_idle, prc_idle * 100)
    
    loadavg_ = AverageLoad()
    avg_5 = int(loadavg_.average_avg_5_min * 100)
    avg_10 = int(loadavg_.average_avg_10_min * 100)
    avg_15 = int(loadavg_.average_avg_15_min * 100)
    act_tasks = loadavg_.active_tasks
    tot_tasks = loadavg_.total_tasks
    last_pid = loadavg_.last_pid
    loadavg = 'Average load: %2i%% %2i%% %2i%% %i/%i %i'
    loadavg %= (avg_5, avg_10, avg_15, act_tasks, tot_tasks, last_pid)
    
    users_ = Users().users
    you = os.environ['USER']
    def colour_user(user):
        if user == 'root':     return '\033[31m%s\033[39m'
        elif not user == you:  return '\033[33m%s\033[39m'
        else:                  return '%s'
    users = ['%s{%i}' % (colour_user(u) % u, len(users_[u])) for u in users_.keys()]
    users = 'Users: %s' % (' '.join(users))
    
    uname_ = Uname()
    nodename = uname_.nodename
    kernel_release = uname_.kernel_release
    operating_system = uname_.operating_system
    lts = '-lts' if kernel_release.lower().endswith('-lts') else ''
    have_linux_libre, have_pacman = True, True
    linux_installed, linux_latest = None, None
    try:
        linux_installed = Pacman('linux-libre' + lts, True)
    except:
        have_linux_libre = False
        try:
            linux_installed = Pacman('linux' + lts, True)
        except:
            have_pacman = False
    if have_pacman:
        linux_latest = Pacman('linux-libre' + lts if have_linux_libre else 'linux' + lts, False)
        linux_running = kernel_release.split('-')
        linux_running, kernel_release = linux_running[:2], linux_running[2:]
        linux_running = '-'.join(linux_running)
        kernel_release = '-' + '-'.join(kernel_release)
        linux_installed = linux_installed.version
        linux_latest = linux_latest.version
        if linux_installed == linux_latest:
            if linux_running == linux_installed:
                linux_running = '\033[32m%s\033[39m' % linux_running
        else:
            if linux_running == linux_installed:
                linux_running = '\033[33m%s\033[39m' % linux_running
            else:
                linux_running = '\033[31m%s\033[39m' % linux_running
        kernel_release = linux_running + kernel_release
    uname = '%s %s %s'
    uname %= (nodename, kernel_release, operating_system)
    
    memory_ = Memory()
    def colourise(value, format = '%.0f'):
        colour = '39'
        if value > 30:  colour = '32'
        if value > 50:  colour = '33'
        if value > 80:  colour = '31'
        return '\033[%sm%s\033[0m%%' % (colour, format % value)
    mem = 'Mem: ' + colourise(memory_.mem_used * 100 / memory_.mem_total)
    swp = 'Swp: ' + colourise(memory_.swap_used * 100 / memory_.swap_total)
    shm = 'Shm: ' + colourise(memory_.shmem * 100 / memory_.mem_total)
    
    discs_ = Discs()
    discs = []
    discs_used = 0
    discs_total = 0
    for fs in discs_.filesystems:
        if '/' not in fs:
            continue
        disc = discs_.filesystems[fs]
        mp = disc.mountpoint
        if mp in ['/boot', '/rescue']:
            continue
        mp = mp.split('/')[-1]
        if mp == '':
            mp = '/'
        discs_used += disc.used
        discs_total += disc.blocks
        use = disc.used * 100 / disc.blocks
        colour, colour_ = '39', '39'
        if use < 50:  colour = '32'
        if use > 75:  colour = '33'
        if use > 95:  colour = '31'
        if use > 99:  colour, colour_ = '31', '31'
        discs.append((mp, use, colour, colour_))
    discs.sort(key = lambda d : d[0])
    discs = ['\033[%sm%s:\033[%sm%.1f\033[%sm%%\033[0m' % (d[3], d[0][-4:], d[2], d[1], d[3]) for d in discs]
    use = discs_used * 100 / discs_total
    colour = '39'
    if use < 50:  colour = '32'
    if use > 75:  colour = '33'
    if use > 95:  colour = '31'
    if use > 99:  colour = '41;33'
    discs = 'df: %s : \033[%sm%.2f\033[0m%%' % (' '.join(discs), colour, use)
    
    discstats_ = DiscStats()
    discstats = []
    for disc in discstats_.devices.keys():
        if disc.startswith('sd') or disc.startswith('hd'):
            disc = discstats_.devices[disc]
            discstats.append((disc.device, disc.io_current))
    def colourise(value):
        colour = '39'
        if value >= 1:  colour = '32'
        if value >= 3:  colour = '33'
        if value >= 5:  colour = '31'
        return '\033[%sm%i\033[0m' % (colour, value)
    discstats.sort(key = lambda x : x[0])
    discstats = ['%s:%s' % (d[2:], colourise(i)) for d, i in discstats]
    discstats = 'Disc stats: %s' % (' '.join(discstats))
    
    xdisplay_ = XDisplay()
    xdisplay = ':%i tty%i' % (xdisplay_.display, xdisplay_.vt)
    
    moc_ = MOC().state
    moc = {None : 'dead', MOC.STOPPED : 'stopped', MOC.PAUSED : 'paused', MOC.PLAYING : 'playing'}
    moc = 'Moc: %s' % moc[moc_]
    
    cpu_ = CPU()
    last_cpu_idle, last_cpus_idle = cpu_.cpu[CPU.idle], [cpu[CPU.idle] for cpu in cpu_.cpus]
    last_cpu_total, last_cpus_total = sum(cpu_.cpu), [sum(cpu) for cpu in cpu_.cpus]
    time.sleep(1)
    cpu_ = CPU()
    now_cpu_idle, now_cpus_idle = cpu_.cpu[CPU.idle], [cpu[CPU.idle] for cpu in cpu_.cpus]
    now_cpu_total, now_cpus_total = sum(cpu_.cpu), [sum(cpu) for cpu in cpu_.cpus]
    def cpu_usage(now_idle, now_total, last_idle, last_total):
        total = now_total - last_total
        idle = now_idle - last_idle
        return (total - idle) * 100 / total
    def colourise(value):
        colour = '39'
        if value >= 5:   colour = '32'
        if value >= 50:  colour = '33'
        if value >= 90:  colour = '31'
        return '\033[%sm%2.0f\033[0m%%' % (colour, value)
    cpus = zip(now_cpus_idle, now_cpus_total, last_cpus_idle, last_cpus_total)
    cpus = ' '.join([colourise(cpu_usage(*c)) for c in cpus])
    cpu = colourise(cpu_usage(now_cpu_idle, now_cpu_total, last_cpu_idle, last_cpu_total))
    cpu = 'Cpu: %s : %s' % (cpus, cpu)
    
    gb = lambda x : x / 1024 ** 3
    net_ = Network('lo').devices
    net_ = [(dev, gb(net_[dev]['rx_bytes']), gb(net_[dev]['tx_bytes'])) for dev in net_]
    net_each = ['%s: %.2fGB|%.2fGB' % dev for dev in net_]
    net_total = '%.2fGB|%.2fGB' % (sum([rx for _, rx, tx in net_]), sum([tx for _, rx, tx in net_]))
    net = '%s : %s' % (' '.join(net_each), net_total)
    
    
    text = ['%s │ %s │ %s │ %s │ %s │ %s %s │ %s │ %s │ %s │ %s'
           ,'%s │ %s │ %s'
           ,'%s']
    text = '\n'.join(text)
    text %= (date, uptime, idle, loadavg, users, uname, xdisplay, mem, swp, shm, moc,
             discs, discstats, cpu,
             net)
    bar.clear()
    bar.draw_coloured_text(0, 10, 0, 2, text)