selenium通过单例封装driver

from selenium import webdriver


class GetDriver:
    driver = None
    url = "https://www.baidu.com/"
    @classmethod
    def get_driver(cls):
        if cls.driver is None:
            cls.driver = webdriver.Chrome()
            cls.driver.maximize_window()
            cls.driver.get(cls.url)
        return cls.driver

    @classmethod
    def quit_driver(cls):
        if cls.driver:
            cls.driver.quit()
            cls.driver = None

if __name__ == '__main__':
    GetDriver().get_driver()
    GetDriver().quit_driver()