selenium打开浏览器并操作窗口

selenium打开浏览器并操作窗口

from selenium import webdriver
import time

class SeleniumDriver:
    def __init__(self,browser):
        self.driver = self.open_browser('chrome')
    def open_browser(self, browser):
        try:
            if browser == 'chrome':
                self.driver = webdriver.Chrome()
            elif browser == 'firefox':
                self.driver = webdriver.Firefox()
            elif browser == 'ie':
                self.driver = webdriver.Ie()
            else:
                self.driver = webdriver.Edge()
            time.sleep(5)
            driver = self.driver
            return driver
        except:
            print('Browser Error')
            return None


    def get_url(self, url):
        if self.driver !=None:
            if 'http://' in url:
                self.driver.get(url)
            else:
                print('Url Error')
        else:
            print('case failed')

    def handle_browser(self, *args):
        if len(args) == 1:
            value = args[0]
            if value == 'max':
                self.driver.maximize_window()
            elif value == 'min':
                self.driver.minimize_window()
            elif value == 'back':
                self.driver.back()
            elif value == 'go':
                self.driver.forward()
            else:
                self.driver.refresh()
        elif len(args) == 2:
            self.driver.set_window_size(args[0], args[1])
        else:
            print('Parameter Error')
        time.sleep(2)
        self.driver.quit()

seleniumdriver = SeleniumDriver('chrome')
seleniumdriver.handle_browser('max')