Pymonit/agent.py

130 lines
5.3 KiB
Python

#!/usr/bin/python3
import re
import os
import psutil
units_map = [
(1<<50, ' PB'),
(1<<40, ' TB'),
(1<<30, ' GB'),
(1<<20, ' MB'),
(1<<10, ' KB'),
(1, (' byte', ' bytes')),
]
def human_size(bytes, units=units_map):
for factor, suffix in units:
if bytes >= factor:
break
amount = int(bytes / factor)
if isinstance(suffix, tuple):
singular, multiple = suffix
if amount == 1:
suffix = singular
else:
suffix = multiple
return str(amount) + suffix
class ENV:
def __init__(self):
self.hostname = psutil.os.uname().nodename
self.cpu_number = psutil.cpu_count(logical=False)
self.vcpu = psutil.cpu_count()
self.mem_max = human_size(psutil.virtual_memory().total)
class CPU:
def __init__(self):
self.cpu = {}
self.cpu['percent'] = psutil.cpu_percent(interval=0.0, percpu=False)
self.cpu['time_user'] = psutil.cpu_times_percent().user
self.cpu['time_nice'] = psutil.cpu_times_percent().nice
self.cpu['time_system'] = psutil.cpu_times_percent().system
self.cpu['time_idle'] = psutil.cpu_times_percent().idle
self.cpu['time_iowait'] = psutil.cpu_times_percent().iowait
self.cpu['time_irq'] = psutil.cpu_times_percent().irq
self.cpu['time_softirq'] = psutil.cpu_times_percent().softirq
self.cpu['time_steal'] = psutil.cpu_times_percent().steal
self.cpu['time_guest'] = psutil.cpu_times_percent().guest
self.cpu['time_guest_nice'] = psutil.cpu_times_percent().guest_nice
class Memory:
def __init__(self):
self.mem = {}
self.mem['total'] = human_size(psutil.virtual_memory().total)
self.mem['available'] = human_size(psutil.virtual_memory().available)
self.mem['percent'] = psutil.virtual_memory().percent
self.mem['used'] = human_size(psutil.virtual_memory().used)
self.mem['free'] = human_size(psutil.virtual_memory().free)
self.mem['active'] = human_size(psutil.virtual_memory().active)
self.mem['inactive'] = human_size(psutil.virtual_memory().inactive)
self.mem['buffers'] = human_size(psutil.virtual_memory().buffers)
self.mem['cached'] = human_size(psutil.virtual_memory().cached)
self.mem['shared'] = human_size(psutil.virtual_memory().shared)
self.mem['slab'] = human_size(psutil.virtual_memory().slab)
class Swap:
def __init__(self):
self.swap = {}
self.swap['total'] = human_size(psutil.swap_memory().total)
self.swap['used'] = human_size(psutil.swap_memory().used)
self.swap['free'] = human_size(psutil.swap_memory().free)
self.swap['percent'] = psutil.swap_memory().percent
self.swap['sin'] = human_size(psutil.swap_memory().sin)
self.swap['sout'] = human_size(psutil.swap_memory().sout)
class Disks:
def __init__(self):
num=0
for i in psutil.disk_partitions():
exec("self.disk_"+str(num)+" = {}")
exec("self.disk_"+str(num)+"['device'] = '"+str(i.device)+"'")
exec("self.disk_"+str(num)+"['mountpoint'] = '"+str(i.mountpoint)+"'")
exec("self.disk_"+str(num)+"['fstype'] = '"+str(i.fstype)+"'")
exec("self.disk_"+str(num)+"['opts'] = '"+str(i.opts)+"'")
exec("self.disk_"+str(num)+"['maxfile'] = '"+str(i.maxfile)+"'")
exec("self.disk_"+str(num)+"['maxpath'] = '"+str(i.maxpath)+"'")
exec("self.disk_"+str(num)+"['size_total'] = '"+str(human_size(psutil.disk_usage(i.mountpoint).total))+"'")
exec("self.disk_"+str(num)+"['size_used'] = '"+str(human_size(psutil.disk_usage(i.mountpoint).used))+"'")
exec("self.disk_"+str(num)+"['size_free'] = '"+str(human_size(psutil.disk_usage(i.mountpoint).free))+"'")
exec("self.disk_"+str(num)+"['size_percent'] = '"+str(psutil.disk_usage(i.mountpoint).percent)+"'")
num+=1
class Network:
def __init__(self):
num=0
for interface in psutil.net_if_addrs():
exec("self.net_"+str(num)+" = {}")
for detail in psutil.net_if_addrs()[interface]:
exec("self.net_"+str(num)+"['name'] = '"+str(interface)+"'")
ip_pattern = re.compile(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$")
if ip_pattern.match(detail.address):
exec("self.net_"+str(num)+"['address_v4'] = '"+str(detail.address)+"'")
else:
exec("self.net_"+str(num)+"['address_v6'] = '"+str(detail.address)+"'")
if detail.netmask:
if ip_pattern.match(detail.netmask):
exec("self.net_"+str(num)+"['netmask_v4'] = '"+str(detail.netmask)+"'")
else:
exec("self.net_"+str(num)+"['netmask_v6'] = '"+str(detail.netmask)+"'")
num+=1
class Monit:
def __init__(self):
self.env = ENV()
self.cpu = CPU()
self.mem = Memory()
self.swap = Swap()
self.disks = Disks()
self.network = Network()
Localhost = Monit()
full = {}
full.update(vars(Localhost.env))
full.update(vars(Localhost.cpu))
full.update(vars(Localhost.mem))
full.update(vars(Localhost.swap))
full.update(vars(Localhost.disks))
full.update(vars(Localhost.network))
print (full)