screenshot.py

from selenium import webdriver
from selenium.webdriver import ChromeOptions
from webdriver_manager.chrome import ChromeDriverManager
import time

options = ChromeOptions()
options.add_argument("--headless=new")
ChromeDriverManager().install()


def get_screen_shot(url: str, load_time: int, path: str):
    driver = webdriver.Chrome(options=options)
    driver.get(url)
    time.sleep(load_time)
    driver.save_screenshot(path)