Viewing file:  collect_information.py (5.15 KB) -rwxr-xr-x
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
#!/opt/alt/python37/bin/python3
# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENCE.TXT
#
import json
import argparse
import os
import logging
from clwpos.logsetup import setup_logging
from clwpos.feature_suites import (
get_allowed_modules,
get_admin_config_directory,
get_admin_config_permissions,
)
from clwpos.utils import acquire_lock, user_uid, get_pw, is_wpos_supported
from clwpos.daemon import whmapi1, WposDaemon
_logger = setup_logging(
caller_name='collect_information',
file_level=logging.INFO,
logfile_path='/var/log/clwpos/collect_information.log',
)
def php_get_vhost_versions():
"""
@return: [
{
"account": "rm3",
"account_owner": "root",
"documentroot": "/home/example/public_html",
"homedir": "/home/rm3",
"is_suspended": 0,
"main_domain": 1,
"php_fpm": 1,
"php_fpm_pool_parms": {
"pm_max_children": 5,
"pm_max_requests": 20,
"pm_process_idle_timeout": 10
},
"phpversion_source": [
{
"domain": "example.com",
"system_default": 1
}
],
"version": "ea-php72",
"vhost": "otherchars.rm3.tld"
}
]
"""
return whmapi1(WposDaemon.DAEMON_PHP_GET_VHOST_VERSIONS_COMMAND)["versions"]
def php_get_installed_versions():
"""
@return: [
"ea-php74"
]
"""
return whmapi1(WposDaemon.DAEMON_PHP_GET_INSTALLED_VERSIONS_COMMAND)["versions"]
def listaccts():
"""
@return: [
{
"backup": 0,
"child_nodes": [
{
"alias": "nodealias",
"workload": "Mail"
}
],
"disklimit": "unlimited",
"diskused": "14M",
"domain": "example.com",
"email": "username@example.com",
"has_backup": 1,
"inodeslimit": "unlimited",
"inodesused": 1,
"ip": "192.168.0.128",
"ipv6": [
"0101:ca75:0101:ca75:0101:ca75:0101:ca77"
],
"is_locked": 0,
"legacy_backup": 0,
"mailbox_format": "mdbox",
"max_defer_fail_percentage": "unlimited",
"max_email_per_hour": "unlimited",
"max_emailacct_quota": "unlimited",
"maxaddons": "unlimited",
"maxftp": "unlimited",
"maxlst": "unlimited",
"maxparked": "unlimited",
"maxpop": "unlimited",
"maxsql": "unlimited",
"maxsub": "unlimited",
"min_defer_fail_to_trigger_protection": "unlimited",
"outgoing_mail_hold": 1,
"outgoing_mail_suspended": 0,
"owner": "root",
"partition": "home",
"plan": "packagename",
"shell": "/bin/bash",
"startdate": "13 May 22 16:03",
"suspended": 0,
"suspendreason": "not suspended",
"suspendtime": 1594040856,
"temporary": 0,
"theme": "paper_lantern",
"uid": 1001,
"unix_startdate": 1369256589,
"user": "username"
}
]
"""
return whmapi1("listaccts")["acct"]
if __name__ == '__main__' and is_wpos_supported():
with acquire_lock(os.path.join('/var/run/collect_information.lock',), attempts=None):
parser = argparse.ArgumentParser(description="Utility to collect information about user for AccelerateWP")
parser.add_argument("user", type=str, nargs="?")
args = parser.parse_args()
installed_versions = php_get_installed_versions()
vhost_versions = php_get_vhost_versions()
target_accounts = listaccts()
target_accounts = list(filter(lambda x: get_allowed_modules(int(x["uid"])), target_accounts))
if args.user is not None:
target_accounts = list(filter(lambda x: int(x["uid"]) == user_uid(username=args.user), target_accounts))
for account in target_accounts:
username = account["user"]
filtered_vhost_versions = [version for version in vhost_versions if version["account"] == username]
try:
pw = get_pw(username=username)
admin_config_dir = get_admin_config_directory(pw.pw_uid)
info_json = os.path.join(admin_config_dir, "info.json")
with open(info_json, "w") as f:
json.dump(
{"installed_versions": installed_versions,
"vhost_versions": filtered_vhost_versions},
f,
)
owner, group, mode = get_admin_config_permissions(pw.pw_gid)
os.chown(info_json, owner, group)
os.chmod(info_json, mode)
except Exception as e:
_logger.exception("Error during collecting information for %s: %s", username, e)
continue
|