Indicator for Network Status

最新推荐文章于 2025-05-04 21:14:43 发布

原创最新推荐文章于 2025-05-04 21:14:43 发布 · 737 阅读

0 ·

CC 4.0 BY-SA版权

网络专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个用于GNOME桌面环境的轻量级网络指示器应用。该应用通过Python实现，能够实时显示网络上传和下载速度，并根据不同状态更换图标。文章提供了完整的源代码，便于读者理解和使用。

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)