feat: add clipping option for screenshots and implement element capture

Author: thalissonvs

pydoll/commands/page.py

@@ -50,7 +50,9 @@ def set_download_path(cls, path: str) -> dict:
         return command
 
     @classmethod
-    def screenshot(cls, format: str = 'png', quality: int = 100) -> dict:
+    def screenshot(
+        cls, format: str = 'png', quality: int = 100, clip: dict = None
+    ) -> dict:
         """
         Generates the command to capture a screenshot of the current page.
 
@@ -69,6 +71,8 @@ def screenshot(cls, format: str = 'png', quality: int = 100) -> dict:
         command = cls.SCREENSHOT_TEMPLATE.copy()
         command['params']['format'] = format
         command['params']['quality'] = quality
+        if clip:
+            command['params']['clip'] = clip
         return command
 
     @classmethod

pydoll/element.py

@@ -1,15 +1,18 @@
 import asyncio
 import json
 
+import aiofiles
 from bs4 import BeautifulSoup
 
 from pydoll import exceptions
 from pydoll.commands.dom import DomCommands
 from pydoll.commands.input import InputCommands
+from pydoll.commands.page import PageCommands
 from pydoll.commands.runtime import RuntimeCommands
 from pydoll.connection import ConnectionHandler
 from pydoll.constants import Scripts
 from pydoll.mixins.find_elements import FindElementsMixin
+from pydoll.utils import decode_image_to_bytes
 
 
 class WebElement(FindElementsMixin):
@@ -125,7 +128,7 @@ async def get_bounds_using_js(self) -> list:
         response = await self._execute_script(
             Scripts.BOUNDS, return_by_value=True
         )
-        return response['result']['result']['value']
+        return json.loads(response['result']['result']['value'])
 
     async def _execute_script(
         self, script: str, return_by_value: bool = False
@@ -170,6 +173,30 @@ async def _is_element_on_top(self):
         )
         return result['result']['result']['value']
 
+    async def get_screenshot(self, path: str):
+        """
+        Takes a screenshot of the element.
+
+        Args:
+            path (str): The path where the screenshot will be saved.
+        """
+        bounds = await self.get_bounds_using_js()
+        clip = {
+            'x': bounds['x'],
+            'y': bounds['y'],
+            'width': bounds['width'],
+            'height': bounds['height'],
+            'scale': 1,
+        }
+        screenshot = await self._connection_handler.execute_command(
+            PageCommands.screenshot(
+                format='png', clip=clip
+            )
+        )
+        async with aiofiles.open(path, 'wb') as file:
+            image_bytes = decode_image_to_bytes(screenshot['result']['data'])
+            await file.write(image_bytes)
+
     async def get_element_text(self) -> str:
         """
         Retrieves the text of the element.
@@ -241,7 +268,6 @@ async def click(self, x_offset: int = 0, y_offset: int = 0):
             )
         except KeyError:
             element_bounds = await self.get_bounds_using_js()
-            element_bounds = json.loads(element_bounds)
             position_to_click = (
                 element_bounds['x'] + element_bounds['width'] / 2,
                 element_bounds['y'] + element_bounds['height'] / 2,