51 lines
1.5 KiB
51 lines
1.5 KiB
"""EasyEngine MySQL core classes."""
|
|
import pymysql
|
|
import configparser
|
|
from os.path import expanduser
|
|
import sys
|
|
from ee.core.logging import Log
|
|
|
|
|
|
class EEMysql():
|
|
"""Method for MySQL connection"""
|
|
|
|
def execute(self, statement):
|
|
config = configparser.RawConfigParser()
|
|
cnfpath = expanduser("~")+"/.my.cnf"
|
|
if [cnfpath] == config.read(cnfpath):
|
|
user = config.get('client', 'user')
|
|
passwd = config.get('client', 'password')
|
|
try:
|
|
host = config.get('client', 'host')
|
|
except configparser.NoOptionError as e:
|
|
host = 'localhost'
|
|
|
|
try:
|
|
port = config.get('client', 'port')
|
|
except configparser.NoOptionError as e:
|
|
port = '3306'
|
|
|
|
try:
|
|
conn = pymysql.connect(host=host, port=int(port),
|
|
user=user, passwd=passwd)
|
|
cur = conn.cursor()
|
|
except Exception as e:
|
|
Log.error(self, 'Unable to connect to database: {0} {1}'
|
|
.format(e, "[FAIL]"))
|
|
sys.exit(1)
|
|
|
|
try:
|
|
cur.execute(statement)
|
|
except Exception as e:
|
|
Log.error(self, 'Unable to execute statement: {0} {1}'
|
|
.format(e, "[FAIL]"))
|
|
cur.close()
|
|
conn.close()
|
|
sys.exit(1)
|
|
|
|
cur.close()
|
|
conn.close()
|
|
|
|
# def __del__(self):
|
|
# self.cur.close()
|
|
# self.conn.close()
|
|
|