Spaces:

dan92
/

notdiamond2api2

Paused

App Files Files Community

dan92 commited on Nov 26, 2024

Commit

858b443

verified ·

1 Parent(s): 199c4e9

Upload 7 files

Browse files

Files changed (3) hide show

Dockerfile +4 -3
app.py +102 -36
register_bot.py +303 -303

Dockerfile CHANGED Viewed

@@ -10,9 +10,10 @@ COPY requirements.txt .
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt gunicorn
-# 复制应用程序文件
 COPY app.py .
 COPY register_bot.py .
 # 设置环境变量
 ENV FLASK_APP=app.py
@@ -22,5 +23,5 @@ ENV PYTHONUNBUFFERED=1
 # 暴露端口
 EXPOSE 3000
-# 使用 gunicorn 作为生产级 WSGI 服务器
-CMD ["gunicorn", "--bind", "0.0.0.0:3000", "--workers", "4", "app:app"]

 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt gunicorn
+# 复制应用程序文件和模板
 COPY app.py .
 COPY register_bot.py .
+COPY templates templates/
 # 设置环境变量
 ENV FLASK_APP=app.py
 # 暴露端口
 EXPOSE 3000
+# 使用 gunicorn 作为生产级 WSGI 服务器，添加错误日志
+CMD ["gunicorn", "--bind", "0.0.0.0:3000", "--workers", "4", "--log-level", "debug", "--error-logfile", "-", "app:app"]

app.py CHANGED Viewed

@@ -19,6 +19,7 @@ import urllib3
 from cachetools import TTLCache
 import threading
 from datetime import datetime
 # 新增导入
 import register_bot
@@ -490,7 +491,7 @@ def get_auth_credentials():
             if data.get('status') == 'success' and data.get('content'):
                 content = data['content']
                 credentials = []
-                # 分割多个凭据（如果��的话）
                 for cred in content.split(';'):
                     if '|' in cred:
                         email, password = cred.strip().split('|')
@@ -555,44 +556,83 @@ def get_accounts_status():
 @app.route('/', methods=['GET'])
 def root():
-    accounts_status = get_accounts_status()
-    if request.headers.get('Accept') == 'application/json':
-        return get_json_status(accounts_status)
-    return render_template('monitor.html', **get_template_data(accounts_status))
 def get_template_data(accounts_status):
-    total_accounts = len(accounts_status)
-    valid_accounts = sum(1 for acc in accounts_status if acc["is_valid"])
-    accounts_data = []
-    total_requests = 0
-    for auth_manager in multi_auth_manager.auth_managers:
-        success_rate = 0
-        if auth_manager.total_requests > 0:
-            success_rate = (auth_manager.success_requests / auth_manager.total_requests) * 100
-        account_info = {
-            "email": auth_manager._email,
-            "is_valid": auth_manager.is_token_valid(),
-            "total_requests": auth_manager.total_requests,
-            "success_requests": auth_manager.success_requests,
-            "failed_requests": auth_manager.failed_requests,
-            "success_rate": success_rate,
-            "last_used_time": auth_manager.last_used_time.strftime('%m/%d/%Y, %I:%M:%S %p') if auth_manager.last_used_time else "从未使用"
         }
-        accounts_data.append(account_info)
-        total_requests += auth_manager.total_requests
-    return {
-        "total_accounts": total_accounts,
-        "valid_accounts": valid_accounts,
-        "total_requests": total_requests,
-        "accounts": accounts_data,
-        "last_update": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
-    }
 def get_json_status(accounts_status):
     template_data = get_template_data(accounts_status)
@@ -789,7 +829,7 @@ def make_request(payload, auth_manager, model_id):
                 if response.status_code == 403:  # Forbidden, 模型使用限制
                     logger.warning(f"Model {model_id} usage limit reached for account {auth_manager._email}")
-                    # 立即触发注册
                     if trigger_registration():
                         # 重试请求
                         return make_request(payload, None, model_id)
