.

salt/modules/_modules/mysql.py

@@ -0,0 +1,75 @@
+"""
+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)
+