x

main.py

@@ -0,0 +1,182 @@
+import asyncio
+import sys
+
+# Windows平台特殊处理，强制使用SelectorEventLoop
+if sys.platform.startswith('win'):
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
+from typing import Dict, List
+
+from core.config import Config
+from core.database import DatabaseManager
+from core.logger import setup_logger
+from register.register_worker import RegisterWorker
+from services.email_manager import EmailManager
+from services.fetch_manager import FetchManager
+from services.proxy_pool import ProxyPool
+from services.token_pool import TokenPool
+
+
+class CursorRegister:
+    def __init__(self):
+        self.config = Config.from_yaml()
+        self.logger = setup_logger(self.config)
+        self.db_manager = DatabaseManager(self.config)
+        self.fetch_manager = FetchManager(self.config)
+        self.proxy_pool = ProxyPool(self.config, self.fetch_manager)
+        self.token_pool = TokenPool(self.config)
+        self.email_manager = EmailManager(self.config, self.db_manager)
+        self.register_worker = RegisterWorker(
+            self.config, 
+            self.fetch_manager,
+            self.email_manager
+        )
+        
+    async def initialize(self):
+        """初始化数据库"""
+        await self.db_manager.initialize()
+        
+    async def cleanup(self):
+        """清理资源"""
+        await self.db_manager.cleanup()
+        
+    async def batch_register(self, num: int):
+        """批量注册"""
+        try:
+            self.logger.info(f"开始批量注册 {num} 个账号")
+            
+            # 1. 先获取token对
+            token_pairs = await self.token_pool.batch_generate(num)
+            if not token_pairs:
+                self.logger.error("获取token失败，终止注册")
+                return []
+            
+            actual_num = len(token_pairs)  # 根据实际获取到的token对数量调整注册数量
+            if actual_num < num:
+                self.logger.warning(f"只获取到 {actual_num} 对token，将减少注册数量")
+                num = actual_num
+            
+            # 2. 获取邮箱账号
+            email_accounts = await self.email_manager.batch_get_accounts(num)
+            if len(email_accounts) < num:
+                self.logger.warning(f"可用邮箱账号不足，仅获取到 {len(email_accounts)} 个")
+                num = len(email_accounts)
+            
+            # 3. 获取代理
+            proxies = await self.proxy_pool.batch_get(num)
+            
+            # 在关键位置添加详细日志
+            self.logger.debug(f"代理列表: {proxies}")
+            self.logger.debug(f"邮箱账号: {[a.email for a in email_accounts]}")
+            self.logger.debug(f"尝试使用的token对数量: {len(token_pairs)}")
+            
+            # 4. 创建注册任务
+            tasks = []
+            for account, proxy, token_pair in zip(email_accounts, proxies, token_pairs):
+                task = self.register_worker.register(proxy, token_pair, account)
+                tasks.append(task)
+            
+            # 5. 并发执行
+            results = await asyncio.gather(*tasks, return_exceptions=True)
+            
+            # 6. 处理结果
+            successful = []
+            failed = []
+            skipped = 0
+            
+            for i, result in enumerate(results):
+                if isinstance(result, Exception):
+                    self.logger.error(f"注册任务 {i+1} 失败: {str(result)}")
+                    failed.append(str(result))
+                elif result is None:  # 跳过的账号
+                    skipped += 1
+                else:
+                    try:
+                        # 更新数据库
+                        await self.email_manager.update_account(
+                            result['account_id'],
+                            result['cursor_password'],
+                            result['cursor_cookie'],
+                            result['cursor_jwt']
+                        )
+                        self.logger.debug(f"更新数据库成功 - 账号ID: {result['account_id']}")
+                        successful.append(result)
+                    except Exception as e:
+                        self.logger.error(f"更新数据库失败 - 账号ID: {result['account_id']}, 错误: {str(e)}")
+                        failed.append(str(e))
+            
+            self.logger.info(f"注册完成: 成功 {len(successful)}, 失败 {len(failed)}, 跳过 {skipped}")
+            return successful
+            
+        except Exception as e:
+            self.logger.error(f"批量注册失败: {str(e)}")
+            return []
+            
+    async def _process_results(self, results: List[Dict]):
+        """处理注册结果"""
+        successful = []
+        failed = []
+        
+        for result in results:
+            if isinstance(result, Exception):
+                failed.append(str(result))
+            else:
+                # 更新数据库
+                await self.email_manager.update_account(
+                    result['account_id'],
+                    result['cursor_password'],
+                    result['cursor_cookie']
+                )
+                successful.append(result)
+                
+        print(f"Successfully registered: {len(successful)}")
+        print(f"Failed registrations: {len(failed)}")
+        
+        return successful
+
+
+async def main():
+    register = CursorRegister()
+    await register.initialize()
+
+    try:
+        batch_size = register.config.register_config.batch_size
+        total_registered = 0
+        
+        while True:
+            # 检查是否还有可用的邮箱账号
+            available_accounts = await register.email_manager.batch_get_accounts(1)
+            if not available_accounts:
+                register.logger.info("没有更多可用的邮箱账号，注册完成")
+                break
+
+            # 释放检查用的账号
+            await register.email_manager.update_account_status(available_accounts[0].id, 'pending')
+            
+            # 执行批量注册
+            register.logger.info(f"开始新一轮批量注册，批次大小: {batch_size}")
+            results = await register.batch_register(batch_size)
+            
+            # 统计结果
+            successful = len(results)
+            total_registered += successful
+            
+            register.logger.info(f"当前总进度: 已注册 {total_registered} 个账号")
+            
+            # 如果本批次注册失败率过高，暂停一段时间
+            if successful < batch_size * 0.5:  # 成功率低于50%
+                register.logger.warning("本批次成功率过低，暂停60秒后继续")
+                await asyncio.sleep(60)
+            else:
+                # 正常等待一个较短的时间再继续下一批
+                await asyncio.sleep(5)
+                
+    except Exception as e:
+        register.logger.error(f"程序执行出错: {str(e)}")
+    finally:
+        register.logger.info(f"程序结束，总共成功注册 {total_registered} 个账号")
+        await register.cleanup()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())