Selenium二次封装（一）

from selenium.webdriver.support import expected_conditions as EC
from selenium import webdriver
import time
class SeleniumDriver:

    def __init__(self,browser):
    """判断打开浏览器"""                      
       self.browser = browser                   
       if self.browser == 'firefox':                 #通过browser来判断打开浏览器
            self.driver = webdriver.Firefox()
       elif self.browser == 'chrome':
            self.driver = webdriver.Chrome()
       elif self.browser == 'ie':
            self.driver = webdriver.Ie()
       else:
            self.driver = webdriver.Edge()
       time.sleep(2)
       #return None
    def get_url(self,url):
        if self.driver !=None:
            if 'http://' in url:                     #通过判断Url中是否包含有http://来判断打开网页正确性
                self.driver.get(url)
            else:
                print('你的url有问题！')
        else:
            print("case失败")

    def handle_windows(self,*args):  
    """判断浏览器操作"""        
        value = len(args)                 #获取args的长度
        if value == 1:
           if args[0] == 'max':
               self.driver.maximize_window()  #浏览器最大化
           elif args[0] == 'mix':
               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(2)
        self.driver.quit()

selenium_driver = SeleniumDriver('chrome')
#selenium_driver.open_browser()
selenium_driver.handle_windows('max')