Upload register_bot.py

register_bot.py

@@ -18,7 +18,7 @@ import logging
 
 # 配置日志（仅控制台输出）
 logging.basicConfig(
-    level=logging.INFO, 
+    level=logging.DEBUG,  # 改为DEBUG级别，获取更详细的日志
     format='%(asctime)s - %(levelname)s - %(message)s', 
     stream=sys.stdout
 )
@@ -63,9 +63,11 @@ def get_temp_email():
     try:
         response = session.get(TEMP_EMAIL_API)
         if response.status_code == 200:
-            return response.json()[0]
+            email = response.json()[0]
+            logging.info(f"Generated temp email: {email}")
+            return email
     except requests.exceptions.SSLError as e:
-        logging.error(f"SSL Error: {e}")
+        logging.error(f"SSL Error getting temp email: {e}")
     return None
 
 def register(email, password):
@@ -92,8 +94,11 @@ def register(email, password):
         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 with email: {email}")
+                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.")
@@ -103,20 +108,18 @@ def register(email, password):
                 return False
         
         except requests.exceptions.RequestException as e:
-            logging.error(f"Request error: {e}")
+            logging.error(f"Request error during registration: {e}")
             if attempt < MAX_RETRIES - 1:
-                logging.info(f"Retrying in {RETRY_DELAY} seconds...")
+                logging.info(f"Retrying registration in {RETRY_DELAY} seconds...")
                 time.sleep(RETRY_DELAY)
             else:
-                logging.error("Max retries reached.")
+                logging.error("Max registration retries reached.")
                 return False
     
     return False
 
 def save_account(email, password):
-    # 上传到指定 URL
     try:
-        # 先获取现有内容
         session = requests.Session()
         
         # 首先进行认证
@@ -129,11 +132,15 @@ def save_account(email, 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"认证失败，状态码：{response.status_code}")
-            return
+            logging.error(f"Authentication failed, status code: {response.status_code}")
+            return False
         
         # 从响应获取现有内容
         existing_data = response.json()
@@ -145,6 +152,8 @@ def save_account(email, password):
         else:
             new_content = f"{email}|{password}"
         
+        logging.info(f"New content to upload: {new_content}")
+        
         # 上传新内容
         upload_headers = {
             'Authorization': 'Basic emhvdWRhbjp6aG91ZGFu',
@@ -155,15 +164,22 @@ def save_account(email, password):
             '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"Error uploading account: {e}")
+        logging.error(f"Comprehensive error uploading account: {e}")
+        return False
 
 def check_email(email):
     domain = email.split('@')[1]
@@ -176,56 +192,66 @@ def check_email(email):
     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):
-        response = session.get(mailbox_url)
-        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: {full_message}")
+        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)
                         
-                        # 提取验证链接
-                        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}")
+                        if message_response.status_code == 200:
+                            full_message = message_response.json()
+                            logging.info(f"Full message content found: {full_message}")
                             
-                            # 解析 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)
-                            
-                            if verify_response.status_code == 200:
-                                logging.info(f"Email verified for {email}")
-                                return True
+                            # 提取验证链接
+                            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 verify email for {email}: {verify_response.text}")
+                                logging.error(f"Failed to extract verification link from email for {email}")
                                 return False
-                        else:
-                            logging.error(f"Failed to extract verification link from email for {email}")
-                            return False
-        time.sleep(5)
+            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
 
@@ -252,8 +278,13 @@ def process_account(email):
         password = generate_strong_password()
         
         if register(email, password):
-            save_account(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}")
@@ -263,7 +294,7 @@ 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']}")