在selenium的自動化指令碼中，有許多東西是總是呼叫的，比如登陸，那麼就可以將其模組化，獨立出來，從而可以使自動化指令碼更加簡潔．如下以登陸為例來實現模組化：

現有一個自動化指令碼如下，為驗證是否正常登陸的：

test1.py

# -*- coding: UTF-8 -*-
'''
Created on 2015年3月28日

@author: dff
'''
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
from selenium.webdriver.support.wait import WebDriverWait

class Login_user(unittest.TestCase):
    def setUp(self):
        self.driver =webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://192.168.11.14/"
        self.verificationErrors = []
        self.accept_next_alert = True
        self.driver.get(self.base_url + "/auth/login/")  
        
            
    def test_login_admin(self):
        '''管理員使用者登陸'''
        driver=self.driver
        time.sleep(0.5)
        driver.find_element_by_id("id_username").clear()
        driver.find_element_by_id("id_username").send_keys("admin")        
        driver.find_element_by_id("id_password").clear()
        driver.find_element_by_id("id_password").send_keys("ADMIN_PASS")
        driver.find_element_by_xpath("//button[@type='submit']").click()
        driver.get_window_position(windowHandle)
        d=driver.get_window_size('current')
        print d
        print driver.get_window_position("current")
        url=driver.current_url
        print url
        handle=driver.current_window_handle
        print handle
        self.assertEqual( u"使用概況 - Delicate Cloud Dashboard", driver.title,'login fail')
   
    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException: return False
        return True
    
    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException: return False
        return True
    
    def close_alert_and_get_its_text(self):
        try:
            alert = self.driver.switch_to_alert()
            alert_text = alert.text
            if self.accept_next_alert:
                alert.accept()
            else:
                alert.dismiss()
            return alert_text
        finally: self.accept_next_alert = True
    
    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    unittest.main()
 

        driver.find_element_by_id("id_username").clear()
        driver.find_element_by_id("id_username").send_keys("admin")        
        driver.find_element_by_id("id_password").clear()
        driver.find_element_by_id("id_password").send_keys("ADMIN_PASS")
        driver.find_element_by_xpath("//button[@type='submit']").click()
 

# -*- coding:UTF-8 -*-
'''
Created on 2015年4月2日

@author: dff
'''
import unittest,time
from  selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
from selenium.webdriver.support.wait import WebDriverWait
class lo():
    def __init__(self,driver):
        self._driver=driver
    def login_(self,name='admin',pw='ADMIN_PASS'): 
        self._driver.find_element_by_id("id_username").clear()
        self._driver.find_element_by_id("id_username").send_keys(name)
        self._driver.find_element_by_id("id_password").clear()
        self._driver.find_element_by_id("id_password").send_keys(pw)
        self._driver.find_element_by_xpath("//button[@type='submit']").click()
 

webtest-->src-->justfortest/test1.py

                             common/action.py

那麼在test1.py中呼叫可以用如下方法：

本例中將此登陸呼叫放在setup函式中：

首先需要import此模組，如

from common.action import lo

修改後部分程式碼如下：

class Login_user(unittest.TestCase):
    def setUp(self):
        self.driver =webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://192.168.11.14/"
        self.verificationErrors = []
        self.accept_next_alert = True
        self.driver.get(self.base_url + "/auth/login/")
        lo(self.driver).login_()

{u'width': 1164, u'height': 629}
{u'y': 24, u'x': 436}
http://192.168.11.14/admin/
{ce3619ce-bb1b-492f-ba92-6585e05f8416}
.
----------------------------------------------------------------------
Ran 1 test in 10.524s

OK

如此即可將test_admin_login函式中的登陸部分去掉，實現模組化指令碼；

以此為方法，可將其他部分模組化，如退出，切換目錄等；