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
|
# -*- coding: utf8 -*-
from __future__ import division # python2 compatibility
from time import time, sleep
"""
Module for displaying current network transfer rate.
@author shadowprince
@license Eclipse Public License
"""
CACHED_TIME = 1 # update time (in seconds)
POSITION = 0 # bar position
DEVFILE = "/proc/net/dev" # location of dev file under /proc
INTERFACES = [] # list of interfaces to track
ALL_INTERFACES = True # ignore INTERFACES, but not INTERFACES_BLACKLIST
INTERFACES_BLACKLIST = ["lo"] # list of interfaces to ignore
"""
Format of status string.
Placeholders:
interface - name of interface
total - total rate
up - upload rate
down - download rate
"""
FORMAT = "{interface}: {total}"
PRECISION = 1 # amount of numbers after dot
MULTIPLIER_TOP = 1024 # if value is greater, divide it with UNIT_MULTI and get next unit from UNITS
LEFT_ALIGN = len(str(MULTIPLIER_TOP)) + 1 + PRECISION # == 6 characters (from MULTIPLIER_TOP + dot + PRECISION)
"""
Format of total, up and down placeholders under FORMAT.
As default, substitutes LEFT_ALIGN and PRECISION as %s and %s
Placeholders:
value - value (float)
unit - unit (string)
"""
VALUE_FORMAT = "{value:%s.%sf} {unit}" % (LEFT_ALIGN, PRECISION)
INITIAL_MULTI = 1024 # initial multiplier, if you want to get rid of first bytes, set to 1 to disable
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...
NO_CONNECTION = "! no data" # when there is no data transmitted from the start of the plugin
HIDE_IF_NO = False # hide indicator if rate == 0
"""
Get statistics from devfile in list of lists of words
"""
def get_stat():
def dev_filter(x):
# get first word and remove trailing interface number
x = x.strip().split(" ")[0][:-1]
if x in INTERFACES_BLACKLIST:
return False
if ALL_INTERFACES:
return True
if x in INTERFACES:
return True
return False
# read devfile, skip two header files
x = filter(dev_filter, open(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
"""
Divide a value and return formatted string
"""
def divide_and_format(value):
for i, unit in enumerate(UNITS):
if value > MULTIPLIER_TOP:
value /= UNIT_MULTI
else:
break
return VALUE_FORMAT.format(value=value, unit=unit)
class Py3status:
def __init__(self, *args, **kwargs):
self.last_stat = get_stat()
self.last_time = time()
self.last_interface = None
def currentSpeed(self, json, i3status_config):
ns = 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 = 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 = HIDE_IF_NO
# 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 = HIDE_IF_NO
return (POSITION, {
'transformed': True,
'full_text': "" if hide else
FORMAT.format(
total=divide_and_format(delta['total']),
up=divide_and_format(delta['up']),
down=divide_and_format(delta['down']),
interface=interface[:-1],
) if interface else NO_CONNECTION,
'name': 'speed',
'cached_until': time() + CACHED_TIME,
})
if __name__ == "__main__":
x = Py3status()
while True:
print(x.currentSpeed(1, 1))
sleep(1)
|
