Fix none reference img

Signed-off-by: Shawn Hsu <ambersun1019.shawn@gmail.com>

Author: ambersun1234

src/core/image_checker.py

@@ -13,14 +13,16 @@ def check_page(self, page_url: str) -> (list, bool):
 
         page_fail_results = []
         try:
+            current_image = None
+
             page = self.context.new_page()
             cdp_session = page.context.new_cdp_session(page)
             cdp_session.send("Network.setCacheDisabled", {"cacheDisabled": True})
 
             try:
                 response = page.goto(page_url, wait_until="load", timeout=90000)
             except Exception:
-                logger.debug(f"Timeout loading {page_url}, skipping")
+                logger.warning(f"Timeout loading {page_url}, skipping")
                 raise
 
             if not response or response.status != 200:
@@ -70,6 +72,7 @@ def check_page(self, page_url: str) -> (list, bool):
 
             images = page.locator("img").all()
             for index, img in enumerate(images):
+                current_image = img
                 width, height = self._check_image(img, page_url)
                 logger.debug(
                     f"Checking image",
@@ -92,7 +95,7 @@ def check_page(self, page_url: str) -> (list, bool):
             page_fail_results.append(
                 {
                     "page_url": page_url,
-                    "image_url": img.get_attribute("src"),
+                    "image_url": "unknown" if current_image is None else current_image.get_attribute("src"),
                 }
             )
         finally: