76 lines
1.9 KiB
Python
76 lines
1.9 KiB
Python
"""
|
|
mysql execution module
|
|
"""
|
|
|
|
import salt
|
|
import subprocess
|
|
import logging
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
__virtualname__ = "mysql"
|
|
|
|
def __virtual__():
|
|
"""
|
|
only load module if mysql client is installed
|
|
"""
|
|
if salt.utils.path.which("mysql") is None:
|
|
return (False, "mysql isn't installed on this minion")
|
|
elif salt.utils.path.which("mysqladmin") is None:
|
|
return (False, "mysqladmin isn't installed on this minion")
|
|
else:
|
|
return __virtualname__
|
|
|
|
def _execute(cmd):
|
|
"""
|
|
executes the cmd and returns a salt ret dictionary
|
|
"""
|
|
ret = {}
|
|
LOG.debug(f"Executing: {cmd}")
|
|
process = subprocess.Popen(cmd.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
stdout, stderr = process.communicate()
|
|
if stdout:
|
|
ret["out"] = stdout
|
|
if stderr:
|
|
ret["err"] = stderr
|
|
LOG.debug(f"Exit code: { process.returncode }")
|
|
if process.returncode == 0:
|
|
ret["result"] = True
|
|
else:
|
|
ret["result"] = False
|
|
return ret
|
|
|
|
def check_credentials(username, password, host="127.0.0.1", port="3306"):
|
|
"""
|
|
Check if able to login with credentials
|
|
|
|
:param str username: The username to check
|
|
|
|
:param str password: The password for username
|
|
|
|
:param str host: Optional, the IP or fqdn to the server (default: 127.0.0.1)
|
|
|
|
:param str port: Optional, port the mysql is listening on (default: 3306)
|
|
|
|
CLI Example:
|
|
|
|
salt '*' mysql.check_credentials root secret
|
|
"""
|
|
cmd = f"mysql --host={host} --port={port} --user={username} --password={password} --execute=;"
|
|
return _execute(cmd)["result"]
|
|
|
|
|
|
def set_root_password(password):
|
|
"""
|
|
Set the mysql/mariadb root password
|
|
|
|
:param str password: The password to set
|
|
|
|
CLI Example:
|
|
|
|
salt '*' mysql.set_root_password secret
|
|
"""
|
|
cmd = f"mysqladmin password {password}"
|
|
return _execute(cmd)
|
|
|