Screenshot

screenshot.py

from seleniumwire import webdriver

def take_image(self, link: str):
    options = webdriver.ChromeOptions()
    options.add_argument('--headless')
    options.add_argument('window-size=1280x720')
    options.add_argument('--no-sandbox')
    options.add_argument('--disable-dev-shm-usage')

    try:
        browser = webdriver.Chrome(
            options=options,
        )
        browser.get(link)
        result = browser.get_screenshot_as_png()
        browser.quit()

        return result
    except Exception as e:
        self.logger.error(f'{e}, link:{link}', exc_info=True)