python 接口自动化测试--代码实现(七)
数据库还原脚本:
#! /usr/bin/python # coding:utf-8 import time from Engine.SqlEngine import MSSQL COUNT=1 def restoreRelease(): global COUNT checkConSql = "select spid from sysprocesses where dbid in (select dbid from sysdatabases where name='SOATest')" restoreSql = "RESTORE DATABASE SOATest FROM DATABASE_SNAPSHOT = 'SOATest_ss'" dbc=MSSQL('192.168.1.2','yht','yht','Master') conNum = dbc.ExecQuery(checkConSql) if COUNT < 5: if len(conNum) == 0: print(u'%d条连接数,可以还原数据库,还原中...'%len(conNum)) dbc.ExecNonQuery(restoreSql) print(u'数据库还原完成') return True else: print(u'%d条连接数占用无法还原数据库,5秒后重试'%len(conNum)) time.sleep(5) COUNT=COUNT+1 restoreRelease() else: print(u'%d条连接数始终占用,已试过5次依然无法还原数据库'%len(conNum)) return False
SqlEngine.py
#! /usr/bin/python # coding:utf-8 import pymssql import sys from Engine.DataEngine import decoCatchExcep reload(sys) sys.setdefaultencoding('utf-8') class MSSQL(object): """ 对pymssql的简单封装 pymssql库,该库到这里下载:http://www.lfd.uci.edu/~gohlke/pythonlibs/#pymssql 使用该库时,需要在Sql Server Configuration Manager里面将TCP/IP协议开启 用法: """ def __init__(self,host,user,pwd,db): self.host = host self.user = user self.pwd = pwd self.db = db def __GetConnect(self): """ 得到连接信息 返回: conn.cursor() """ if not self.db: raise(NameError,"没有设置数据库信息") self.conn = pymssql.connect(host=self.host,user=self.user,password=self.pwd,database=self.db,charset="utf8") cur = self.conn.cursor() if not cur: raise(NameError,"连接数据库失败") else: return cur @decoCatchExcep def ExecQuery(self,sql): cur = self.__GetConnect() cur.execute(sql) resList = cur.fetchall() self.conn.close() return resList @decoCatchExcep def ExecNonQuery(self,sql): cur = self.__GetConnect() self.conn.autocommit(True) cur.execute(sql) self.conn.autocommit(False) self.conn.commit() self.conn.close()
装饰器decoCatchExcep:
def decoCatchExcep(func): def _decoCatchExcep(*args, **kwargs): try: ret = func(*args, **kwargs) return ret except Exception,e: print(func.__name__+' : '+str(e).encode('gb18030')) LogPro.writeException(str(e).encode('gb18030')) return _decoCatchExcep