python3 关于函数封装简单介绍，以百度搜索为例子

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