From f537b44b9b0ac0f540188b187338977a1556d872 Mon Sep 17 00:00:00 2001 From: Sandro Date: Thu, 30 Jun 2022 23:20:12 +0200 Subject: [PATCH] Code style (Black) and linting (PyFlakes) changes --- plugins/network/nft_counters | 85 +++++++++++++++++++++++------------- 1 file changed, 54 insertions(+), 31 deletions(-) diff --git a/plugins/network/nft_counters b/plugins/network/nft_counters index 66f46d21..0abf717b 100755 --- a/plugins/network/nft_counters +++ b/plugins/network/nft_counters @@ -97,6 +97,7 @@ examples [4] provided by Arturo Borrero Gonzalez. """ import sys + try: from nftables import Nftables from nftables import json @@ -137,6 +138,7 @@ def nft_cmd(nftlib, cmd): return ruleset + def getCounters(): nft = Nftables() @@ -161,7 +163,6 @@ class MuninNftCountersPlugin(MuninPlugin): plugin_name = "nft_counters" isMultigraph = True - def __init__(self, argv=(), env=None, debug=False): """ @@ -194,34 +195,46 @@ class MuninNftCountersPlugin(MuninPlugin): if self._argv[1] == "autoconf": return else: - print("# No counters in nftables. Try adding some first.", - "# See 'munin-doc %s' for more information." % self.plugin_name, - sep="\n") + print( + "# No counters in nftables. Try adding some first.", + "# See 'munin-doc %s' for more information." % self.plugin_name, + sep="\n", + ) raise - except Exception as err: + except Exception: if self._argv[1] == "autoconf": return else: - print("# Plugin needs to be run as root since nftables can only be", - "# run as root.", - "#", - "# Use the following setting in the configuration file", - "# to enable root privileges:", - "#", - "# [%s]" % self.plugin_name, - "# user root", - sep="\n") + print( + "# Plugin needs to be run as root since nftables can only be", + "# run as root.", + "#", + "# Use the following setting in the configuration file", + "# to enable root privileges:", + "#", + "# [%s]" % self.plugin_name, + "# user root", + sep="\n", + ) raise count_only = self.envGet("count_only") # Create the graphs if not (count_only == "bytes"): - graph_packets = MuninGraph("nftables counters (packets)", graph_category, - vlabel="packets / second", args="--base 1000") + graph_packets = MuninGraph( + "nftables counters (packets)", + graph_category, + vlabel="packets / second", + args="--base 1000", + ) if not (count_only == "packets"): - graph_bytes = MuninGraph("nftables counters (bytes)", graph_category, - vlabel="bytes / second", args="--base 1024") + graph_bytes = MuninGraph( + "nftables counters (bytes)", + graph_category, + vlabel="bytes / second", + args="--base 1024", + ) # Define filter to allow for tuning of counters graphed self.envRegisterFilter("counters") @@ -232,23 +245,32 @@ class MuninNftCountersPlugin(MuninPlugin): # Until it does, use counter name as info try: field_info = counter["comment"] - except: + except Exception: field_info = counter["name"] if self.envCheckFilter("counters", counter["name"]): if not (count_only == "bytes"): - graph_packets.addField(counter["name"], counter["name"], min=0, - type="DERIVE", info=field_info) + graph_packets.addField( + counter["name"], + counter["name"], + min=0, + type="DERIVE", + info=field_info, + ) if not (count_only == "packets"): - graph_bytes.addField(counter["name"], counter["name"], min=0, - type="DERIVE", info=field_info) + graph_bytes.addField( + counter["name"], + counter["name"], + min=0, + type="DERIVE", + info=field_info, + ) if not (count_only == "bytes"): self.appendGraph("nft_counters_packets", graph_packets) if not (count_only == "packets"): self.appendGraph("nft_counters_bytes", graph_bytes) - def retrieveVals(self): """ @@ -264,12 +286,13 @@ class MuninNftCountersPlugin(MuninPlugin): for counter in self.counters: if self.envCheckFilter("counters", counter["name"]): if self.hasGraph("nft_counters_packets"): - self.setGraphVal("nft_counters_packets", counter["name"], - counter["packets"]) + self.setGraphVal( + "nft_counters_packets", counter["name"], counter["packets"] + ) if self.hasGraph("nft_counters_bytes"): - self.setGraphVal("nft_counters_bytes", counter["name"], - counter["bytes"]) - + self.setGraphVal( + "nft_counters_bytes", counter["name"], counter["bytes"] + ) def autoconf(self): """ @@ -283,9 +306,9 @@ class MuninNftCountersPlugin(MuninPlugin): """ try: - counters = getCounters() + getCounters() return True - except: + except Exception: return False