cyberes
/
icinga2-checks
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

error handling

This commit is contained in:
Cyberes 2023-04-21 23:54:18 -06:00
parent b32f46f4fd
commit 8232fc2472
1 changed files with 64 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
check_bandwidth.py
View File

@ -1,8 +1,11 @@
#!/usr/bin/env python3
import argparse
import sys
import traceback
import psutil
import checker.nagios as nagios
from checker.markdown import list_to_markdown_table
# Parse the bandwidth limit argument
@ -12,15 +15,17 @@ parser.add_argument('--critical', type=int, default=75, help='Critical if percen
parser.add_argument('--warn', type=int, default=50, help='Warning if percent of bandwidth usage is greater than or equal to this.')
args = parser.parse_args()
data = []
warn_value = (args.bandwidth * args.warn / 100)
crit_value = (args.bandwidth * args.critical / 100)
# Get network interface statistics
net_io_counters = psutil.net_io_counters(pernic=True)
def main():
data = []
warn_value = (args.bandwidth * args.warn / 100)
crit_value = (args.bandwidth * args.critical / 100)
# Calculate bandwidth utilization for each interface
for interface, stats in net_io_counters.items():
# Get network interface statistics
net_io_counters = psutil.net_io_counters(pernic=True)
# Calculate bandwidth utilization for each interface
for interface, stats in net_io_counters.items():
# Get the number of bytes sent and received
bytes_sent = stats.bytes_sent
bytes_recv = stats.bytes_recv
@ -37,33 +42,46 @@ for interface, stats in net_io_counters.items():
bandwidth_utilization = (8 * (new_bytes_sent - bytes_sent + new_bytes_recv - bytes_recv)) / (1 * 1000 * 1000)
data.append([interface, bandwidth_utilization, 'none'])
critical = []
warn = []
ok = []
perf_data = []
for i in range(len(data)):
exit_code = nagios.OK
critical = []
warn = []
ok = []
perf_data = []
for i in range(len(data)):
interface = data[i][0]
bandwidth_utilization = data[i][1]
if bandwidth_utilization >= crit_value:
critical.append(interface)
data[i][2] = 'critical'
exit_code = nagios.CRITICAL
elif bandwidth_utilization >= warn_value:
warn.append(interface)
data[i][2] = 'warning'
exit_code = nagios.WARNING
else:
ok.append(interface)
data[i][2] = 'ok'
perf_data.append(f'{interface}={round(bandwidth_utilization, 2)}MB;{warn_value};{crit_value}; ')
if len(ok):
if len(ok):
print(f'OK: {", ".join(ok)}')
if len(warn):
if len(warn):
print(f'WARNING: {", ".join(warn)}')
if len(critical):
if len(critical):
print(f'CRITICAL: {", ".join(critical)}')
data = [(x[0], f'{round(x[1], 2)} Mbps', x[2]) for x in data]
data.insert(0, ('Interface', 'Bandwidth', 'State'))
print(list_to_markdown_table(data, align='left'))
data = [(x[0], f'{round(x[1], 2)} Mbps', x[2]) for x in data]
data.insert(0, ('Interface', 'Bandwidth', 'State'))
print(list_to_markdown_table(data, align='left'))
print(f' |{"".join(perf_data)}')
print(f' |{"".join(perf_data)}')
sys.exit(exit_code)
if __name__ == "__main__":
try:
main()
except Exception as e:
print(f'UNKNOWN: exception "{e}"')
print(traceback.format_exc())
sys.exit(nagios.UNKNOWN)