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
|
# -*- coding: utf8 -*-
"""
Module for displaying current network transfer rate.
@author shadowprince
@license Eclipse Public License
"""
from __future__ import division # python2 compatibility
from time import time
INITIAL_MULTI = 1024 # initial multiplier, if you want to get rid of first bytes, set to 1 to disable
MULTIPLIER_TOP = 999 # if value is greater, divide it with UNIT_MULTI and get next unit from UNITS
UNIT_MULTI = 1024 # value to divide if rate is greater than MULTIPLIER_TOP
UNITS = ["kb/s", "mb/s", "gb/s", "tb/s", ] # list of units, first one - value/INITIAL_MULTI, second - value/1024, third - value/1024^2, etc...
class Py3status:
"""
Confiuration parameters:
- all_interfaces : ignore self.interfaces, but not self.interfaces_blacklist
- devfile : location of dev file under /proc
- format_no_connection : when there is no data transmitted from the start of the plugin
- hide_if_zero : hide indicator if rate == 0
- interfaces : comma separated list of interfaces to track
- interfaces_blacklist : comma separated list of interfaces to ignore
- precision : amount of numbers after dot
Format of status string placeholders:
interface - name of interface
total - total rate
up - upload rate
down - download rate
"""
# available configuration parameters
all_interfaces = True
cache_timeout = 2
devfile = '/proc/net/dev'
format = "{interface}: {total}"
format_no_connection = ''
hide_if_zero = False
interfaces = ''
interfaces_blacklist = 'lo'
precision = 1
def __init__(self, *args, **kwargs):
"""
Format of total, up and down placeholders under self.format.
As default, substitutes self.left_align and self.precision as %s and %s
Placeholders:
value - value (float)
unit - unit (string)
"""
self.last_interface = None
self.last_stat = self._get_stat()
self.last_time = time()
# == 6 characters (from MULTIPLIER_TOP + dot + self.precision)
self.left_align = len(str(MULTIPLIER_TOP)) + 1 + self.precision
self.value_format = "{value:%s.%sf} {unit}" % (self.left_align, self.precision)
def currentSpeed(self, i3s_output_list, i3s_config):
# parse some configuration parameters
if not isinstance(self.interfaces, list):
self.interfaces = self.interfaces.split(',')
if not isinstance(self.interfaces_blacklist, list):
self.interfaces_blacklist = self.interfaces_blacklist.split(',')
ns = self._get_stat()
deltas = {}
try:
# time from previous check
timedelta = time() - self.last_time
# calculate deltas for all interfaces
for old, new in zip(self.last_stat, ns):
down = int(new[1]) - int(old[1])
up = int(new[9]) - int(old[9])
down /= timedelta * INITIAL_MULTI
up /= timedelta * INITIAL_MULTI
deltas[new[0]] = {'total': up+down, 'up': up, 'down': down, }
# update last_ info
self.last_stat = self._get_stat()
self.last_time = time()
# get the interface with max rate
interface = max(deltas, key=lambda x: deltas[x]['total'])
# if there is no rate - show last active interface, or hide
if deltas[interface]['total'] == 0:
interface = self.last_interface
hide = self.hide_if_zero
# if there is - update last_interface
else:
self.last_interface = interface
hide = False
# get the deltas into variable
delta = deltas[interface] if interface else None
except TypeError:
delta = None
interface = None
hide = self.hide_if_zero
return {
'cached_until': time() + self.cache_timeout,
'full_text': "" if hide else
self.format.format(
total=self._divide_and_format(delta['total']),
up=self._divide_and_format(delta['up']),
down=self._divide_and_format(delta['down']),
interface=interface[:-1],
) if interface else self.format_no_connection
}
def _get_stat(self):
"""
Get statistics from devfile in list of lists of words
"""
def dev_filter(x):
# get first word and remove trailing interface number
x = x.strip().split(" ")[0][:-1]
if x in self.interfaces_blacklist:
return False
if self.all_interfaces:
return True
if x in self.interfaces:
return True
return False
# read devfile, skip two header files
x = filter(dev_filter, open(self.devfile).readlines()[2:])
try:
# split info into words, filter empty ones
return [list(filter(lambda x: x, _x.split(" "))) for _x in x]
except StopIteration:
return None
def _divide_and_format(self, value):
"""
Divide a value and return formatted string
"""
for i, unit in enumerate(UNITS):
if value > MULTIPLIER_TOP:
value /= UNIT_MULTI
else:
break
return self.value_format.format(value=value, unit=unit)
if __name__ == "__main__":
"""
Test this module by calling it directly.
"""
from time import sleep
x = Py3status()
while True:
print(x.currentSpeed([], {}))
sleep(1)
|
