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/Makefile | 11 +++ plasma/datasource/contents/code/main.py | 159 ++++++++++++++++++++++++++++++++ plasma/datasource/metadata.desktop | 17 ++++ 3 files changed, 187 insertions(+) create mode 100644 plasma/datasource/Makefile create mode 100644 plasma/datasource/contents/code/main.py create mode 100644 plasma/datasource/metadata.desktop diff --git a/plasma/datasource/Makefile b/plasma/datasource/Makefile new file mode 100644 index 0000000..076755c --- /dev/null +++ b/plasma/datasource/Makefile @@ -0,0 +1,11 @@ +all: plasmapkg + +plasmapkg: + zip -r ../../net_monitor_data.zip contents/ metadata.desktop + +clean: + rm -f ../../net_monitor_plasma_data.zip + +test: clean plasmapkg + plasmapkg -t dataengine -r net_monitor_data + plasmapkg -t dataengine -i ../../net_monitor_data.zip 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) diff --git a/plasma/datasource/metadata.desktop b/plasma/datasource/metadata.desktop new file mode 100644 index 0000000..7746f58 --- /dev/null +++ b/plasma/datasource/metadata.desktop @@ -0,0 +1,17 @@ +[Desktop Entry] +Encoding=UTF-8 +Name=Network Monitor data source +Type=Service +ServiceTypes=Plasma/DataEngine +Icon=chronometer +X-Plasma-API=python +X-Plasma-MainScript=code/main.py +X-KDE-PluginInfo-Author=Eugeni Dodonov +X-KDE-PluginInfo-Email=eugeni@mandriva.com +X-KDE-PluginInfo-Name=net_monitor_data +X-KDE-PluginInfo-Version=1.0 +X-KDE-PluginInfo-Website=http://git.mandriva.com/ +X-KDE-PluginInfo-Category=Network +X-KDE-PluginInfo-Depends= +X-KDE-PluginInfo-License=GPL +X-KDE-PluginInfo-EnabledByDefault=true -- cgit v1.2.1