一、函數的定義，並引用

public_model.py

#登陸
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("退出").click()
driver.quit()

diaoyong_publicmodel.py 調用模塊化的函數文件

#coding=utf-8
#調用public文件的登錄與退出函數
from selenium import webdriver
from public_model import login,logout

driver = webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")

#登陸
login(driver)

#退出
logout(driver)

二、使用類進行模塊化及引用

public_class.py

#encoding=utf-8
class Login():
def __init__(self):
self.driver=driver
def login(self):
self.driver.find_element_by_id("idInput").clear()
self.driver.find_element_by_id("idInput").send_keys(‘username‘)
self.driver.find_element_by_id("pwdInput").clear()
self.driver.find_element_by_id("pwdInput").send_keys(‘password‘)
self.driver.find_element_by_id("loginBtn").click()

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

diaoyong_publicclass.py

#coding=utf-8
from selenium import webdriver
#調用public文件的Login類裏的函數
from public_class import Login

driver = webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")

#登陸
Login(driver).login()

#退出
Login(driver).logout()

selenium 模塊化實例