chore: 整理代码

This commit is contained in:
cheng zhen
2025-01-03 18:41:28 +08:00
parent 1fa59ba330
commit 86352bd2e2
5 changed files with 2 additions and 276 deletions

80
get_email_code.py Normal file
View File

@@ -0,0 +1,80 @@
from DrissionPage.common import Keys
import time
import re
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:
print("正在处理...")
# 打开新标签页访问临时邮箱
tab_mail = self.browser.new_tab(self.mail_url)
self.browser.activate_tab(tab_mail)
# 输入用户名
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:
print(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:
tab.actions.click("@class=mail")
break
else:
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)
print("马上就要成功了")
else:
print("执行失败")
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")