Refactored

wo/core/mysql.py

@@ -0,0 +1,136 @@
+"""WordOps MySQL core classes."""
+import pymysql
+from pymysql import connections, DatabaseError, Error
+import configparser
+from os.path import expanduser
+import sys
+import os
+from wo.core.logging import Log
+from wo.core.variables import WOVariables
+
+
+class MySQLConnectionError(Exception):
+    """Custom Exception when MySQL server Not Connected"""
+    pass
+
+
+class StatementExcecutionError(Exception):
+    """Custom Exception when any Query Fails to execute"""
+    pass
+
+
+class DatabaseNotExistsError(Exception):
+    """Custom Exception when Database not Exist"""
+    pass
+
+
+class WOMysql():
+    """Method for MySQL connection"""
+
+    def connect(self):
+        """Makes connection with MySQL server"""
+        try:
+            if os.path.exists('/etc/mysql/conf.d/my.cnf'):
+                connection = pymysql.connect(read_default_file='/etc/mysql/conf.d/my.cnf')
+            else:
+                connection = pymysql.connect(read_default_file='~/.my.cnf')
+            return connection
+        except ValueError as e:
+            Log.debug(self, str(e))
+            raise MySQLConnectionError
+        except pymysql.err.InternalError as e:
+            Log.debug(self, str(e))
+            raise MySQLConnectionError
+
+    def dbConnection(self, db_name):
+        try:
+            if os.path.exists('/etc/mysql/conf.d/my.cnf'):
+                connection = pymysql.connect(db=db_name,read_default_file='/etc/mysql/conf.d/my.cnf')
+            else:
+                connection = pymysql.connect(db=db_name,read_default_file='~/.my.cnf')
+
+            return connection
+        except DatabaseError as e:
+            if e.args[1] == '#42000Unknown database \'{0}\''.format(db_name):
+                raise DatabaseNotExistsError
+            else:
+                raise MySQLConnectionError
+        except pymysql.err.InternalError as e:
+            Log.debug(self, str(e))
+            raise MySQLConnectionError
+        except Exception as e :
+            Log.debug(self, "[Error]Setting up database: \'" + str(e) + "\'")
+            raise MySQLConnectionError
+
+    def execute(self, statement, errormsg='', log=True):
+        """Get login details from /etc/mysql/conf.d/my.cnf & Execute MySQL query"""
+        connection = WOMysql.connect(self)
+        log and Log.debug(self, "Exceuting MySQL Statement : {0}"
+                          .format(statement))
+        try:
+            cursor = connection.cursor()
+            sql = statement
+            cursor.execute(sql)
+
+            # connection is not autocommit by default.
+            # So you must commit to save your changes.
+            connection.commit()
+        except AttributeError as e:
+            Log.debug(self, str(e))
+            raise StatementExcecutionError
+        except Error as e:
+            Log.debug(self, str(e))
+            raise StatementExcecutionError
+        finally:
+            connection.close()
+
+    def backupAll(self):
+        import subprocess
+        try:
+            Log.info(self, "Backing up database at location: "
+                     "/var/wo-mysqlbackup")
+            # Setup Nginx common directory
+            if not os.path.exists('/var/wo-mysqlbackup'):
+                Log.debug(self, 'Creating directory'
+                          '/var/wo-mysqlbackup')
+                os.makedirs('/var/wo-mysqlbackup')
+
+            db = subprocess.check_output(["mysql -Bse \'show databases\'"],
+                                         universal_newlines=True,
+                                         shell=True).split('\n')
+            for dbs in db:
+                if dbs == "":
+                    continue
+                Log.info(self, "Backing up {0} database".format(dbs))
+                p1 = subprocess.Popen("mysqldump {0}"
+                                      " --max_allowed_packet=1024M"
+                                      " --single-transaction".format(dbs),
+                                      stdout=subprocess.PIPE,
+                                      stderr=subprocess.PIPE, shell=True)
+                p2 = subprocess.Popen("gzip -c > /var/wo-mysqlbackup/{0}{1}.s"
+                                      "ql.gz".format(dbs, WOVariables.wo_date),
+                                      stdin=p1.stdout,
+                                      shell=True)
+
+                # Allow p1 to receive a SIGPIPE if p2 exits
+                p1.stdout.close()
+                output = p1.stderr.read()
+                p1.wait()
+                if p1.returncode == 0:
+                    Log.debug(self, "done")
+                else:
+                    Log.error(self, output.decode("utf-8"))
+        except Exception as e:
+            Log.error(self, "Error: process exited with status %s"
+                            % e)
+
+    def check_db_exists(self, db_name):
+        try:
+            if WOMysql.dbConnection(self, db_name):
+                return True
+        except DatabaseNotExistsError as e:
+            Log.debug(self, str(e))
+            return False
+        except MySQLConnectionError as e:
+            Log.debug(self, str(e))
+            return False