import MySQLdb,time,traceback,logging
class DBOperateAction:
def __init__(self,dbhost,dbaccount,dbpasswd,dbname,port=3306,charset="utf8"):
self.dbhost=dbhost
self.dbaccount=dbaccount
self.dbpasswd=dbpasswd
self.dbname=dbname
self.charset=charset
self.port=port
self.db_conn=""
self.db_cursor=""
def connect(self):
try:
self.db_conn=MySQLdb.connect(host=self.dbhost,user=self.dbaccount,passwd=self.dbpasswd,
db=self.dbname,port=self.port,charset=self.charset)
self.db_cursor=self.db_conn.cursor()
return True
except MySQLdb.OperationalError:
logging.error("Connect to "+self.dbhost+" Failed")
logging.exception("exception message:")
return False
def re_connect(self):
logging.error("connect to mysql server failed, reconnect")
try:
self.db_conn=MySQLdb.connect(host=self.dbhost,user=self.dbaccount,passwd=self.dbpasswd,
db=self.dbname,port=self.port,charset="utf8")
self.db_cursor=self.db_conn.cursor()
return True
except MySQLdb.OperationalError:
logging.error("Reconnect MySQL failed")
return False
def get_all_result(self,sql):
try:
logging.info("Execute sql: "+sql)
self.db_cursor.execute(sql)
self.db_conn.commit()
result=self.db_cursor.fetchall()
return result
except MySQLdb.OperationalError:
logging.exception("Execure SQL except"+sql)
for i in range(0,3):
time.sleep(5)
if self.re_connect():
logging.info("reconnect database successfully")
return False
def get_one_result(self,sql):
try:
logging.info("Execute sql: "+sql)
self.db_cursor.execute(sql)
self.db_conn.commit()
result=self.db_cursor.fetchone()
return result
except MySQLdb.OperationalError:
traceback.print_exc()
for i in range(0,3):
time.sleep(5)
if self.re_connect():
break
def close_connection(self):
self.db_conn.close()