学步园

GNOME可以用的最小的网络指示器

#!/usr/bin/python
# -*- coding: utf-8 -*-
"""
Created on Tue Nov 11 21:26:17 2014

@author: terry
"""

import pygtk
pygtk.require('2.0')
import sys
import os
import shutil
import json
import time
from threading import Thread, Event
import subprocess
import psutil as ps
import copy
import re
import gtk
gtk.gdk.threads_init()
import appindicator
import logging



class StatusUpdateIndicator(Thread):
    B_UNITS = ['', 'KB', 'MB', 'GB', 'TB']
    interval = 2

    def __init__(self):
        Thread.__init__(self)
        self.isRunning = True
        self.last = ps.cpu_times()
        self.last_net_usage = [0, 0]  # (up, down)

        self.network_indicator = appindicator.Indicator ("indicator-sysmonitor", "sysmonitor", appindicator.CATEGORY_APPLICATION_STATUS)
        self.network_indicator.set_status (appindicator.STATUS_ACTIVE)
        self.network_indicator.set_icon("gnome-netstatus-idle")
        self.network_indicator.set_label("Init...")

        # create a menu
        self.menu = gtk.Menu()
        image = gtk.ImageMenuItem(gtk.STOCK_QUIT)
        image.connect("activate", self.quit)
        image.show()
        self.menu.append(image)
        self.menu.show()
        self.network_indicator.set_menu(self.menu)

    def bytes_to_human(self,bytes_):
        unit = 0
        while bytes_ > 1024:
            unit += 1
            bytes_ /= 1024

        return '{}{}'.format(int(bytes_), self.B_UNITS[unit])

    def update(self):
        """It returns the bytes sent and recieved in bytes/second"""
        current = [0, 0]
        for _, iostat in ps.network_io_counters(pernic=True).items():
            current[0] += iostat.bytes_recv
            current[1] += iostat.bytes_sent
        dummy = copy.deepcopy(current)

        current[0] -= self.last_net_usage[0]
        current[1] -= self.last_net_usage[1]
        self.last_net_usage = dummy
        current[0] /= self.interval
        current[1] /= self.interval
        if current[0] == 0 and current[1] == 0:
            self.network_indicator.set_icon("gnome-netstatus-idle")
        elif current[0] > 0 and current[1] == 0:
            self.network_indicator.set_icon("gnome-netstatus-rx")
        elif current[0] == 0 and current[1] > 0:
            self.network_indicator.set_icon("gnome-netstatus-tx")
        else:
            self.network_indicator.set_icon("gnome-netstatus-txrx")
        traffic = '{}/s | {}/s'.format(self.bytes_to_human(current[1]),
                                    self.bytes_to_human(current[0]))
#        print traffic
        self.network_indicator.set_label(traffic)

    def run(self):
        """It is the main loop."""
        while self.isRunning:
            self.update()
            time.sleep(self.interval)

    def quit(self, widget, data=None):
        self.isRunning = False
        gtk.main_quit()


def main():
    gtk.main()
    return 0

if __name__ == "__main__":
    indicator = StatusUpdateIndicator()
    indicator.setDaemon(True)
    indicator.start()
    main()

#network_indicator = appindicator.Indicator("indicator-sysmonitor",
#                                            "sysmonitor",
#                                            appindicator.CATEGORY_SYSTEM_SERVICES)
#network_indicator.set_status(appindicator.STATUS_ACTIVE)
#network_indicator.set_label("Init...")
#network_indicator.set_icon("gnome-netstatus-idle")
#
#updater = StatusUpdater(network_indicator)