python开发数据库_用Python实现数据库的自动化测试（2）

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()