refact: fix on changes with ruff and applied changes to tests

Author: ciceroadf

pydoll/browser/page.py

@@ -205,7 +205,9 @@ async def get_screenshot(self, path: str):
         if fmt not in ['jpeg', 'jpg', 'png']:
             raise InvalidFileExtension(f'{fmt} extension is not supported.')
 
-        response = await self._execute_command(PageCommands.screenshot(fmt=fmt))
+        response = await self._execute_command(
+            PageCommands.screenshot(fmt=fmt)
+        )
         screenshot_b64 = response['result']['data'].encode('utf-8')
         screenshot_bytes = decode_image_to_bytes(screenshot_b64)
         async with aiofiles.open(path, 'wb') as file:

pydoll/element.py

@@ -189,7 +189,7 @@ async def get_screenshot(self, path: str):
             'scale': 1,
         }
         screenshot = await self._connection_handler.execute_command(
-            PageCommands.screenshot(format='jpeg', clip=clip)
+            PageCommands.screenshot(fmt='jpeg', clip=clip)
         )
         async with aiofiles.open(path, 'wb') as file:
             image_bytes = decode_image_to_bytes(screenshot['result']['data'])

pydoll/exceptions.py

@@ -81,8 +81,9 @@ class ElementNotInteractable(Exception):
     def __str__(self):
         return self.message
 
+
 class InvalidFileExtension(Exception):
     message = 'The file extension provided is not supported'
 
     def __str__(self):
-        return self.message
+        return self.message

tests/test_page_commands.py

@@ -33,7 +33,7 @@ def test_screenshot_jpeg():
         },
     }
     assert (
-        PageCommands.screenshot(format='jpeg', quality=80) == expected_command
+        PageCommands.screenshot(fmt='jpeg', quality=80) == expected_command
     )
 
 
@@ -45,7 +45,7 @@ def test_screenshot_png():
             'quality': 100,
         },
     }
-    assert PageCommands.screenshot(format='png') == expected_command
+    assert PageCommands.screenshot(fmt='png') == expected_command
 
 
 def test_screenshot_with_clip():