import os from subprocess import PIPE, Popen def get_users_count_from_cllib(): """ Get user count using the common CloudLinux library. This number is more accurate for systems with a control panel installed. """ if not os.path.exists('/opt/cloudlinux/venv/bin'): raise ValueError("CloudLinux virtual environment not found") cmd = '/opt/cloudlinux/venv/bin/python3 -c "from clcommon.cpapi import cpusers; print(cpusers())"' process = Popen(cmd, shell=True, stdout=PIPE, stderr=PIPE) output, errors = [result.decode().strip() for result in process.communicate()] if errors: raise ValueError(f"Failed to get users from CloudLinux library: {errors}") return len(output[1:-1].split(', ')) def get_users_count_generic(): """ Fallback method to get the user count by looking into the /etc/passwd file. """ from up2date_client.clpwd import ClPwd pwd = ClPwd() return len(pwd.get_uid_dict()) def count_server_users(): """ Get the total count of users on the server. Tries the CloudLinux library first, falls back to generic method if it fails. """ try: users_count = get_users_count_from_cllib() except Exception: users_count = get_users_count_generic() return users_count