first commit

account_sync.py

@@ -0,0 +1,131 @@
+import os
+import json
+import time
+import random
+import string
+import requests
+import logging
+from datetime import datetime
+from typing import Optional
+from urllib3.exceptions import InsecureRequestWarning
+
+# 禁用 SSL 警告
+requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
+
+class AccountSync:
+    def __init__(self, api_choice: int = 1, max_retries: int = 3, retry_delay: int = 5):
+        self.api_base_url = "https://cursorapi.nosqli.com/admin"  # API地址写死
+        self.api_choice = api_choice  # 1=听泉助手池, 2=高质量号池
+        self.max_retries = max_retries
+        self.retry_delay = retry_delay
+
+    def get_machine_id(self) -> str:
+        """生成随机machine_id"""
+        return "".join(random.choices(string.ascii_letters + string.digits, k=32))
+
+    def sync_account(self, account_info: dict) -> bool:
+        """同步账号信息到服务器"""
+        retry_count = 0
+        while retry_count < self.max_retries:
+            try:
+                # 根据选择确定API端点
+                endpoint = f"{self.api_base_url}/api.account/{'addpool' if self.api_choice == 2 else 'add'}"
+                logging.info(f"使用API端点: {endpoint}")
+                headers = {
+                    "Content-Type": "application/json",
+                    "Accept": "application/json"
+                }
+                
+                # 获取machine_id
+                machine_id = self.get_machine_id()
+                logging.info(f"使用随机生成的 machine_id: {machine_id}")
+                
+                proxy_host = account_info.get("proxy_host", "")
+                proxy_port = account_info.get("proxy_port", "")
+                proxy_username = account_info.get("proxy_username", "")
+                proxy_password = account_info.get("proxy_password", "")
+                current_proxies = {
+                    'http': f'http://{proxy_username}:{proxy_password}@{proxy_host}:{proxy_port}',
+                    'https': f'http://{proxy_username}:{proxy_password}@{proxy_host}:{proxy_port}'
+                }
+                # 准备请求数据
+                data = {
+                    "email": account_info["email"],
+                    "password": account_info["password"],
+                    "access_token": account_info.get("token", ""),  # 使用token作为access_token
+                    "refresh_token": account_info.get("token", ""), # 使用token作为refresh_token
+                    "machine_id": machine_id,
+                    "user_agent": os.getenv("BROWSER_USER_AGENT", account_info.get("user_agent", ""),),
+                    "registration_time": account_info.get("register_time", datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
+                    "first_name": account_info.get("first_name", ""),
+                    "last_name": account_info.get("last_name", ""),
+                    "ip_address": proxy_host,
+                    "proxy_info": current_proxies,
+                    "proxy_data": account_info.get("proxy_data", ""),
+                    "country": account_info.get("country", ""),
+                    "batch_index": account_info.get("batch_index", "")
+                }
+                logging.info(f"请求数据: {json.dumps(data, ensure_ascii=False)}")
+                
+                # 打印请求信息
+                logging.info(f"请求 URL: {endpoint}")
+                logging.info(f"请求数据: {json.dumps(data, ensure_ascii=False)}")
+                
+                # 发送POST请求
+                response = requests.post(
+                    endpoint,
+                    json=data,
+                    headers=headers,
+                    timeout=10,
+                    verify=False,
+                    proxies=current_proxies
+                )
+                
+                # 打印响应状态和内容
+                logging.info(f"响应状态码: {response.status_code}")
+                logging.info(f"响应头: {dict(response.headers)}")
+                logging.info(f"响应内容: {response.text[:200]}")
+                
+                if response.status_code == 502:
+                    logging.error("服务器网关错误(502)，请检查 API 地址是否正确")
+                    time.sleep(self.retry_delay)
+                    retry_count += 1
+                    continue
+                    
+                try:
+                    response_data = response.json()
+                    if response_data["code"] == 200 and response_data.get("msg") == "添加成功":
+                        logging.info(f"账号 {account_info['email']} 同步成功")
+                        return True
+                    elif response_data["code"] == 400:
+                        logging.warning(f"账号 {account_info['email']} 已存在")
+                        return False
+                    elif response_data["code"] == 500:
+                        logging.error(f"服务器内部错误(500)，第 {retry_count + 1} 次重试")
+                        time.sleep(self.retry_delay)
+                        retry_count += 1
+                        continue
+                    else:
+                        logging.error(f"账号 {account_info['email']} 同步失败: {response_data.get('msg', '未知错误')}")
+                        time.sleep(self.retry_delay)
+                        retry_count += 1
+                        continue
+                except json.JSONDecodeError:
+                    logging.error(f"服务器返回的不是有效的JSON数据: {response.text[:200]}")
+                    time.sleep(self.retry_delay)
+                    retry_count += 1
+                    continue
+                    
+            except requests.exceptions.RequestException as e:
+                logging.error(f"请求失败: {str(e)}")
+                time.sleep(self.retry_delay)
+                retry_count += 1
+                continue
+            except Exception as e:
+                logging.error(f"同步账号 {account_info['email']} 时出错: {str(e)}")
+                time.sleep(self.retry_delay)
+                retry_count += 1
+                continue
+                
+        logging.error(f"账号 {account_info['email']} 同步失败，已重试 {self.max_retries} 次")
+        return False