web自动化关于浏览器的操作

web自动化中关于浏览器的操作简单整理了一下

from selenium import webdriver


class MyBassClass:
    def __init__(self):
        driver = webdriver.Chrome()
        self.driver = driver
        self.driver.maximize_window()

    def open_window(self, url):
        """
        打开链接
        :param url: 资源定位符
        """
        self.driver.get(url)

    def maximize_window(self):
        """
        浏览器最大化
        """
        self.driver.maximize_window()

    def set_window_size(self):
        """
        设置浏览器尺寸大小
        """
        self.driver.set_window_size(1440, 900)

    def set_window_back(self):
        """
        浏览器后退
        """
        self.driver.back()

    def set_window_forward(self):
        """
        浏览器向前
        """
        self.driver.forward()

    def refresh_window(self):
        """
        浏览器刷新
        """
        self.driver.refresh()

    def close_window(self):
        """
        关闭当前浏览器窗口，或最后打开的窗口
        """
        self.driver.close()

    def quit_window(self):
        """
        退出浏览器
        """
        self.driver.quit()

    def screenshot_window(self, file):
        """
        屏幕截图
        :param file: 文件名称
        """
        self.driver.get_screenshot_as_file(file)