selenium练习2【po模式优化代码】

from selenium import webdriver
from selenium_po.page.contact_page import ContactPage

class MainPage:
    def goto_contact_page(self):
        opt = webdriver.ChromeOptions()
        opt.debugger_address = "127.0.0.1:9222"
        driver = webdriver.Chrome(options=opt)
        driver.implicitly_wait(10)
        driver.get("https://work.weixin.qq.com/wework_admin/frame")
        driver.find_element_by_id("menu_contacts").click()

        return ContactPage()

import time

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.support.wait import WebDriverWait


class ContactPage:

    def click_add_member(self):
        opt = webdriver.ChromeOptions()
        opt.debugger_address = "127.0.0.1:9222"
        driver = webdriver.Chrome(options=opt)
        driver.implicitly_wait(10)
        from selenium_po.page.add_member_page import AddMemberPage

        ele = (By.CSS_SELECTOR, ".ww_operationBar .js_add_member")
        WebDriverWait(driver, 10).until(expected_conditions.element_to_be_clickable(ele))
        while True:
            # ele = driver.find_element_by_css_selector(".ww_operationBar .js_add_member")
            # ele.click()  #不停点，直到添加成员页面出现
            driver.find_element(*ele).click()# 解数组
            element = driver.find_elements_by_id("username")#用elements找到username,说明进入到添加成员页面
            if len(element) > 0:
                break


        return AddMemberPage()
    def get_member(self):
        opt = webdriver.ChromeOptions()
        opt.debugger_address = "127.0.0.1:9222"
        driver = webdriver.Chrome(options=opt)
        driver.implicitly_wait(10)

        time.sleep(2)
        eles = driver.find_elements_by_css_selector('.member_colRight_memberTable_td:nth-child(2)')
        name_list = []
        for value in eles:
            # 获取元素属性title的值，存入list内
            print(value.get_attribute("title"))
            # if value.get_attribute("title") == "七ds月3":
            #     return True
            # return False
            name_list.append(value.get_attribute("title"))
        # 断言目标名字是否在列表内
        return name_list

from selenium import webdriver


class AddMemberPage:
    def add_member(self):
        from selenium_po.page.contact_page import ContactPage
        opt = webdriver.ChromeOptions()
        opt.debugger_address = "127.0.0.1:9222"
        driver = webdriver.Chrome(options=opt)
        driver.implicitly_wait(10)
        driver.find_element_by_id("username").send_keys("七ds月1")
        driver.find_element_by_id("memberAdd_english_name").send_keys("uii")
        driver.find_element_by_id('memberAdd_acctid').send_keys("t9jg6")
        driver.find_element_by_id("memberAdd_phone").send_keys("15304005545")
        driver.find_element_by_id("memberAdd_mail").send_keys("22896809@qq.com")
        driver.find_element_by_css_selector(".js_btn_save").click()
        return ContactPage()

from selenium_po.page.main_page import MainPage


class TestLogin:
    def setup(self):
        self.main = MainPage()
    def teardown(self):
        pass
    def test_login(self):
        namelist = self.main.goto_contact_page().click_add_member().add_member().get_member()
        print(namelist)
        assert "七ds月1" in namelist

``