python-mylib/mylib/mysql.py

59 lines
1.4 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
""" MySQL client """
import logging
import sys
import MySQLdb
log = logging.getLogger(__name__)
class MyDB:
""" MySQL client """
host = ""
user = ""
pwd = ""
db = ""
con = 0
def __init__(self, host, user, pwd, db):
self.host = host
self.user = user
self.pwd = pwd
self.db = db
def connect(self):
""" Connect to MySQL server """
if self.con == 0:
try:
con = MySQLdb.connect(self.host, self.user, self.pwd, self.db)
self.con = con
except Exception:
log.fatal('Error connecting to MySQL server', exc_info=True)
sys.exit(1)
def doSQL(self,sql):
""" Run INSERT/UPDATE/DELETE/... SQL query """
cursor = self.con.cursor()
try:
cursor.execute(sql)
self.con.commit()
return True
except Exception:
log.error('Error during SQL request "%s"', sql, exc_info=True)
self.con.rollback()
return False
def doSelect(self, sql):
""" Run SELECT SQL query and return result as dict """
cursor = self.con.cursor()
try:
cursor.execute(sql)
return cursor.fetchall()
except Exception:
log.error('Error during SQL request "%s"', sql, exc_info=True)
return False