python线性脚本,我们都很熟悉,以下代码百度搜索demo,其代码是线性脚本,我们先看下。
以下是demo1文件的代码
from selenium import webdriver
from time import sleep
dr = webdriver.Chrome()
dr.get('http://www.baidu.com')
sleep(3)
dr.find_element_by_id('kw').send_keys('demo')
dr.find_element_by_id('su').click()
sleep(3)
dr.quit()
从上面看,脚本从上往下按步骤执行。假如我想把以下的代码封装起来呢,
dr.find_element_by_id(‘kw’).send_keys(‘demo’)
dr.find_element_by_id(‘su’).click()
那么我们怎么修改才能实现函数的封装,然后调用函数呢?
首先我们新建一个demo2 的python文件。新建demo2的目的是把函数放在公共的模块,方便调用,其代码如下
'''函数传参必须传入dr驱动,def search_text(dr):'''
def search_text(dr):
dr.find_element_by_id('kw').send_keys('demo')
dr.find_element_by_id('su').click()
demo1文件的代码修改如下
from selenium import webdriver
from time import sleep
from demo2 import search_text #导入demo2的search_text函数
dr = webdriver.Chrome()
dr.get('http://www.baidu.com')
sleep(3)
'''函数传参必须传入dr驱动,search_text(dr)'''
search_text(dr) #调用demo2的search_text(dr)
sleep(3)
dr.quit()