✨ feat(screenshot): Save screenshot to 800x600px webp (#25)

ecoindex_scraper/scrap.py

@@ -19,6 +19,7 @@
     ScreenShot,
     WindowSize,
 )
+from ecoindex_scraper.utils import convert_screenshot_to_webp
 
 
 class EcoindexScraper:
@@ -99,6 +100,7 @@ async def scrap_page(self) -> Tuple[PageMetrics, PageType]:
     async def generate_screenshot(self) -> None:
         if self.screenshot and self.screenshot.folder and self.screenshot.id:
             self.driver.save_screenshot(self.screenshot.__str__())
+            convert_screenshot_to_webp(self.screenshot)
 
     async def scroll_to_bottom(self) -> None:
         try:

ecoindex_scraper/utils.py

@@ -0,0 +1,18 @@
+from os import unlink
+
+from PIL import Image
+
+from ecoindex_scraper.models import ScreenShot
+
+
+def convert_screenshot_to_webp(screenshot: ScreenShot) -> None:
+    image = Image.open(rf"{screenshot.__str__()}")
+    width, height = image.size
+    ratio = 800 / height if width > height else 600 / width
+
+    image.convert("RGB").resize(size=(int(width * ratio), int(height * ratio))).save(
+        rf"{screenshot.folder}/{screenshot.id}.webp",
+        format="webp",
+    )
+
+    unlink(f"{screenshot.__str__()}")

pyproject.toml

@@ -15,6 +15,7 @@ include = [
 python = "^3.8"
 ecoindex = "^5.0.0"
 undetected-chromedriver = "^3.1.5"
+Pillow = "^9.2.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1"