@@ -841,3 +881,29 @@ if __name__ == "__main__":
     port = int(os.environ.get("PORT", 3000))
     app.run(debug=False, host='0.0.0.0', port=port, threaded=True)

 from cachetools import TTLCache
 import threading
 from datetime import datetime
+from werkzeug.exceptions import HTTPException
 # 新增导入
 import register_bot
             if data.get('status') == 'success' and data.get('content'):
                 content = data['content']
                 credentials = []
+                # 分割多个凭据（如果的话）
                 for cred in content.split(';'):
                     if '|' in cred:
                         email, password = cred.strip().split('|')
 @app.route('/', methods=['GET'])
 def root():
+    try:
+        accounts_status = get_accounts_status()
+        if request.headers.get('Accept') == 'application/json':
+            return get_json_status(accounts_status)
+        template_data = get_template_data(accounts_status)
+        return render_template('monitor.html', **template_data)
+    except Exception as e:
+        logger.error(f"Error in root route: {str(e)}", exc_info=True)
+        if request.headers.get('Accept') == 'application/json':
+            return jsonify({
+                "error": "Internal Server Error",
+                "message": str(e)
+            }), 500
+        # 对于 HTML 请求，返回一个简单的错误页面
+        error_html = """
+        <html>
+            <head><title>Error</title></head>
+            <body>
+                <h1>Internal Server Error</h1>
+                <p>An error occurred while processing your request.</p>
+                <p>Error details: {}</p>
+                <p><a href="javascript:location.reload()">Retry</a></p>
+            </body>
+        </html>
+        """.format(str(e) if app.debug else "Please try again later")
+        return error_html, 500
 def get_template_data(accounts_status):
+    try:
+        if not multi_auth_manager or not multi_auth_manager.auth_managers:
+            return {
+                "total_accounts": 0,
+                "valid_accounts": 0,
+                "total_requests": 0,
+                "accounts": [],
+                "last_update": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+            }
+        total_accounts = len(accounts_status)
+        valid_accounts = sum(1 for acc in accounts_status if acc["is_valid"])
+        accounts_data = []
+        total_requests = 0
+        for auth_manager in multi_auth_manager.auth_managers:
+            try:
+                success_rate = 0
+                if auth_manager.total_requests > 0:
+                    success_rate = (auth_manager.success_requests / auth_manager.total_requests) * 100
+                account_info = {
+                    "email": auth_manager._email,
+                    "is_valid": auth_manager.is_token_valid(),
+                    "total_requests": auth_manager.total_requests,
+                    "success_requests": auth_manager.success_requests,
+                    "failed_requests": auth_manager.failed_requests,
+                    "success_rate": success_rate,
+                    "last_used_time": auth_manager.last_used_time.strftime('%m/%d/%Y, %I:%M:%S %p') if auth_manager.last_used_time else "从未使用"
+                }
+                accounts_data.append(account_info)
+                total_requests += auth_manager.total_requests
+            except Exception as e:
+                logger.error(f"Error processing account {auth_manager._email}: {str(e)}", exc_info=True)
+                continue
+        return {
+            "total_accounts": total_accounts,
+            "valid_accounts": valid_accounts,
+            "total_requests": total_requests,
+            "accounts": accounts_data,
+            "last_update": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         }
+    except Exception as e:
+        logger.error(f"Error in get_template_data: {str(e)}", exc_info=True)
+        raise
 def get_json_status(accounts_status):
     template_data = get_template_data(accounts_status)
                 if response.status_code == 403:  # Forbidden, 模型使用限制
                     logger.warning(f"Model {model_id} usage limit reached for account {auth_manager._email}")
+                    # ��即触发注册
                     if trigger_registration():
                         # 重试请求
                         return make_request(payload, None, model_id)
     port = int(os.environ.get("PORT", 3000))
     app.run(debug=False, host='0.0.0.0', port=port, threaded=True)
