diff --git a/parser/utils.py b/parser/utils.py index 6733995..4c829dd 100644 --- a/parser/utils.py +++ b/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'] diff --git a/start.sh b/start.sh index dabf310..f0d9216 100644 --- a/start.sh +++ b/start.sh @@ -1,6 +1,6 @@ docker build -t replace-bot . docker run --net=br0 \ --name=replacebot \ ---mount type=bind,source="$(pwd)",target=/app $@ \ +-v "$(pwd)":/app $@ \ --ip=10.0.0.3 --restart=unless-stopped \ replace-bot