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