+# 添加错误处理器
+@app.errorhandler(Exception)
+def handle_exception(e):
+    """处理所有异常"""
+    logger.error(f"Unhandled exception: {str(e)}", exc_info=True)
+    # 如果是 HTTP 异常，返回其状态码
+    if isinstance(e, HTTPException):
+        return jsonify({
+            "error": {
+                "type": "http_error",
+                "code": e.code,
+                "name": e.name,
+                "description": e.description
+            }
+        }), e.code
+    # 其他异常返回 500
+    return jsonify({
+        "error": {
+            "type": "server_error",
+            "message": "Internal Server Error",
+            "details": str(e) if app.debug else "An unexpected error occurred"
+        }
+    }), 500

register_bot.py CHANGED Viewed

@@ -1,303 +1,303 @@
-import requests
-from bs4 import BeautifulSoup
-from requests.adapters import HTTPAdapter
-from requests.packages.urllib3.util.retry import Retry
-import uuid
-import json
-import time
-import re
-import hashlib
-import base64
-import random
-import string
-import os
-import sys
-from urllib.parse import quote, urlparse, parse_qs
-import concurrent.futures
-import logging
-# 配置日志（仅控制台输出）
-logging.basicConfig(
-    level=logging.DEBUG,  # 改为DEBUG级别，获取更详细的日志
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    stream=sys.stdout
-)
-# 获取环境变量
-PASTE_API_URL = os.getenv('PASTE_API_URL')
-PASTE_API_PASSWORD = os.getenv('PASTE_API_PASSWORD')
-# 构建上传 URL
-UPLOAD_URL = PASTE_API_URL.replace('/api/paste/', '/api/admin/paste/') + '/content'
-# 新的临时邮箱API
-TEMP_EMAIL_API = "https://www.1secmail.com/api/v1/?action=genRandomMailbox"
-# 注册 API 地址
-REGISTER_API_URL = "https://spuckhogycrxcbomznwo.supabase.co/auth/v1/signup?redirect_to=https%3A%2F%2Fchat.notdiamond.ai"
-# Supabase API
-SUPABASE_API_KEY = os.getenv('SUPABASE_API_KEY')
-# 请求限制和重试相关配置
-MAX_RETRIES = 3
-RETRY_DELAY = 10  # 秒
-RATE_LIMIT_DELAY = 60  # 秒
-def generate_strong_password():
-    """固定密码为password123"""
-    return "password123"
-def generate_code_verifier():
-    return base64.urlsafe_b64encode(uuid.uuid4().bytes).decode('utf-8').rstrip('=')
-def generate_code_challenge(code_verifier):
-    code_challenge = hashlib.sha256(code_verifier.encode('utf-8')).digest()
-    return base64.urlsafe_b64encode(code_challenge).decode('utf-8').rstrip('=')
-def get_temp_email():
-    session = requests.Session()
-    retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
-    session.mount('https://', HTTPAdapter(max_retries=retries))
-    try:
-        response = session.get(TEMP_EMAIL_API)
-        if response.status_code == 200:
-            email = response.json()[0]
-            logging.info(f"Generated temp email: {email}")
-            return email
-    except requests.exceptions.SSLError as e:
-        logging.error(f"SSL Error getting temp email: {e}")
-    return None
-def register(email, password):
-    for attempt in range(MAX_RETRIES):
-        session = requests.Session()
-        headers = {
-            'apikey': SUPABASE_API_KEY,
-            'authorization': f'Bearer {SUPABASE_API_KEY}',
-            'content-type': 'application/json'
-        }
-        code_verifier = generate_code_verifier()
-        code_challenge = generate_code_challenge(code_verifier)
-        payload = {
-            'email': email,
-            'password': password,
-            'code_challenge': code_challenge,
-            'code_challenge_method': 's256',
-            'data': {},
-            'gotrue_meta_security': {}
-        }
-        try:
-            response = session.post(REGISTER_API_URL, headers=headers, json=payload)
-            logging.info(f"Registration response status: {response.status_code}")
-            logging.info(f"Registration response content: {response.text}")
-            if response.status_code == 200:
-                logging.info(f"Registered successfully with email: {email}")
-                return True
-            elif response.status_code == 429:
-                logging.warning(f"Rate limit reached. Waiting {RATE_LIMIT_DELAY} seconds.")
-                time.sleep(RATE_LIMIT_DELAY)
-            else:
-                logging.error(f"Failed to register with email: {email}. Response: {response.text}")
-                return False
-        except requests.exceptions.RequestException as e:
-            logging.error(f"Request error during registration: {e}")
-            if attempt < MAX_RETRIES - 1:
-                logging.info(f"Retrying registration in {RETRY_DELAY} seconds...")
-                time.sleep(RETRY_DELAY)
-            else:
-                logging.error("Max registration retries reached.")
-                return False
-    return False
-def save_account(email, password):
-    try:
-        session = requests.Session()
-        # 首先进行认证
-        auth_url = PASTE_API_URL
-        auth_headers = {
-            'accept': '*/*',
-            'accept-language': 'zh-CN,zh;q=0.9',
-            'user-agent': 'Mozilla/5.0',
-            'x-password': PASTE_API_PASSWORD
-        }
-        # 获取现有内容
-        logging.info(f"Authenticating with URL: {auth_url}")
-        response = session.get(auth_url, headers=auth_headers)
-        logging.info(f"Authentication response status: {response.status_code}")
-        logging.info(f"Authentication response content: {response.text}")
-        if response.status_code != 200:
-            logging.error(f"Authentication failed, status code: {response.status_code}")
-            return False
-        # 从响应获取现有内容
-        existing_data = response.json()
-        existing_content = existing_data.get('content', '')
-        # 构建新的内容
-        if existing_content:
-            new_content = f"{existing_content};{email}|{password}"
-        else:
-            new_content = f"{email}|{password}"
-        logging.info(f"New content to upload: {new_content}")
-        # 上传新内容
-        upload_headers = {
-            'Authorization': 'Basic emhvdWRhbjp6aG91ZGFu',
-            'Content-Type': 'application/json'
-        }
-        upload_payload = {
-            'content': new_content
-        }
-        logging.info(f"Uploading to URL: {UPLOAD_URL}")
-        upload_response = session.put(UPLOAD_URL, headers=upload_headers, json=upload_payload)
-        logging.info(f"Upload response status: {upload_response.status_code}")
-        logging.info(f"Upload response content: {upload_response.text}")
-        if upload_response.status_code == 200:
-            logging.info(f"Successfully uploaded account {email}")
-            return True
-        else:
-            logging.error(f"Failed to upload account {email}. Status code: {upload_response.status_code}")
-            return False
-    except Exception as e:
-        logging.error(f"Comprehensive error uploading account: {e}")
-        return False
-def check_email(email):
-    domain = email.split('@')[1]
-    login = email.split('@')[0]
-    # 获取消息ID
-    mailbox_url = f"https://www.1secmail.com/api/v1/?action=getMessages&login={login}&domain={domain}"
-    session = requests.Session()
-    retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
-    session.mount('https://', HTTPAdapter(max_retries=retries))
-    logging.info(f"Checking email for {email}")
-    time.sleep(10)
-    for _ in range(20):
-        try:
-            response = session.get(mailbox_url)
-            logging.info(f"Email check response status: {response.status_code}")
-            if response.status_code == 200 and response.json():
-                for message in response.json():
-                    if "Confirm Your Signup" in message.get('subject', ''):
-                        # 获取完整邮件内容
-                        message_url = f"https://www.1secmail.com/api/v1/?action=readMessage&login={login}&domain={domain}&id={message['id']}"
-                        message_response = session.get(message_url)
-                        if message_response.status_code == 200:
-                            full_message = message_response.json()
-                            logging.info(f"Full message content found: {full_message}")
-                            # 提取验证链接
-                            match = re.search(r'https?://chat\.notdiamond\.ai/auth/confirm\?[^\s]+', full_message.get('body', ''))
-                            if match:
-                                verification_link = match.group(0)
-                                logging.info(f"Verification link found: {verification_link}")
-                                # 解析 URL 并获取参数
-                                parsed_url = urlparse(verification_link)
-                                query_params = parse_qs(parsed_url.query)
-                                # 使用 Supabase API 进行验证
-                                verify_url = "https://spuckhogycrxcbomznwo.supabase.co/auth/v1/verify"
-                                headers = {
-                                    'apikey': SUPABASE_API_KEY,
-                                    'authorization': f'Bearer {SUPABASE_API_KEY}',
-                                    'content-type': 'application/json'
-                                }
-                                payload = {
-                                    'token_hash': query_params['token_hash'][0],
-                                    'type': 'signup'  # 硬编码为 'signup'
-                                }
-                                verify_response = session.post(verify_url, headers=headers, json=payload)
-                                logging.info(f"Verification response status: {verify_response.status_code}")
-                                if verify_response.status_code == 200:
-                                    logging.info(f"Email verified for {email}")
-                                    return True
-                                else:
-                                    logging.error(f"Failed to verify email for {email}: {verify_response.text}")
-                                    return False
-                            else:
-                                logging.error(f"Failed to extract verification link from email for {email}")
-                                return False
-            time.sleep(5)
-        except Exception as e:
-            logging.error(f"Error during email verification: {e}")
-    logging.warning(f"No verification email found for {email}")
-    return False
-def register_and_verify(num_accounts=1):
-    successful_accounts = []
-    with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
-        futures = []
-        for _ in range(num_accounts):
-            email = get_temp_email()
-            if email:
-                futures.append(executor.submit(process_account, email))
-                time.sleep(RETRY_DELAY)
-        for future in concurrent.futures.as_completed(futures):
-            result = future.result()
-            if result:
-                successful_accounts.append(result)
-    return successful_accounts
-def process_account(email):
-    try:
-        # 使用固定密码
-        password = generate_strong_password()
-        if register(email, password):
-            save_result = save_account(email, password)
-            if not save_result:
-                logging.error(f"Failed to save account {email}")
-                return None
-            if check_email(email):
-                logging.info(f"Account fully processed: {email}")
-                return {'email': email, 'password': password}
-    except Exception as e:
-        logging.error(f"Error processing account {email}: {e}")
-    return None
-def main():
-    num_accounts = 5  # 可以根据需要修改注册数量
-    successful_accounts = register_and_verify(num_accounts)
-    # 仅输出到控制台
-    print(f"Successfully registered {len(successful_accounts)} accounts")
-    for account in successful_accounts:
-        print(f"Email: {account['email']}, Password: {account['password']}")
-if __name__ == "__main__":
-    main()

