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

完