from selenium.webdriver import Chrome, ChromeOptions
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as ec
from loguru import logger
from collections import namedtuple
Locator = namedtuple('Locator', ('type', 'value'))
class Browser:
def __init__(self, driver=None, config=None):
self.driver = driver or Chrome()
self.config = config or {}
self.waiter = WebDriverWait(self.driver, timeout=self.config.get('timeout', 20))
# 页面级操作
def get(self, url):
logger.info(f'打开网页 => {url}')
self.driver.maximize_window()
self.driver.set_page_load_timeout(self.config.get('page_max_load_time', 20))
self.driver.get(url)
def f5(self):
self.driver.refresh()
def close(self):
self.driver.close()
@property
def html(self):
return self.driver.page_source
# js操作
def js(self, script: str, *args):
self.driver.execute_script(script, *args)
# 元素级操作
def _accord_wait(self, scanner):
"""显示等待"""
return self.waiter.until(scanner)
def find_element(self, locator: Locator):
return self._accord_wait(ec.presence_of_element_located(locator))
def find_elements(self, locator):
return self._accord_wait(ec.presence_of_all_elements_located(locator))
def input(self, locator: Locator, text):
self.find_element(locator).send_keys(text)
def click(self, locator: Locator):
self.find_element(locator).click()
def text(self, locator: Locator):
return self.find_element(locator).text
def test():
options = ChromeOptions()
options.add_argument('--headless') # *面
options.add_argument('--no-sandbox') # root权限
options.add_argument('--disable-dev-shm-usage') # 禁止共享内存
driver = Chrome(options=options)
browser = Browser(driver=driver)
browser.get('http://www.baidu.com')