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 import quopri
from loguru import logger from loguru import logger
import re import re
from browser import PrimaryLocationClicker, TravelVerificationClicker from browser import PrimaryLocationClicker, TravelVerificationRetriever
import asyncio import asyncio
from line_notifier import LineNotifier from line_notifier import LineNotifier
@ -35,7 +35,7 @@ class NetflixHandler:
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)
elif urls := VERIFY_URL_PATTERN.findall(data): elif urls := VERIFY_URL_PATTERN.findall(data):
with TravelVerificationClicker() as browser: with TravelVerificationRetriever() as browser:
url = urls[0] url = urls[0]
logger.info(f'Found Travel Verification link: {url}') logger.info(f'Found Travel Verification link: {url}')
code = browser.get_code(url) code = browser.get_code(url)