#coding=utf-8
from selenium import webdriver
import time
#类使用
class SeleniumDriver:
def int(self,browser):
self.driver = self.open_browser(browser)
def open_browser(self,browser):
try:
if browser == “chrome”:
driver = webdriver.Chrome()
elif browser == “firefox”:
driver = webdriver.firefox()
elif browser == “ie”:
driver = webdriver.Ie()
else:
driver = webdriver.Edge()
time.sleep(5)
return driver
except:
print(“打开浏览器失败”)
return None

def get_url(self,url):
    if self.driver !=None:
        if 'http://' in url:
           self.driver.get(url)
           time.sleep(5)
        else:
           print("你的url有问题")
    else:
        print("case失败")

def handle_windows(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("你输入的window页面操作有问题")
        time.sleep(5)
        self.driver.quit()

selenium_driver = SeleniumDriver(‘chrome’)
#selenium_driver.get_url(‘http://www.baidu.com’)
selenium_driver.handle_windows(‘max’)