#coding=utf8'''該模式功能是實現自動運行sql命令，命令執行成功后保存成功截圖并下載結果文件；運行出錯，則保存錯誤截圖；開始運行保存，設置命令截圖。該模塊初始化包含如下信息：self.url：用來保存要訪問的網站鏈接self.driver:創建一個Chrome對象實例，并啟動Chrome瀏覽器self.driver.maximize_window():窗口最大化self.driver.get(self.url)：打開網站鏈接self.driver.implicitly_wait(10)：設置隱性等待時間10秒'''import timefrom selenium import webdriverfrom selenium.webdriver.support.ui import WebDriverWait  #設置默認編碼格式為utf8import sysreload(sys)sys.setdefaultencoding("utf8")class RunXQL(object):    def __init__(self):        self.url=""        self.driver=webdriver.Chrome()        self.driver.maximize_window()        self.driver.get(self.url)        self.driver.implicitly_wait(10)        self.run()          def openCmd(self):        #設置命令文件的路徑，使用RunAll.py腳本時,是.//CmdFile//CMD        #如果直接運行該腳本要更改為：..//CmdFile//CMD        filePath=".//CmdFile//CMD"        #打開文件        cmd=file(filePath,"r+")        #讀取文件        cmd=cmd.read()        #把文件轉換成string類型并解碼        cmds=str(cmd).decode("utf8")        #根據換行符對命令文件進行拆分        #如果不進行該項處理，在命令行結尾會莫名添加一個括號）        cmd=cmds.split("/n")          #返回命令行列表        return cmd        def SetCmdImage(self):        try:            #獲取當前時間并轉換為字符串            now=time.strftime("%Y%m%d%H%M%S")            #設置圖片格式與路徑            #如果直接運行該腳本，變成..//SetCmdImage            #以下涉及路徑的于此處理方法一樣            fileName=".//SetCmdImage"+"//"+now+"xpl.png"            #保存屏幕截圖到相應路徑            self.driver.get_screenshot_as_file(fileName)        except Exception,e:            PRint e        def RunErrorImage(self):        try:            now=time.strftime("%Y%m%d%H%M%S")            fileName=".//RunErrorImage"+"//"+now+"xqlErro.png"            self.driver.get_screenshot_as_file(fileName)        except Exception,e:            print e                         def SuccessImage(self):        try:            now=time.strftime("%Y%m%d%H%M%S")            fileName=".//ExecResutlImage"+"//"+now+"success.png"            self.driver.get_screenshot_as_file(fileName)        except Exception,e:            print e            def inputCMD(self,cmd):        try:            #獲取輸入命令行元素對象            cmdArea=self.driver.find_element_by_class_name("ace_text-input")            #如果元素存在            if cmdArea:                #把命令輸入到文本區域中                cmdArea.send_keys(cmd)                #獲取屏幕截圖                self.SetCmdImage()            else:                print "The element %s not exists!" %("ace_text-input")        except Exception,e:            print e                def runCmd(self):        try:            #獲取執行命令的元素對象            runEle=self.driver.find_element_by_xpath("//*[@id=/"new-tab-cont/"]/div/div[1]/ul[1]/li[1]/a")            #如果元素存在            if runEle:                #點擊該元素運行命令                runEle.click()            else:                print "The element %s not exists!" %("exec")        except Exception,e:            print e                def clickDown(self):        try:            #獲取下載元素對象            download=self.driver.find_element_by_xpath("//*[@id=/"new-tab-cont/"]/div/div[3]/ul/li[1]/a[1]")            #如果元素存在            if download:                #點擊該元素執行下載操作                download.click()            else:                print "The element %s not exists!" %("fa fa-download j-dload")        except Exception,e:            print e             def GetResutl(self):        try:            try:                #獲取命令出錯或執行異常，顯示異常信息的元素對象                #超過60秒如果找不到該元素，失敗                text_danger=WebDriverWait(self.driver, 60).until(lambda driver: driver.find_element_by_xpath("//*[@id=/"new-tab-cont/"]/div/div[3]/div/div[1]/pre"))                #如果元素存在                if text_danger:                    #獲取元素的文本信息                    errInfro=text_danger.text                    #獲取屏幕截圖                    self.RunErrorImage()                    print errInfro            except Exception,e:                pass            try:                     #獲取成功時顯示的文本信息，超過300秒元素不存在則失敗                text_success=WebDriverWait(self.driver, 300).until(lambda driver: driver.find_element_by_class_name("text-success"))                     #如果元素存在                    if text_success:                    #獲取屏幕截圖                    self.SuccessImage()                     print "sql run successs......"                    #點擊下載按鈕下載運行結果文件                    self.clickDown()            except Exception,e:                print e               finally:            #最后關閉瀏覽器            self.driver.close()                def run(self):        try:            cmd=self.openCmd()            self.inputCMD(cmd)            self.runCmd()            self.GetResutl()            print "success......"        except Exception,e:            print e        finally:            #最后關閉瀏覽器            self.driver.close()                    def test():    RunXQL()    if __name__=="__main__":    test()                完整項目源碼下載鏈接：http://download.csdn.net/download/henni_719/9771946