From cba44c60fb7cdcf704b2b0b9f5a594d4d594543e Mon Sep 17 00:00:00 2001 From: Eugeni Dodonov Date: Sun, 16 Jan 2011 20:29:57 -0200 Subject: added datasource --- plasma/datasource/contents/code/main.py | 159 ++++++++++++++++++++++++++++++++ 1 file changed, 159 insertions(+) create mode 100644 plasma/datasource/contents/code/main.py (limited to 'plasma/datasource/contents') diff --git a/plasma/datasource/contents/code/main.py b/plasma/datasource/contents/code/main.py new file mode 100644 index 0000000..8d9a8a2 --- /dev/null +++ b/plasma/datasource/contents/code/main.py @@ -0,0 +1,159 @@ +# -*- coding: utf-8 -*- +# Net_monitor plasma data source +# +# Copyright, (C) Eugeni Dodonov , 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) -- cgit v1.2.1