From 43f5ec022e2bddf0e6618426658559a37603a2f9 Mon Sep 17 00:00:00 2001 From: ibrokemypie Date: Sat, 3 Jun 2017 23:40:50 +1000 Subject: [PATCH] [modules/traffic] Add option to only show currently up interfaces --- bumblebee/modules/traffic.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/bumblebee/modules/traffic.py b/bumblebee/modules/traffic.py index 0da9fd1..afe911e 100644 --- a/bumblebee/modules/traffic.py +++ b/bumblebee/modules/traffic.py @@ -4,6 +4,7 @@ Parameters: * traffic.exclude: Comma-separated list of interface prefixes to exclude (defaults to "lo,virbr,docker,vboxnet,veth") + * traffic.onlyup: Whether to oly show traffic for Up interfaces (defaults to False) """ import re @@ -47,7 +48,12 @@ class Module(bumblebee.engine.Module): return widget def _update_widgets(self, widgets): - interfaces = [ i for i in netifaces.interfaces() if not i.startswith(self._exclude) ] + allinterfaces = [ i for i in netifaces.interfaces() if not i.startswith(self._exclude) ] + + if self.parameter("onlyup", False): + interfaces = [ i for i in allinterfaces if netifaces.AF_INET in netifaces.ifaddresses(i) ] + else: + interfaces = allinterfaces counters = psutil.net_io_counters(pernic=True) for interface in interfaces: