用python-webdriver实现自动填表

  在日常工作中常常需要重复填写某些表单,如果人工完成,费时费力,而且网络延迟令人十分崩溃。如果能够用程序实现自动填表,效率可以提高一倍以上,并且能够移植到多台计算机,进一步提高工作效率。webdriver是python的selenium库中的一个自动化测试工具,它能完全模拟浏览器的操作,无需处理复杂的request、post,对爬虫初学者十分友好。

一、环境配置

  python3.6+selenium库+xlrd库+xlwt库

  其中xlrd和xlwt库用于读写excel表中的数据。

  还要下载一个浏览器的driver文件用于打开浏览器,注意要选择与计算机系统相符合的版本(max/windows64位/windows32位)

  ChromeDriver:http://npm.taobao.org/mirrors/chromedriver/

     IEDriver:http://selenium-release.storage.googleapis.com/index.html

  将下载下来的driver.exe放到浏览器根目录和python的根目录

二、打开网页

  以IE浏览器为例,以下两行代码就可以实现打开一个IE浏览器并且访问我们需要填表的网站

    driver= webdriver.Ie()
    driver.get('http://xxxx.com/')

  如果网站需要登陆(需要填表的一般是公司内部网站),再写一个login函数,将driver作为参数调用

  driver = login(driver)

  注意一定要将driver传回,这样driver才能继续接受程序的指令

三、元素定位

  webdriver的工作原理是找到网页中某一个元素,可以对其进行填入数据或点击等操作。

  关于元素定位可以参考这篇博客https://blog.csdn.net/bananasssss/article/details/51316369

  我主要用到的元素定位方式有

  driver.find_element_by_id('someid')#通过元素的id定位
  driver.find_element_by_css_selector("input[value='确定'")#查找一个input元素,它的value属性值为'确定'
  driver.find_element_by_xpath("//span[contains(@style,'COLOR: red')]/span[1]")#查找一个style属性值为'COLOR:red'的span元素的第一个span子元素

 (1)通过id定位

  如果我们想在网页表单的某一个位置填某项值或者点击某个按钮,我们首先要用开发者工具查看这个元素的源代码,然后首先观察它有没有id,如果有id,直接用id定位该元素。然后,用

  driver.find_element_by_id('someid').click()#点击元素
  driver.find_element_by_id('someid').send_keys('somekeys')#填入'somekeys'
  driver.find_element_by_id('someid').clear()#清空输入框中已有的值

  实现我们想要做的操作。

 (2)通过ccs selector定位

  如果我们想要操作的元素没有ID,那么我们就要找到它跟网页其他元素不同的特征,ccs selector是一种十分灵活的定位方式,其中用value定位是一个不错的选择。以

  driver.find_element_by_css_selector("input[value='确定'")

  为例,双引号中的input可以换成任何网页元素(div、span、input、a等),中括号中是该元素的某一个属性(style、id、value、class等),等号后面是该属性的值。

  注意,如果网页中有多个元素同时满足ccs selector的条件,如有多个value=“确定” 的input,那么find_element_by_css_selector只会定位到在html源代码中最靠前的一个,而find_elements_by_css_selector会找到源代码中所有满足条件的元素,并以列表的形式返回这些找到的元素。例如,网页中弹出很多个提示框,我们要一一去点确定,可以这样操作

    list=driver.find_elements_by_css_selector("input[value=' 确定 ']")
    for l in list:
        l.click()

  但是,如果这些提示框是重叠出现的,而最上层的提示框实际上在源码中更靠后的位置,那么列表中第一个“确定”元素就会被叠在上面的提示框遮挡,无法点击,这个时候倒序一下数组就可以了,从最后一个“确定”元素开始点击

     query=driver.find_elements_by_css_selector("input[value=' 确定 ']")
     for q in query[::-1]:
         q.click()

 (3)通过xpath定位

  关于xpath定位详解可以参考https://www.jianshu.com/p/820dcd013993

  xpath定位比较复杂但是非常全面,当这个元素的class、style属性和其他元素一样,实在没什么特点可以一步定位的时候,我们就可以用xpath,先找到我们想要的元素的父子兄弟元素,再定位到我们想要的元素。例如

   driver.find_element_by_xpath('//*[@class="submit clear"]/input[1]').click()
  text =driver.find_element_by_xpath("//input[@value=' 确定 ']/../preceding-sibling::div[1]").text
   driver.find_elements_by_xpath("//span[contains(@style,'COLOR: red')]/span[1]")

  引号中的//表示相对定位,表示从源代码中任何地方开始寻找。

  //后可以跟任何元素,*代表任意元素,即定位符合属性筛选任何元素。

  中括号内是属性的筛选条件,@后可以加任意属性。contains(@style,'COLOR: red')表示的筛选条件是:style属性中包含”COLOR:red“。这里为什么不直接用@style='COLOR: red'

