python selenium page object_selenium-python PageObject设计模式

from selenium import webdriver

from selenium.webdriver.common.by import By

from time import sleep

from pip._vendor.distlib.locators import Page

from selenium.webdriver.support.ui import Select#导入下拉框函数

import unittest

class Page(object):

login_url = 'http://qf-uatqapp-w2/ProductApplication/Application'

def __init__(self, selenium_driver, base_url=login_url):

self.base_url = base_url

self.driver = selenium_driver

self.timeout = 30

self.driver.maximize_window()

def on_page(self):

return self.driver.current_url == (self.base_url + self.url)

def _open(self, url):

url=self.base_url

self.driver.get(url)

self.driver.maximize_window()

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

def switch_frame(self, loc):

return self.driver.switch_to_frame()

class LoginPage(Page):

url = '/'

username_loc = (By.ID, "Account")

password_loc = (By.ID, "Pwd")

submit_loc = (By.ID,"Log_Submit")

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

class ProductPage(Page):

customername_loc = (By.ID, "customerName")

customerID_loc = (By.ID,"customerIDCard")

applyCity_loc = (By.ID,"applyCity")

# productCode_loc = (By.ID,"//option[@value='productCodesyd-6-24']")

# productCode_value = (By.XPATH,"//option[@value='BYQSF0000']")

# platform_loc = (By.ID,"platform")

#platform_value = (By.XPATH,"//option[@value='BYQSF0000']")

def type_customername(self, customername):

self.find_element(*self.customername_loc).send_keys(customername)

def type_customerID(self, customerId):

self.find_element(*self.customerID_loc).send_keys(customerId)

def type_applyCity(self, applyCityvalue):

Select(self.find_element(*self.applyCity_loc)).select_by_value(applyCityvalue)#下拉框函数

def test_user_login(driver, username, password):

page = LoginPage(driver)

page.open()

page.type_username(username)

page.type_password(password)

page.submit()

def test_apply_product(driver,customername, customerID, applyCityvalue):

page1 = ProductPage(driver)

page1.type_customername(customername)

page1.type_customerID(customerID)

page1.type_applyCity(applyCityvalue)

def main():

driver = webdriver.Chrome()

username = 'jiahua'

password = 'Quarkhj05'

customername = u'huajia'

customerId = '310104198408020057'

applyCityvalue = '025,025'

test_user_login(driver,username,password)

driver.implicitly_wait(30)

test_apply_product(driver,customername, customerId, applyCityvalue)

if __name__ == '__main__':

main()

unittest.main()