Operation   : Linux premium131.web-hosting.com 4.18.0-553.44.1.lve.el8.x86_64 #1 SMP Thu Mar 13 14:29:12 UTC 2025 x86_64
Software     : Apache
Server IP    : 162.0.232.56 | Your IP: 216.73.216.111
Domains      : 1034 Domain(s)
Permission   : [ 0755 ]

# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT

import argparse
from xml.dom import minidom, DOMException, NotFoundErr
from xml.parsers.expat import ExpatError


def set_governor_save_uid(config_path="/etc/container/mysql-governor.xml", save_uid=True):
    """

    :param str config_path:
    :param bool save_uid:
    :return:
    """
    try:
        governor_config = minidom.parse(config_path)
    except (IOError, ExpatError, DOMException):
        pass
    else:
        try:
            if save_uid:
                governor_config.getElementsByTagName("statistic")[0].setAttribute("save_uid", "On")
            else:
                governor_config.getElementsByTagName("statistic")[0].removeAttribute("save_uid")
        except (NotFoundErr, IndexError, KeyError):
            pass
        else:
            with open(config_path, 'w', encoding='utf-8') as config_file:
                governor_config.writexml(config_file)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--save_uid',
        action='store_true',
        dest='save_uid',
        default=False)
    opts = parser.parse_args()
    if opts.save_uid:
        set_governor_save_uid(save_uid=True)
    else:
        set_governor_save_uid(save_uid=False)