from selenium import webdriver
import time
class SeleniumDriver:
def init(self,browser):
self.driver = self.open_browser(browser)

def open_browser(self,browser):
    if browser == "chrome":      #如果是谷歌浏览器
        driver = webdriver.Chrome()
    elif browser == "firefox":   #如果是火狐浏览器
        driver = webdriver.Firefox()
    elif browser == "ie":         #如果是火狐浏览器
        driver = webdriver.Ie()
    else:                          #如果以上都是，那么就是Edge浏览器
        driver = webdriver.Edge()
    time.sleep(1)                  #等待1秒


def get_url(self,url):
    if self.driver != None:   #如果浏览器不等于空的时候，再运行以下代码。
        self.driver.maximize_window()
        if "http://" in url:
            self.driver.get(url)
        else:
            print("你的URL有问题")
    else:
        print("case失败")
    self.driver.quit()

def handle_window(self,*args):
    value = len(args)
    if value == 1:
        if args[0] == "max":
            self.driver.maximize_window()
        elif args[0] == "min":
            self.driver.minimize_window()
        elif args[0] == "back":
            self.driver.back()
        elif args[0] == "go":
            self.driver.forward()
        else:
            self.driver.refresh()
    elif value == 2:
        self.driver.set_window_size(args[0],args[1])
    else:
        print("你传递的参数有问题")
    time.sleep(5)
    self.driver.quit()

selfnium_driver = SeleniumDriver(“chrome”)
selfnium_driver.handle_window(“max”)