python selenium 模块化编码

公共模块

public.py

#coding=utf-8

#登陆
def login(driver):
driver.find_element_by_id("idInput").clear()
driver.find_element_by_id("idInput").send_keys("username")
driver.find_element_by_id("pwdInput").clear()
driver.find_element_by_id("pwdInput").send_keys("password")
driver.find_element_by_id("loginBtn").click()

#退出
def logout(driver):
driver.find_element_by_link_text(u"退出").click()
driver.quit()

把driver当成函数变量

当函数被独立到单独的文件中时做了一点调整，主要在函数的传参上，因为函数内部的操作需要使用
driver，但是 driver 并没有在此文件中定义，所以需要调用的用例传递 driver 给调用的函数。

 

功能模块

mail126.py

 

#coding=utf-8
from selenium import webdriver
import public
driver = webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")
#调用登陆模块
public.login(driver)  #并传入driver实参
#收信、写信、删除信件等操作
#调用退出模块
public.logout(driver)   #并传入driver实参

首先，需要导入当前目录中的 public.py，在需要的位置调用文件中的 login()和和 logout()函数。