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
|
# -*- coding: utf-8 -*-
# Net_monitor plasma data source
#
# Copyright, (C) Eugeni Dodonov <eugeni@mandriva.com>, 2011
#
from PyQt4.QtCore import Qt, QVariant
from PyKDE4.plasma import Plasma
from PyKDE4 import plasmascript
# localization
import gettext
try:
gettext.install("net_monitor")
except IOError:
_ = str
from net_monitor import Monitor
class NetMonitorDataEngine(plasmascript.DataEngine):
def __init__(self,parent,args=None):
plasmascript.DataEngine.__init__(self,parent)
def init(self):
"""Initialize Monitor class"""
self.setMinimumPollingInterval(333)
self.monitor = Monitor()
self.ifaces = self.monitor.readnet()
self.enabled_ifaces = []
self.wireless_ifaces = filter(self.monitor.has_wireless, self.ifaces.keys())
sorted_ifaces = self.ifaces.keys()
sorted_ifaces.sort()
net=self.monitor.readnet()
for iface in sorted_ifaces:
device_exists, data_in, data_out = self.monitor.get_traffic(iface,net)
self.ifaces[iface] = {'data_in': 0,
'data_out': 0,
'total_in': 0,
'total_out': 0,
'graph': None,
'histogram': [],
'address': "",
}
if self.monitor.has_network_accounting(iface):
self.enabled_ifaces.append(iface)
self.refresh_connections()
def refresh_connections(self):
"""Updates connections"""
for proto in ["tcp", "udp"]:
connections = self.monitor.get_connections(proto=proto)
for loc_addr, loc_port, rem_addr, rem_port, status in connections:
#print "%s - %s - %s - %s - %s - %s" % (proto, loc_addr, loc_port, rem_addr, rem_port, status)
pass
def sources(self):
"""List of provided sources"""
sources = self.ifaces.keys()
print sources
return sources
def sourceRequestEvent(self, name):
"""Prepare source event"""
self.refresh_connections()
return self.updateSourceEvent(name)
def updateSourceEvent(self, name):
"""Returns monitoring data"""
print "Getting info for %s " % name
net=self.monitor.readnet()
interval = 1 # update interval
wifi_stats = self.monitor.wireless_stats()
for iface in [name]:
iface = str(iface)
status = self.monitor.get_status(iface)
old_data_in = self.ifaces[iface]['data_in']
old_data_out = self.ifaces[iface]['data_out']
total_in = self.ifaces[iface]['total_in']
total_out = self.ifaces[iface]['total_out']
# get the uptime
uptime = self.monitor.get_uptime(iface)
# update widgets
ip, mac = self.monitor.get_address(iface)
device_exists, data_in, data_out = self.monitor.get_traffic(iface, net)
# is it a wireless interface?
if iface in self.wireless_ifaces:
essid = self.monitor.wifi_get_essid(iface)
mode = self.monitor.wifi_get_mode(iface)
bitrate = self.monitor.wifi_get_bitrate(iface)
ap = self.monitor.wifi_get_ap(iface)
link = wifi_stats.get(iface, 0)
# calculate link quality
if "max_quality" in self.ifaces[iface]:
max_quality = self.ifaces[iface]["max_quality"]
if max_quality != 0:
quality = link * 100.0 / max_quality
else:
quality = 0
else:
quality = 0
else:
essid = None
mode = None
bitrate = None
ap = None
quality = 0
# is it the first measure?
if old_data_in == 0 and old_data_out == 0:
old_data_in = data_in
old_data_out = data_out
# check if device exists
if not device_exists:
old_data_in = data_in
old_data_out = data_out
# check total download
diff_in = data_in - old_data_in
diff_out = data_out - old_data_out
# checking for 32bits overflow
if diff_in < 0:
diff_in += 2**32
if diff_out < 0:
diff_out += 2**32
total_in += diff_in
total_out += diff_out
# speed
speed_in = diff_in / interval
speed_out = diff_out / interval
# update saved values
self.ifaces[iface]['data_in'] = data_in
self.ifaces[iface]['data_out'] = data_out
self.ifaces[iface]['total_in'] = total_in
self.ifaces[iface]['total_out'] = total_out
# now set the applet data
self.setData(iface, "data_in", QVariant(data_in))
self.setData(iface, "data_out", QVariant(data_in))
self.setData(iface, "total_in", QVariant(data_in))
self.setData(iface, "total_out", QVariant(data_in))
for item, value in [('ip_address', ip),
('status', status),
('hw_address', mac),
('essid', essid),
('mode', mode),
('bitrate', bitrate),
('ap', ap),
('quality', "%d%%" % quality),
('widget_uptime', uptime),
]:
self.setData(iface, item, QVariant(value))
return True
def CreateDataEngine(parent):
return NetMonitorDataEngine(parent)
|