+import requests
+from bs4 import BeautifulSoup
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
+import uuid
+import json
+import time
+import re
+import hashlib
+import base64
+import random
+import string
+import os
+import sys
+from urllib.parse import quote, urlparse, parse_qs
+import concurrent.futures
+import logging
+# 配置日志（仅控制台输出）
+logging.basicConfig(
+    level=logging.DEBUG,  # 改为DEBUG级别，获取更详细的日志
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    stream=sys.stdout
+)
+# 获取环境变量
+PASTE_API_URL = os.getenv('PASTE_API_URL')
+PASTE_API_PASSWORD = os.getenv('PASTE_API_PASSWORD')
+# 构建上传 URL
+UPLOAD_URL = PASTE_API_URL.replace('/api/paste/', '/api/admin/paste/') + '/content'
+# 新的临时邮箱API
+TEMP_EMAIL_API = "https://www.1secmail.com/api/v1/?action=genRandomMailbox"
+# 注册 API 地址
+REGISTER_API_URL = "https://spuckhogycrxcbomznwo.supabase.co/auth/v1/signup?redirect_to=https%3A%2F%2Fchat.notdiamond.ai"
+# Supabase API
+SUPABASE_API_KEY = os.getenv('SUPABASE_API_KEY')
+# 请求限制和重试相关配置
+MAX_RETRIES = 3
+RETRY_DELAY = 10  # 秒
+RATE_LIMIT_DELAY = 60  # 秒
+def generate_strong_password():
+    """固定密码为password123"""
+    return "password123"
+def generate_code_verifier():
+    return base64.urlsafe_b64encode(uuid.uuid4().bytes).decode('utf-8').rstrip('=')
+def generate_code_challenge(code_verifier):
+    code_challenge = hashlib.sha256(code_verifier.encode('utf-8')).digest()
+    return base64.urlsafe_b64encode(code_challenge).decode('utf-8').rstrip('=')
+def get_temp_email():
+    session = requests.Session()
+    retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
+    session.mount('https://', HTTPAdapter(max_retries=retries))
+    try:
+        response = session.get(TEMP_EMAIL_API)
+        if response.status_code == 200:
+            email = response.json()[0]
+            logging.info(f"Generated temp email: {email}")
+            return email
+    except requests.exceptions.SSLError as e:
+        logging.error(f"SSL Error getting temp email: {e}")
+    return None
+def register(email, password):
+    for attempt in range(MAX_RETRIES):
+        session = requests.Session()
+        headers = {
+            'apikey': SUPABASE_API_KEY,
+            'authorization': f'Bearer {SUPABASE_API_KEY}',
+            'content-type': 'application/json'
+        }
+        code_verifier = generate_code_verifier()
+        code_challenge = generate_code_challenge(code_verifier)
+        payload = {
+            'email': email,
+            'password': password,
+            'code_challenge': code_challenge,
+            'code_challenge_method': 's256',
+            'data': {},
+            'gotrue_meta_security': {}
+        }
+        try:
+            response = session.post(REGISTER_API_URL, headers=headers, json=payload)
+            logging.info(f"Registration response status: {response.status_code}")
+            logging.info(f"Registration response content: {response.text}")
+            if response.status_code == 200:
+                logging.info(f"Registered successfully with email: {email}")
+                return True
+            elif response.status_code == 429:
+                logging.warning(f"Rate limit reached. Waiting {RATE_LIMIT_DELAY} seconds.")
+                time.sleep(RATE_LIMIT_DELAY)
+            else:
+                logging.error(f"Failed to register with email: {email}. Response: {response.text}")
+                return False
+        except requests.exceptions.RequestException as e:
+            logging.error(f"Request error during registration: {e}")
+            if attempt < MAX_RETRIES - 1:
+                logging.info(f"Retrying registration in {RETRY_DELAY} seconds...")
+                time.sleep(RETRY_DELAY)
+            else:
+                logging.error("Max registration retries reached.")
+                return False
+    return False
+def save_account(email, password):
+    try:
+        session = requests.Session()
+        # 首先进行认证
+        auth_url = PASTE_API_URL
+        auth_headers = {
+            'accept': '*/*',
+            'accept-language': 'zh-CN,zh;q=0.9',
+            'user-agent': 'Mozilla/5.0',
+            'x-password': PASTE_API_PASSWORD
+        }
+        # 获取现有内容
+        logging.info(f"Authenticating with URL: {auth_url}")
+        response = session.get(auth_url, headers=auth_headers)
+        logging.info(f"Authentication response status: {response.status_code}")
+        logging.info(f"Authentication response content: {response.text}")
+        if response.status_code != 200:
+            logging.error(f"Authentication failed, status code: {response.status_code}")
+            return False
+        # 从响应获取现有内容
+        existing_data = response.json()
+        existing_content = existing_data.get('content', '')
+        # 构建新的内容
+        if existing_content:
+            new_content = f"{existing_content};{email}|{password}"
+        else:
+            new_content = f"{email}|{password}"
+        logging.info(f"New content to upload: {new_content}")
+        # 上传新内容
+        upload_headers = {
+            'Authorization': 'Basic emhvdWRhbjp6aG91ZGFu',
+            'Content-Type': 'application/json'
+        }
+        upload_payload = {
+            'content': new_content
+        }
+        logging.info(f"Uploading to URL: {UPLOAD_URL}")
+        upload_response = session.put(UPLOAD_URL, headers=upload_headers, json=upload_payload)
+        logging.info(f"Upload response status: {upload_response.status_code}")
+        logging.info(f"Upload response content: {upload_response.text}")
+        if upload_response.status_code == 200:
+            logging.info(f"Successfully uploaded account {email}")
+            return True
+        else:
+            logging.error(f"Failed to upload account {email}. Status code: {upload_response.status_code}")
+            return False
+    except Exception as e:
+        logging.error(f"Comprehensive error uploading account: {e}")
+        return False
+def check_email(email):
+    domain = email.split('@')[1]
+    login = email.split('@')[0]
+    # 获取消息ID
+    mailbox_url = f"https://www.1secmail.com/api/v1/?action=getMessages&login={login}&domain={domain}"
+    session = requests.Session()
+    retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
+    session.mount('https://', HTTPAdapter(max_retries=retries))
+    logging.info(f"Checking email for {email}")
+    time.sleep(10)
+    for _ in range(20):
+        try:
+            response = session.get(mailbox_url)
+            logging.info(f"Email check response status: {response.status_code}")
+            if response.status_code == 200 and response.json():
+                for message in response.json():
+                    if "Confirm Your Signup" in message.get('subject', ''):
+                        # 获取完整邮件内容
+                        message_url = f"https://www.1secmail.com/api/v1/?action=readMessage&login={login}&domain={domain}&id={message['id']}"
+                        message_response = session.get(message_url)
+                        if message_response.status_code == 200:
+                            full_message = message_response.json()
+                            logging.info(f"Full message content found: {full_message}")
+                            # 提取验证链接
+                            match = re.search(r'https?://chat\.notdiamond\.ai/auth/confirm\?[^\s]+', full_message.get('body', ''))
+                            if match:
+                                verification_link = match.group(0)
+                                logging.info(f"Verification link found: {verification_link}")
+                                # 解析 URL 并获取参数
+                                parsed_url = urlparse(verification_link)
+                                query_params = parse_qs(parsed_url.query)
+                                # 使用 Supabase API 进行验证
+                                verify_url = "https://spuckhogycrxcbomznwo.supabase.co/auth/v1/verify"
+                                headers = {
+                                    'apikey': SUPABASE_API_KEY,
+                                    'authorization': f'Bearer {SUPABASE_API_KEY}',
+                                    'content-type': 'application/json'
+                                }
+                                payload = {
+                                    'token_hash': query_params['token_hash'][0],
+                                    'type': 'signup'  # 硬编码为 'signup'
+                                }
+                                verify_response = session.post(verify_url, headers=headers, json=payload)
+                                logging.info(f"Verification response status: {verify_response.status_code}")
+                                if verify_response.status_code == 200:
+                                    logging.info(f"Email verified for {email}")
+                                    return True
+                                else:
+                                    logging.error(f"Failed to verify email for {email}: {verify_response.text}")
+                                    return False
+                            else:
+                                logging.error(f"Failed to extract verification link from email for {email}")
+                                return False
+            time.sleep(5)
+        except Exception as e:
+            logging.error(f"Error during email verification: {e}")
+    logging.warning(f"No verification email found for {email}")
+    return False
+def register_and_verify(num_accounts=1):
+    successful_accounts = []
+    with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
+        futures = []
+        for _ in range(num_accounts):
+            email = get_temp_email()
+            if email:
+                futures.append(executor.submit(process_account, email))
+                time.sleep(RETRY_DELAY)
+        for future in concurrent.futures.as_completed(futures):
+            result = future.result()
+            if result:
+                successful_accounts.append(result)
+    return successful_accounts
+def process_account(email):
+    try:
+        # 使用固定密码
+        password = generate_strong_password()
+        if register(email, password):
+            save_result = save_account(email, password)
+            if not save_result:
+                logging.error(f"Failed to save account {email}")
+                return None
+            if check_email(email):
+                logging.info(f"Account fully processed: {email}")
+                return {'email': email, 'password': password}
+    except Exception as e:
+        logging.error(f"Error processing account {email}: {e}")
+    return None
+def main():
+    num_accounts = 5  # 可以根据需要修改注册数量
+    successful_accounts = register_and_verify(num_accounts)
+    # 仅输出到控制台
+    print(f"Successfully registered {len(successful_accounts)} accounts")
+    for account in successful_accounts:
+        print(f"Email: {account['email']}, Password: {account['password']}")
+if __name__ == "__main__":
+    main()