diff --git a/mail_server.py b/mail_server.py index 200cd3f..d224a53 100644 --- a/mail_server.py +++ b/mail_server.py @@ -2,7 +2,7 @@ from aiosmtpd.controller import Controller import quopri from loguru import logger import re -from browser import PrimaryLocationClicker, TravelVerificationClicker +from browser import PrimaryLocationClicker, TravelVerificationRetriever import asyncio from line_notifier import LineNotifier @@ -35,7 +35,7 @@ class NetflixHandler: except Exception as e: logger.error(e) elif urls := VERIFY_URL_PATTERN.findall(data): - with TravelVerificationClicker() as browser: + with TravelVerificationRetriever() as browser: url = urls[0] logger.info(f'Found Travel Verification link: {url}') code = browser.get_code(url)