You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.8 KiB
58 lines
1.8 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, errormsg=''):
|
|
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:
|
|
if errormsg:
|
|
Log.debug(self, '{0}'
|
|
.format(e))
|
|
Log.error(self, '{0}'
|
|
.format(errormsg))
|
|
else:
|
|
Log.debug(self, '{0}'
|
|
.format(e))
|
|
Log.error(self, 'Unable to connect to database: {0}'
|
|
.format(e))
|
|
|
|
try:
|
|
cur.execute(statement)
|
|
except Exception as e:
|
|
Log.error(self, 'Unable to execute statement:')
|
|
Lod.debug(self, "{0}".format(e))
|
|
cur.close()
|
|
conn.close()
|
|
sys.exit(1)
|
|
|
|
cur.close()
|
|
conn.close()
|
|
|
|
# def __del__(self):
|
|
# self.cur.close()
|
|
# self.conn.close()
|
|
|