object设计模式.py

from selenium import webdriverfrom selenium.webdriver.common.by import Byfrom time import sleep#创建page类：class Page(object):    """    基础类、用于页面对象类的继承    """    login_url = "http://www.126.com"    def __init__(self,selenium_driver,base_url = login_url):        """        首先创建一个基础类Page，在初始化方法init）中定义驱动（driver）、基本的URL        （base_url）和超时时间（timeout）等。        :param selenium_driver:        :param base_url:        """        self.base_url = base_url        self.driver = selenium_driver        self.timeout = 30

    def on_page(self):        return self.driver.current_url == (self.base_url + self.url)

    def _open(self,url):        url = self.base_url + url        self.driver.get(url)        assert self.on_page(),"Did not land on %s" %url

    def open(self):        self._open(self.url)

    def find_element(self,*loc):        return self.driver.find_element(*loc)

"""定义open）方法用于打开URL网站，但它本身并未做这件事情，而是交由_open）方法来实现。关于URL地址的断言部分，则交由on page（）方法来实现，而find_element）方法用于元素的定位。"""

#创建LoginPage类：class LoginPage(Page):    """    126邮箱登录页面模型    LoginPage类中主要对登录页面上的元素进行封装，使其成为更具体的操作方法。例如，用户名、密码和登录按钮都被封装成了方法。    """    url = "/"    #定位器    username_loc = (By.ID,"idInput")    password_loc = (By.ID,"pwdInput")    submit_loc = (By.ID,"loginBtn")    #Action    def type_username(self,username):        self.find_element(*self.username_loc).send_keys(username)

    def type_password(self,password):        self.find_element(*self.password_loc).send_keys(password)

    def submit(self):        self.find_element(*self.submit_loc).click()

def test_user_login(driver,username,password):    """    测试获取的用户名/密码是否可以登录    test_user_login）函数将单个的元素操作组成一个完整的动作，而这个动作包含了打开浏览器、输入用户名/密码、点击登录等单步操作。在使用该函数时需要将driver、username、password等信息作为函数的入参，这样该函数具有很强的可重用性。    :param driver:    :param username:    :param password:    :return:    """    login_page = LoginPage(driver)    login_page.open()    login_page.type_username(username)    login_page.type_password(password)    login_page.submit()

def main():    try:        driver = webdriver.Firefox()        username = "username"        password = "123456"        test_user_login(driver,username,password)        sleep(3)        text = driver.find_element_by_xpath("//span[@id=‘spnUid‘]").text        assert (text == "[email protected]") #用户名称不匹配、登录失败！    finally:        #关闭浏览器窗口：        driver.close()

if __name__ == ‘__main__‘:    main()

原文地址：https://www.cnblogs.com/zhang-da/p/12210535.html