# 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)