parser/utils.py

@@ -1,4 +1,5 @@
 from bs4 import BeautifulSoup
+from typing import Any
 
 def table_parser(soup: BeautifulSoup, output):
     #Date parser
@@ -24,8 +25,10 @@ def table_parser(soup: BeautifulSoup, output):
 
 
 def image_parser(soup: BeautifulSoup):
+    image: Any
+    extension = ('png', 'jpg')
     main = soup.find("main")
-    image = main.select('img[src$=".png"]')
-    output = image[0]['src']
-
-    return output
+    for ext in extension:
+        image = main.select(f'img[src$=".{ext}"]')
+        if image:
+            return image[0]['src']