def save_element_screenshot(element: WebElement, name: str, scroll=1): """ Save screenshot of WebElement and scroll page """ element.screenshot(f'{name}.png') if (scroll): driver = element._parent size = element.size h_elem = size['height'] script = f'window.scrollTo(0, window.scrollY + {h_elem + 10})' driver.execute_script(script)
def save_screenshot(self, element: WebElement, filename): screenshot = element.screenshot(filename) return screenshot