tema/replace-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
tema 2022-11-14 17:53:04 +02:00
parent 66283e0730
commit 7b5af42082
Signed by: tema
GPG Key ID: 21FDB6D162488F6F
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
parser/utils.py
View File

@ -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']