fix(google): revise regex for better handling of thumbnail ids

kitUIN · Sep 12, 2024 · efbab28 · efbab28
1 parent f405812
commit efbab28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PicImageSearch/model/google.py b/PicImageSearch/model/google.py
@@ -89,7 +89,7 @@ def create_thumbnail_dict(script_list: list[PyQuery]) -> dict[str, str]:
         """
         thumbnail_dict = {}
         base_64_regex = compile(r"data:image/(?:jpeg|jpg|png|gif);base64,[^'\"]+")
-        id_regex = compile(r"dimg_\d+")
+        id_regex = compile(r"dimg_[^']+")
 
         for script in script_list:
             base_64_match = base_64_regex.findall(script.text())