1757e5ed2f
- Introduced `BrowserManager` class in `browser_utils.py` for streamlined browser initialization and configuration. - Added `EmailVerificationHandler` class in `get_veri_code.py` to handle email verification code retrieval. - Updated `cursor_pro_keep_alive.py` to utilize the new browser and email handling classes, improving code organization and readability. - Modified `config.ini` to update the email address for account configuration. - Removed obsolete `log.txt` file to clean up the project structure. - Added unit tests for email verification functionality in `test/get_veri_code_test.py`.
89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
from DrissionPage.common import Keys
|
|
import time
|
|
import re
|
|
import logging
|
|
|
|
|
|
class EmailVerificationHandler:
|
|
def __init__(self, browser, mail_url="https://tempmail.plus"):
|
|
self.browser = browser
|
|
self.mail_url = mail_url
|
|
|
|
def get_verification_code(self, email):
|
|
"""获取邮箱验证码"""
|
|
username = email.split("@")[0]
|
|
code = None
|
|
|
|
try:
|
|
# 打开新标签页访问临时邮箱
|
|
tab_mail = self.browser.new_tab(self.mail_url)
|
|
self.browser.activate_tab(tab_mail)
|
|
logging.info("打开邮箱页面")
|
|
|
|
# 输入用户名
|
|
self._input_username(tab_mail, username)
|
|
|
|
# 等待并获取最新邮件
|
|
code = self._get_latest_mail_code(tab_mail)
|
|
|
|
# 清理邮件
|
|
self._cleanup_mail(tab_mail)
|
|
|
|
# 关闭标签页
|
|
tab_mail.close()
|
|
|
|
except Exception as e:
|
|
logging.error(f"获取验证码失败: {str(e)}")
|
|
|
|
return code
|
|
|
|
def _input_username(self, tab, username):
|
|
"""输入用户名"""
|
|
while True:
|
|
if tab.ele("@id=pre_button"):
|
|
tab.actions.click("@id=pre_button")
|
|
time.sleep(0.5)
|
|
tab.run_js('document.getElementById("pre_button").value = ""')
|
|
time.sleep(0.5)
|
|
tab.actions.input(username).key_down(Keys.ENTER).key_up(Keys.ENTER)
|
|
break
|
|
time.sleep(1)
|
|
|
|
def _get_latest_mail_code(self, tab):
|
|
"""获取最新邮件中的验证码"""
|
|
code = None
|
|
while True:
|
|
new_mail = tab.ele("@class=mail")
|
|
if new_mail:
|
|
if new_mail.text:
|
|
logging.info(f"最新的邮件：{new_mail.text}")
|
|
tab.actions.click("@class=mail")
|
|
break
|
|
else:
|
|
logging.info(str(new_mail))
|
|
break
|
|
time.sleep(1)
|
|
|
|
if tab.ele("@class=overflow-auto mb-20"):
|
|
email_content = tab.ele("@class=overflow-auto mb-20").text
|
|
verification_code = re.search(
|
|
r"verification code is (\d{6})", email_content
|
|
)
|
|
if verification_code:
|
|
code = verification_code.group(1)
|
|
logging.info(f"验证码：{code}")
|
|
else:
|
|
logging.warning("未找到验证码")
|
|
|
|
return code
|
|
|
|
def _cleanup_mail(self, tab):
|
|
"""清理邮件"""
|
|
if tab.ele("@id=delete_mail"):
|
|
tab.actions.click("@id=delete_mail")
|
|
time.sleep(1)
|
|
|
|
if tab.ele("@id=confirm_mail"):
|
|
tab.actions.click("@id=confirm_mail")
|
|
logging.info("删除邮件")
|