selenium contextmanager

sselenium lifehack

@contextmanager
def selen(driver_patch: str, profile_patch: str, config='windows') -> webdriver.Chrome:
    """
    Самый удобный метод использования веб драйвера =)
    :param driver_patch: str: patch to chromedriver(exe)
    :param profile_patch: str: the path to the folder storing cookies
    :param config: str: operating system
    :return: webdriver.Chrome
    """

    def windows_conf():
        options = webdriver.ChromeOptions()
        options.add_argument(f'user-data-dir={profile_patch}')
        driver = webdriver.Chrome(executable_path=driver_patch, chrome_options=options)
        return driver

    def linux_conf():
        options = webdriver.ChromeOptions()
        arguments = ('headless', '--no-sandbox', '--disable-dev-shm-usage', f'user-data-dir={profile_patch}')
        for option in arguments:
            options.add_argument(option)

        driver = webdriver.Chrome(executable_path=driver_patch, chrome_options=options)
        return driver

    configs = {'windows': windows_conf, 'linux': linux_conf}
    web_driver = configs.get(config)()

    try:
        yield web_driver
        web_driver.close()
        time.sleep(1)
        web_driver.quit()

    except Exception as error:
        web_driver.close()
        time.sleep(1)
        web_driver.quit()
        raise Exception(error)