的原因是,可能在我们审查源代码的时候这个元素的style属性只有'COLOR: red'这一条,但是动态界面的style属性经常变化,程序运行时直接用等于是定位不到这个元素的。

  我们通常需要靠先找到某个有id的元素,再通过层级关系定位到我们真正想要定位的元素,关于兄弟父子元素定位请参考https://blog.csdn.net/huilan_same/article/details/52541680

  /..  可以定位这个元素的父亲元素

  /  可以定位这个元素的子元素

  /preceding-sibling::  可以定位这个元素的哥哥元素

  /following-sibling::  可以定位这个元素的弟弟元素

  如/input[1]表示子元素中第一个input、/../preceding-sibling::div[1]表示父元素的哥哥元素中的第一个div

 

 (4)通过当前节点定位

  有时候我们会遇到需要判断一下元素当前的状态(是否被选择)再决定接下来的操作的情况,这时就需要用一个变量来保存当前节点

   LTE=driver.find_element_by_xpath("//input[@id='LTE']/../span[1]"

  然后再用get_attribute获得当前节点元素的属性,在这个例子里,如果元素为蓝色,就不需要点击。代码实现为:

    if LTE.get_attribute("style")=="COLOR: blue":
        pass
    else:
        LET.click()

 

  需要筛选出特定文本的情况:

    red=driver.find_elements_by_xpath("//span[contains(@style,'COLOR: red')]/span[1]")#找出所有红色的文本
    for r in red:
        if '低消' in r.text:#如果文本信息中包含‘低消’
            r.find_element_by_xpath("./../preceding-sibling::input[1]").click()#注意从当前节点定位的时候要以‘./’开头
            break

  

  如果寻找的元素需要滚动界面才能看到,这个时候可以用js聚焦此元素,页面便会滚动到该元素的位置

    target=driver.find_element_by_css_selector("input[value=' 确定 ']")
    driver.execute_script("arguments[0].scrollIntoView();", target)
    target.click()

  

四、不确定情况处理

 (1)有可能出现的弹窗

  在填表过程中,有些地方有可能出现一个弹框也有可能不出现,这个时候,无论这个弹窗是什么,用try..except语句处理就可以解决

  js触发的弹窗:

    try:
        driver.find_element_by_css_selector("input[value=' 确定 ']").click()
    except Exception as e:
        pass

  网页alert弹窗:

    try:
        driver.switch_to.alert.dismiss()
    except Exception:
        pass

  dismiss()对应的是alert弹窗的”取消“项,accept()对应的是”确定“项,driver.switch_to.alert.text 可以获得弹窗的文本内容。

  

 (2)数量不定的弹窗

  对上文提到的多个提示框情况,除了用 query=driver.find_elements_by_css_selector("input[value=' 确定 ']") 一次性找到所有元素再顺序或倒序点击之外,还可以用一个while循环解决

    while(1):
        try:
            driver.find_element_by_css_selector("input[value=' 确定 ']").click()
        except Exception as e:
            break

  

 (3)网络延迟

  有些网页在点击查询信息之后需要加载一段时间,加载中的页面是找不到我们接下来想找的元素的,因此程序就会报错,此时有两种解决方法。

  一种是固定等待一段时间,等待网页加载完毕,这种方法的缺点是很难找到等待的最佳时间,太短的话页面还没加载完,太长就影响效率

    time.sleep(2)

  另一种是用一个while循环一直寻找下一个我们要找的元素

    while(1):
        try:
            driver.find_element_by_id('continueTrade').click()
            break
        except Exception:
            pass

  这种方法的前提是下一个要找的元素必定会出现

 

五、frame处理

  关于frame处理这篇博客写得非常好https://blog.csdn.net/huilan_same/article/details/52200586

  总结起来就是:frameset不用切,frame层层切。最好一系列填表操作完后都用 driver.switch_to.default_content() 回到原文档,这样不容易混乱

  这里再补充一点frame没有id时的切入方法

    frame= self.driver.find_element_by_xpath("/html/body/div[12]/iframe")#先定位frame位置,用一个变量储存这个节点
    self.driver.switch_to_frame(frame)#再切入这个节点

  

六、excel数据读写

  excel数据读写十分简单,看代码就好了:

def read(file):
    data = xlrd.open_workbook(file)#打开excel文件
    table = data.sheets()[0]#读取第一个sheet的数据
    phones = table.col_values(0)#以列表形式存储第一列数据
    peoples = table.col_values(1)#以列表形式存储第二列数据

    return phones,peoples

def write(result):
    file=xlwt.Workbook()#创建一个excel文件
    table = file.add_sheet('sheet1')#添加一个sheet
    for i in range(len(result)):#写入数据
        table.write(i,0,result[i][0])
        table.write(i,1,result[i][1])
        table.write(i,2,result[i][2])
    file.save('result.xls')

  

结语:希望技术能让人们从无意义的重复劳动中解脱:D

 

posted @ 2018-05-07 15:59  luozx207  阅读(29106)  评论(2编辑  收藏  举报