mirror of
https://github.com/yeongpin/cursor-free-vip.git
synced 2025-08-02 12:47:34 +08:00
db3a2032dc
- Updated `config.py` to include a comprehensive browser configuration section, allowing for dynamic retrieval of browser paths and drivers for Chrome, Edge, Firefox, and Brave. - Refactored `new_signup.py`, `new_tempemail.py`, and `oauth_auth.py` to utilize the new browser configuration, improving flexibility and maintainability. - Removed deprecated `get_default_chrome_path` function and replaced it with a more generalized `get_default_browser_path` function in `utils.py`. - Updated localization files to include new strings related to browser path validation and selection.
393 lines
19 KiB
Python
393 lines
19 KiB
Python
from DrissionPage import ChromiumPage, ChromiumOptions
|
||
import time
|
||
import os
|
||
import sys
|
||
from colorama import Fore, Style, init
|
||
import requests
|
||
import random
|
||
import string
|
||
from config import get_config
|
||
from utils import get_random_wait_time, get_default_browser_path as utils_get_default_browser_path
|
||
|
||
# Initialize colorama
|
||
init()
|
||
|
||
class NewTempEmail:
|
||
def __init__(self, translator=None):
|
||
self.translator = translator
|
||
self.page = None
|
||
self.setup_browser()
|
||
|
||
def get_blocked_domains(self):
|
||
"""Get blocked domains list"""
|
||
try:
|
||
block_url = "https://raw.githubusercontent.com/yeongpin/cursor-free-vip/main/block_domain.txt"
|
||
response = requests.get(block_url, timeout=5)
|
||
if response.status_code == 200:
|
||
# Split text and remove empty lines
|
||
domains = [line.strip() for line in response.text.split('\n') if line.strip()]
|
||
if self.translator:
|
||
print(f"{Fore.CYAN}ℹ️ {self.translator.get('email.blocked_domains_loaded', count=len(domains))}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.CYAN}ℹ️ 已加载 {len(domains)} 个被屏蔽的域名{Style.RESET_ALL}")
|
||
return domains
|
||
return self._load_local_blocked_domains()
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.blocked_domains_error', error=str(e))}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 获取被屏蔽域名列表失败: {str(e)}{Style.RESET_ALL}")
|
||
return self._load_local_blocked_domains()
|
||
|
||
def _load_local_blocked_domains(self):
|
||
"""Load blocked domains from local file as fallback"""
|
||
try:
|
||
local_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "block_domain.txt")
|
||
if os.path.exists(local_path):
|
||
with open(local_path, 'r', encoding='utf-8') as f:
|
||
domains = [line.strip() for line in f.readlines() if line.strip()]
|
||
if self.translator:
|
||
print(f"{Fore.CYAN}ℹ️ {self.translator.get('email.local_blocked_domains_loaded', count=len(domains))}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.CYAN}ℹ️ 已从本地加载 {len(domains)} 个被屏蔽的域名{Style.RESET_ALL}")
|
||
return domains
|
||
else:
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.local_blocked_domains_not_found')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 本地被屏蔽域名文件不存在{Style.RESET_ALL}")
|
||
return []
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.local_blocked_domains_error', error=str(e))}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 读取本地被屏蔽域名文件失败: {str(e)}{Style.RESET_ALL}")
|
||
return []
|
||
|
||
def exclude_blocked_domains(self, domains):
|
||
"""Exclude blocked domains"""
|
||
if not self.blocked_domains:
|
||
return domains
|
||
|
||
filtered_domains = []
|
||
for domain in domains:
|
||
if domain['domain'] not in self.blocked_domains:
|
||
filtered_domains.append(domain)
|
||
|
||
excluded_count = len(domains) - len(filtered_domains)
|
||
if excluded_count > 0:
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.domains_excluded', domains=excluded_count)}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 已排除 {excluded_count} 个被屏蔽的域名{Style.RESET_ALL}")
|
||
|
||
return filtered_domains
|
||
|
||
|
||
def get_extension_block(self):
|
||
"""获取插件路径"""
|
||
root_dir = os.getcwd()
|
||
extension_path = os.path.join(root_dir, "PBlock")
|
||
|
||
if hasattr(sys, "_MEIPASS"):
|
||
extension_path = os.path.join(sys._MEIPASS, "PBlock")
|
||
|
||
if not os.path.exists(extension_path):
|
||
raise FileNotFoundError(f"插件不存在: {extension_path}")
|
||
|
||
return extension_path
|
||
|
||
def setup_browser(self):
|
||
"""设置浏览器"""
|
||
try:
|
||
if self.translator:
|
||
print(f"{Fore.CYAN}ℹ️ {self.translator.get('email.starting_browser')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.CYAN}ℹ️ 正在启动浏览器...{Style.RESET_ALL}")
|
||
|
||
# 获取配置
|
||
config = get_config(self.translator)
|
||
|
||
# 获取浏览器类型和路径
|
||
browser_type = config.get('Browser', 'default_browser', fallback='chrome')
|
||
browser_path = config.get('Browser', f'{browser_type}_path', fallback=utils_get_default_browser_path(browser_type))
|
||
|
||
if not browser_path or not os.path.exists(browser_path):
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.browser_path_invalid', browser=browser_type) if self.translator else f'{browser_type} 路径无效，使用默认路径'}{Style.RESET_ALL}")
|
||
browser_path = utils_get_default_browser_path(browser_type)
|
||
|
||
# 为了向后兼容，也检查 Chrome 路径
|
||
if browser_type == 'chrome':
|
||
chrome_path = config.get('Chrome', 'chromepath', fallback=None)
|
||
if chrome_path and os.path.exists(chrome_path):
|
||
browser_path = chrome_path
|
||
|
||
# 创建浏览器选项
|
||
co = ChromiumOptions()
|
||
|
||
# 设置浏览器路径
|
||
co.set_browser_path(browser_path)
|
||
|
||
# 记录浏览器信息
|
||
if self.translator:
|
||
print(f"{Fore.CYAN}🌐 {self.translator.get('email.using_browser', browser=browser_type, path=browser_path) if self.translator else f'使用 {browser_type} 浏览器: {browser_path}'}{Style.RESET_ALL}")
|
||
|
||
# Only use headless for non-OAuth operations
|
||
if not hasattr(self, 'auth_type') or self.auth_type != 'oauth':
|
||
co.set_argument("--headless=new")
|
||
|
||
if sys.platform == "linux":
|
||
# Check if DISPLAY is set when not in headless mode
|
||
if not co.arguments.get("--headless=new") and not os.environ.get('DISPLAY'):
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.no_display_found') if self.translator else 'No display found. Make sure X server is running.'}{Style.RESET_ALL}")
|
||
print(f"{Fore.YELLOW}ℹ️ {self.translator.get('email.try_export_display') if self.translator else 'Try: export DISPLAY=:0'}{Style.RESET_ALL}")
|
||
return False
|
||
|
||
co.set_argument("--no-sandbox")
|
||
co.set_argument("--disable-dev-shm-usage")
|
||
co.set_argument("--disable-gpu")
|
||
|
||
# If running as root, try to use actual user's browser profile
|
||
if os.geteuid() == 0:
|
||
sudo_user = os.environ.get('SUDO_USER')
|
||
if sudo_user:
|
||
actual_home = f"/home/{sudo_user}"
|
||
|
||
# 根据浏览器类型选择配置文件夹
|
||
profile_dirs = {
|
||
'chrome': os.path.join(actual_home, ".config", "google-chrome"),
|
||
'brave': os.path.join(actual_home, ".config", "BraveSoftware", "Brave-Browser"),
|
||
'edge': os.path.join(actual_home, ".config", "microsoft-edge"),
|
||
'firefox': os.path.join(actual_home, ".mozilla", "firefox")
|
||
}
|
||
|
||
user_data_dir = profile_dirs.get(browser_type, profile_dirs['chrome'])
|
||
|
||
if os.path.exists(user_data_dir):
|
||
print(f"{Fore.CYAN}ℹ️ {self.translator.get('email.using_browser_profile', browser=browser_type, user_data_dir=user_data_dir) if self.translator else f'Using {browser_type} profile from: {user_data_dir}'}{Style.RESET_ALL}")
|
||
co.set_argument(f"--user-data-dir={user_data_dir}")
|
||
|
||
co.auto_port() # 自动设置端口
|
||
|
||
# 根据浏览器类型设置扩展参数
|
||
extension_args = {
|
||
'chrome': "--allow-extensions-in-incognito",
|
||
'brave': "--allow-extensions-in-brave-incognito", # Brave 可能使用不同的参数
|
||
'edge': "--allow-extensions-in-incognito",
|
||
'firefox': None # Firefox 可能使用不同的方式加载扩展
|
||
}
|
||
|
||
extension_arg = extension_args.get(browser_type, "--allow-extensions-in-incognito")
|
||
|
||
# 加载 uBlock 插件
|
||
try:
|
||
extension_path = self.get_extension_block()
|
||
if extension_arg: # 如果有扩展参数
|
||
co.set_argument(extension_arg)
|
||
co.add_extension(extension_path)
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.extension_load_error')}: {str(e)}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 加载插件失败: {str(e)}{Style.RESET_ALL}")
|
||
|
||
self.page = ChromiumPage(co)
|
||
return True
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.browser_start_error')}: {str(e)}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}❌ 启动浏览器失败: {str(e)}{Style.RESET_ALL}")
|
||
|
||
if sys.platform == "linux":
|
||
browser_install_suggestions = {
|
||
'chrome': "sudo apt install chromium-browser 或 sudo apt install google-chrome-stable",
|
||
'brave': "sudo apt install brave-browser",
|
||
'edge': "sudo apt install microsoft-edge-stable",
|
||
'firefox': "sudo apt install firefox"
|
||
}
|
||
|
||
suggestion = browser_install_suggestions.get(browser_type, browser_install_suggestions['chrome'])
|
||
|
||
print(f"{Fore.YELLOW}ℹ️ {self.translator.get('email.make_sure_browser_is_properly_installed', browser=browser_type) if self.translator else f'Make sure {browser_type} is properly installed'}{Style.RESET_ALL}")
|
||
print(f"{Fore.YELLOW}ℹ️ {self.translator.get('email.try_install_browser') if self.translator else f'Try: {suggestion}'}{Style.RESET_ALL}")
|
||
return False
|
||
|
||
def create_email(self):
|
||
"""create temporary email"""
|
||
try:
|
||
if self.translator:
|
||
print(f"{Fore.CYAN}ℹ️ {self.translator.get('email.visiting_site')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.CYAN}ℹ️ 正在访问 smailpro.com...{Style.RESET_ALL}")
|
||
|
||
# load blocked domains list
|
||
self.blocked_domains = self.get_blocked_domains()
|
||
|
||
# visit website
|
||
self.page.get("https://smailpro.com/")
|
||
time.sleep(2)
|
||
|
||
# click create email button
|
||
create_button = self.page.ele('xpath://button[@title="Create temporary email"]')
|
||
if create_button:
|
||
create_button.click()
|
||
time.sleep(1)
|
||
|
||
# click Create button in popup
|
||
modal_create_button = self.page.ele('xpath://button[contains(text(), "Create")]')
|
||
if modal_create_button:
|
||
modal_create_button.click()
|
||
time.sleep(2)
|
||
|
||
# get email address - modify selector
|
||
email_div = self.page.ele('xpath://div[@class="text-base sm:text-lg md:text-xl text-gray-700"]')
|
||
if email_div:
|
||
email = email_div.text.strip()
|
||
if '@' in email: # check if it's a valid email address
|
||
# check if domain is blocked
|
||
domain = email.split('@')[1]
|
||
if self.blocked_domains and domain in self.blocked_domains:
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.domain_blocked')}: {domain}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 域名已被屏蔽: {domain}，尝试重新创建邮箱{Style.RESET_ALL}")
|
||
# create email again
|
||
return self.create_email()
|
||
|
||
if self.translator:
|
||
print(f"{Fore.GREEN}✅ {self.translator.get('email.create_success')}: {email}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.GREEN}✅ 创建邮箱成功: {email}{Style.RESET_ALL}")
|
||
return email
|
||
if self.translator:
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.create_failed')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}❌ 创建邮箱失败{Style.RESET_ALL}")
|
||
return None
|
||
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.create_error')}: {str(e)}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}❌ 创建邮箱出错: {str(e)}{Style.RESET_ALL}")
|
||
return None
|
||
|
||
def close(self):
|
||
"""close browser"""
|
||
if self.page:
|
||
self.page.quit()
|
||
|
||
def refresh_inbox(self):
|
||
"""refresh inbox"""
|
||
try:
|
||
if self.translator:
|
||
print(f"{Fore.CYAN}🔄 {self.translator.get('email.refreshing')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.CYAN}🔄 正在刷新邮箱...{Style.RESET_ALL}")
|
||
|
||
# click refresh button
|
||
refresh_button = self.page.ele('xpath://button[@id="refresh"]')
|
||
if refresh_button:
|
||
refresh_button.click()
|
||
time.sleep(2) # wait for refresh to complete
|
||
if self.translator:
|
||
print(f"{Fore.GREEN}✅ {self.translator.get('email.refresh_success')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.GREEN}✅ 邮箱刷新成功{Style.RESET_ALL}")
|
||
return True
|
||
|
||
if self.translator:
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.refresh_button_not_found')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}❌ 未找到刷新按钮{Style.RESET_ALL}")
|
||
return False
|
||
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.refresh_error')}: {str(e)}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}❌ 刷新邮箱出错: {str(e)}{Style.RESET_ALL}")
|
||
return False
|
||
|
||
def check_for_cursor_email(self):
|
||
"""检查是否有 Cursor 的验证邮件"""
|
||
try:
|
||
# find verification email - use more accurate selector
|
||
email_div = self.page.ele('xpath://div[contains(@class, "p-2") and contains(@class, "cursor-pointer") and contains(@class, "bg-white") and contains(@class, "shadow") and .//b[text()="no-reply@cursor.sh"] and .//span[text()="Verify your email address"]]')
|
||
if email_div:
|
||
if self.translator:
|
||
print(f"{Fore.GREEN}✅ {self.translator.get('email.verification_found')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.GREEN}✅ 找到验证邮件{Style.RESET_ALL}")
|
||
# use JavaScript to click element
|
||
self.page.run_js('arguments[0].click()', email_div)
|
||
time.sleep(2) # wait for email content to load
|
||
return True
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.verification_not_found')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 未找到验证邮件{Style.RESET_ALL}")
|
||
return False
|
||
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.verification_error')}: {str(e)}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}❌ 检查验证邮件出错: {str(e)}{Style.RESET_ALL}")
|
||
return False
|
||
|
||
def get_verification_code(self):
|
||
"""获取验证码"""
|
||
try:
|
||
# find verification code element
|
||
code_element = self.page.ele('xpath://td//div[contains(@style, "font-size:28px") and contains(@style, "letter-spacing:2px")]')
|
||
if code_element:
|
||
code = code_element.text.strip()
|
||
if code.isdigit() and len(code) == 6:
|
||
if self.translator:
|
||
print(f"{Fore.GREEN}✅ {self.translator.get('email.verification_code_found')}: {code}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.GREEN}✅ 获取验证码成功: {code}{Style.RESET_ALL}")
|
||
return code
|
||
if self.translator:
|
||
print(f"{Fore.YELLOW}⚠️ {self.translator.get('email.verification_code_not_found')}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}⚠️ 未找到有效的验证码{Style.RESET_ALL}")
|
||
return None
|
||
|
||
except Exception as e:
|
||
if self.translator:
|
||
print(f"{Fore.RED}❌ {self.translator.get('email.verification_code_error')}: {str(e)}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}❌ 获取验证码出错: {str(e)}{Style.RESET_ALL}")
|
||
return None
|
||
|
||
def main(translator=None):
|
||
temp_email = NewTempEmail(translator)
|
||
|
||
try:
|
||
email = temp_email.create_email()
|
||
if email:
|
||
if translator:
|
||
print(f"\n{Fore.CYAN}📧 {translator.get('email.address')}: {email}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"\n{Fore.CYAN}📧 临时邮箱地址: {email}{Style.RESET_ALL}")
|
||
|
||
# test refresh function
|
||
while True:
|
||
if translator:
|
||
choice = input(f"\n{translator.get('email.refresh_prompt')}: ").lower()
|
||
else:
|
||
choice = input("\n按 R 刷新邮箱，按 Q 退出: ").lower()
|
||
if choice == 'r':
|
||
temp_email.refresh_inbox()
|
||
elif choice == 'q':
|
||
break
|
||
|
||
finally:
|
||
temp_email.close()
|
||
|
||
if __name__ == "__main__":
|
||
main() |