selenium獲取頁面源碼，判斷是否存在指定內容，執行不同的操作

本案例用於解決selenium UI自動化，判斷頁面是否存在指定文字，執行后續不同的操作

主要用到browser.page_source

如，保存百度分享文件到自己的百度盤中，會出現文件被刪除無法保存的情況

 

如果不做判斷，腳本即停止

在腳本中加入下面的內容，即可解決這個問題。

browser.page_source獲取當前頁源碼

if u"分享的文件已經被刪除了" in browser.page_source:#如果文件被刪除，跳過本次循環，重新一輪循環開始
   continue
else:

整個腳本如下：

def keep():
    for line in open('C:\\Users\\Beckham\\Desktop\\python\\1.txt'):#循環讀取百度地址和提取碼
        address = line[0:47]#分離出百度盤地址
        code = line[47:51]#分割出提取碼
        browser.get(address)#打開鏈接  
        if u"分享的文件已經被刪除了" in browser.page_source:#如果文件被刪除，跳過本次循環，重新一輪循環開始
            continue
        else:
            browser.find_element_by_id("ksrmwk1v").send_keys(code)#輸入提取碼
            time.sleep(2)
            browser.find_element_by_xpath("//span[contains(text(),'提取文件')]").click()
            time.sleep(2)
            browser.find_element_by_xpath("//span[contains(text(),'保存到網盤')]").click()
            time.sleep(2)
            browser.find_element_by_xpath("//span[contains(text(),'taotu8')]").click()
            time.sleep(2)
            browser.find_element_by_xpath("//span[contains(text(),'確定')]").click()
            time.sleep(5)
keep()

完