python selenium 实现UI自动化POM设计模式

# 对操作对象封装
class BasePage:
  driver = webdriver.Chrome()
# 访问URL
  def visit(self, url):
    self.driver.get(url)# 元素定位
  def locator(self, loc):
     return self.driver.find_element(*loc)# 输入
  def input(self, loc, txt):
     self.locator(loc).send_keys(txt)
     # 点击
  def click(self, loc):
     self.locator(loc).click()# 等待
  def wait(self, time):
     sleep(time)# 关闭
  def close(self):
     self.driver.quit()

# 对文本对象封装
class SearchPage(BasePage):
    url = 'http://www.baidu.com'# 定位百度输入框
    search_input = (By.NAME, 'wd')# 定位百度一下按钮
    search_button = (By.ID, 'su')# 封装实现业务流程的函数
    def search(self, txt):
        self.visit(self.url)
        self.wait(2)
        self.input(self.search_input, txt)