diff --git a/main.py b/main.py index 20ce441..d33e520 100644 --- a/main.py +++ b/main.py @@ -1,5 +1,5 @@ import time - +from retrying import retry from schedule import every, repeat, run_pending from requests_html import HTML, HTMLSession @@ -8,11 +8,16 @@ from lottery import Lottery session = HTMLSession() -def get_data_job(lottery_type='pls'): + +@retry(stop_max_attempt_number=5) +def get_data_job(lottery_type='pls', issue=None): """ :param lottery_type """ - url = f"https://kaijiang.500.com/{lottery_type}.shtml" + if issue is None: + url = f"https://kaijiang.500.com/{lottery_type}.shtml" + else: + url = f"http://kaijiang.500.com/shtml/{lottery_type}/{issue}.shtml" r = session.get(url) if lottery_type == 'pls': issue_elem = r.html.search('排列3 第 {}期')