GOLDEN HOUR

DIR: /usr/local/lib/python3.9/site-packages/agent360/plugins

Name	Size	Action
__init__.py	0 B
[ __pycache__ ]	-
apt-updates.py	1.23 KB
asterisk.py	1012 B
bind.py	732 B
bird.py	841 B
bitninja.py	1.27 KB
cloudlinux-dbgov.py	1.04 KB
cloudlinux.py	1020 B
cpanel.py	1.41 KB
cpu.py	1.97 KB
cpu_freq.py	583 B
dirsize.py	617 B
diskinodes.py	652 B
diskstatus-nvme.py	1.58 KB
diskstatus.py	1.96 KB
diskusage.py	4.04 KB
docker.py	3.36 KB
dovecot.py	1.27 KB
elasticsearch.py	4.24 KB
exim.py	484 B
fail2ban.py	831 B
gpu.py	959 B
haproxy.py	4.17 KB
httpd.py	2.73 KB
iostat.py	5.1 KB
janus.py	705 B
kamailio.py	556 B
litespeed.py	3.04 KB
loadavg.py	332 B
loggedin.py	400 B
mailq.py	528 B
mdstat.py	1.36 KB
megacli.py	2.72 KB
memcached.py	3.09 KB
memory.py	932 B
minecraft.py	2.35 KB
mongodb.py	5.63 KB
mysql.py	5.13 KB
network.py	2.72 KB
nginx.py	3.2 KB
openvpn.py	2.17 KB
phpfpm.py	2.72 KB
ping.py	3.09 KB
plesk-cgroups.py	6.18 KB
plugins.py	2.48 KB
postfix.py	1.73 KB
powerdns.py	4.56 KB
process.py	4.26 KB
proftpd.py	1.1 KB
rabbitmq.py	3.49 KB
redis_stat.py	5.48 KB
sleeper.py	244 B
swap.py	364 B
system.py	5.67 KB
tcpports.py	1.08 KB
temp.py	1.45 KB
unbound.py	3.24 KB
vms.py	6.2 KB
wp-toolkit.py	1.8 KB
yum-updates.py	810 B

Edit: diskusage.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import psutil
import plugins
import json

class Plugin(plugins.BasePlugin):
    __name__ = 'diskusage'

def run(self, config):
        disk = {}
        disk['df-psutil'] = []

for part in psutil.disk_partitions(False):
            valid_part = True
            ignored_partitions = ['/loop', '/snap', 'squashfs', 'cagefs-skeleton']
            
            for ignore in ignored_partitions:
                if ignore in part.device or ignore in part.mountpoint or ignore in part.fstype:
                   valid_part = False
            if valid_part == False:
                continue
            
            if os.name == 'nt':
                if 'cdrom' in part.opts or part.fstype == '':
                    # skip cd-rom drives with no disk in it; they may raise
                    # ENOENT, pop-up a Windows GUI error for a non-ready
                    # partition or just hang.
                    continue
            try:
                usage = psutil.disk_usage(part.mountpoint)
                diskdata = {}
                diskdata['info'] = part
                for key in usage._fields:
                    diskdata[key] = getattr(usage, key)
                disk['df-psutil'].append(diskdata)
            except:
                pass

try:
            force_df = config.get('diskusage', 'force_df')
        except:
            force_df = 'no'

if len(disk['df-psutil']) == 0 or force_df == 'yes':
            try:
                disk['df-psutil'] = []
                df_output_lines = [s.split() for s in os.popen("df -Pl").read().splitlines()]
                del df_output_lines[0]
                for row in df_output_lines:
                    if row[0] == 'tmpfs':
                        continue
                    disk['df-psutil'].append({'info': [row[0], row[5],'',''], 'total': int(row[1])*1024, 'used': int(row[2])*1024, 'free': int(row[3])*1024, 'percent': row[4][:-1]})
            except:
                pass

try:
            zfs_stats = config.get('diskusage', 'zfs')
        except:
            zfs_stats = 'no'

if zfs_stats == 'yes':
            try:
                lines = [s.split(', ') for s in os.popen("zfs list -Hp -t volume").read().splitlines()]
                for row in lines:
                    v = {}
                    v['vg_name'] = row[0]
                    v['vg_size'] = int(row[5][:-1])
                    v['vg_free'] = int(row[6][:-1])
                    v['vg_used'] = int(v['vg_size']-v['vg_free'])
                    v['vg_percentage'] = (v['vg_used']/float(v['vg_size']))*100
                    disk['df-psutil'].append({'info': [v['vg_name'], v['vg_name'], 'zfs', False], 'total': v['vg_size'], 'used': v['vg_used'], 'free': v['vg_free'], 'percent': v['vg_percentage']})
            except Exception as e:
                return e.message

try:
            lvm_stats = config.get('diskusage', 'lvm')
        except:
            lvm_stats = 'no'

# For LVM volume group monitoring, requires sudo access to vgs
        # add vgs to /etc/sudoers
        # agent360 ALL=(ALL) NOPASSWD: /usr/sbin/vgs
        # set lvm = yes right under enabled = yes in /etc/agent360.ini
        if lvm_stats == 'yes':
            try:
                lines = [s.split(', ') for s in os.popen("sudo vgs --all --units b --noheadings --separator ', '").read().splitlines()]
                for row in lines:
                    v = {}
                    v['vg_name'] = row[0]
                    v['vg_size'] = int(row[5][:-1])
                    v['vg_free'] = int(row[6][:-1])
                    v['vg_used'] = int(v['vg_size']-v['vg_free'])
                    v['vg_percentage'] = (v['vg_used']/float(v['vg_size']))*100
                    disk['df-psutil'].append({'info': [v['vg_name'], v['vg_name'], 'lvm', False], 'total': v['vg_size'], 'used': v['vg_used'], 'free': v['vg_free'], 'percent': v['vg_percentage']})
            except Exception as e:
                return e.message

return disk

if __name__ == '__main__':
    Plugin().execute()

GOLDEN HOUR

Edit: diskusage.py

TERMINAL