This commit is contained in:
YuanYu 2024-03-31 19:56:38 +08:00
parent bfcbd72a10
commit e963b33065

View File

@ -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)