[fix] Don't overwrite mysql module

salt/_modules/so.py

@@ -1,4 +1,38 @@
 #!py
 
+import logging
+
 def status():
-  return __salt__['cmd.run']('/usr/sbin/so-status')
+  return __salt__['cmd.run']('/usr/sbin/so-status')
+
+
+def mysql_conn(retry):
+  from MySQLdb import _mysql
+  import time
+
+  log = logging.getLogger(__name__)
+  mainint = __salt__['pillar.get']('sensor:mainint', __salt__['pillar.get']('manager:mainint'))
+  mainip = __salt__['grains.get']('ip_interfaces').get(mainint)[0]
+
+  mysql_up = False
+  for i in range(0, retry):
+      log.debug(f'Connection attempt {i+1}')
+      try:
+          _mysql.connect(
+              host=mainip,
+              user="root",
+              passwd=__salt__['pillar.get']('secrets:mysql')
+          )
+          mysql_up = True
+          break
+      except _mysql.OperationalError as e:
+          log.debug(e)
+      except Exception as e:
+          log.error(e)
+          break
+      time.sleep(1)
+
+  if not mysql_up:
+      log.error(f'Could not connect to MySQL server on {mainip} after {retry} attempts.')
+
+  return mysql_up