????
Current Path : /usr/local/lib/python3.9/site-packages/agent360/plugins/ |
Current File : //usr/local/lib/python3.9/site-packages/agent360/plugins/cpanel.py |
#!/usr/bin/env python # -*- coding: utf-8 -*- import plugins import subprocess import json import re class Plugin(plugins.BasePlugin): __name__ = 'cpanel' def to_bytes(self, size): size = size.upper() units = {"B": 1, "K": 2 ** 10, "M": 2 ** 20, "G": 2 ** 30, "T": 2 ** 40} if not re.match(r' ', size): size = re.sub(r'([KMGT])', r' \1', size) number, unit = [string.strip() for string in size.split()] return int(float(number) * units[unit]) def run(self, config): ''' Plugin to collect cpanel user accounts To enable add to /etc/agent360.ini: [cpanel] enabled = yes ''' data = subprocess.check_output(['whmapi1', '--output=jsonpretty', 'listaccts']) results = {} accounts = json.loads(data) for account in accounts['data']['acct']: results[account['user']] = { 'diskused_bytes': self.to_bytes(account['diskused']), 'inodesused': account['inodesused'], 'is_locked': account['is_locked'], 'has_backup': account['has_backup'], 'outgoing_mail_hold': account['outgoing_mail_hold'], 'outgoing_mail_suspended': account['outgoing_mail_suspended'], 'suspended': account['suspended'], } return results if __name__ == '__main__': Plugin().execute()