# -*- coding: utf-8 -*-
# databaseModule.py

import  config
import  MySQLdb
import  traceback
import  logging as logger

class  MyDB (object):
    def  connect (self, user=config.DB_USER, password=config.DB_PASSWORD, \
            host = config.DB_HOST, database=config.DB_DATABASE, unix_socket=config.UNIX_SOCKET):
        if not  connection:
            try:
                self.connection = MySQLdb.connect ()
            except  mysql.connector.Error as err:
                logger.exception ('Database connection failed for ' + config.DB_USER + '@' + config.DB_HOST + '/' + config.DB_DATABASE)
                exit ()

    def  execute (self, sql, *args):
        retval =  None
        try:
            cursor = self.connection.cursor ()
            retval = cursor.execute (sql, *args) # when you pass a tuple to this method execute get it as is
            self.cursor.fetchall ()
        except  ( AttributeError, MySQLdb.OperationalError ):
            logger.exception ('Connection finished. Retrying connection')
            self.connect ()
            self.connection.commit ()
            self.execute (sql, *args)
        except  MySQLdb.Error:
            logger.exception ('Execution error')
            self.connection.rollback ()
            raise
        return retval #, self.cursor () return also cursor. Return value with dictionary

    def  query (self, sql, *args):
        retval = None
        try:
            retval = self.execute (sql)
        except  ( AttributeError, MySQLdb.OperationalError ):
            logger.exception ('Connection finished. Retrying connection')
            self.connect ()
            self.query (sql, *args)
        except  MySQLdb.Error:
            logger.exception ('Execution error')
            self.connection.rollback ()
            raise
        return  retval

    def  commit (self):
        self.connection.commit ()

    def  disconnect (self):
        self.connection.close ()