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
