129 lines
5.9 KiB
Python
129 lines
5.9 KiB
Python
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, max_retries: int = 3, retry_delay: int = 5):
|
|
self.api_base_url = "https://cursorapi.nosqli.com/admin" # API地址写死
|
|
|
|
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/add"
|
|
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,
|
|
"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 |