Function main

daily/sfsy.py:1109–1196

主函数

()

# ==================== 主程序 ====================
def main():
    """主函数"""
    config = Config()

    env_value = os.getenv(config.ENV_NAME)
    if not env_value:
        print(f"❌ 未找到环境变量 {config.ENV_NAME}，请检查配置")
        return

    account_urls = [url.strip() for url in env_value.split('&') if url.strip()]
    if not account_urls:
        print(f"❌ 环境变量 {config.ENV_NAME} 为空或格式错误")
        return

    # 随机打乱账号顺序
    random.shuffle(account_urls)
    print(f"🔀 已随机打乱账号执行顺序")

    print("=" * 50)
    print(f"🎉 {config.APP_NAME} v{config.VERSION}")
    print(f"👨‍💻 作者: 爱学习的呆子")
    print(f"📱 共获取到 {len(account_urls)} 个账号")
    print(f"⚙️ 并发数量: {CONCURRENT_NUM}")
    print(f"⏰ 执行时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
    print("=" * 50)
    
    # 收集所有账号的统计信息
    all_results = []
    
    if CONCURRENT_NUM <= 1:
        # 串行执行模式
        print("🔄 使用串行模式执行...")
        for index, account_url in enumerate(account_urls):
            account = AccountManager(account_url, index, config)
            result = account.run()
            result['index'] = index
            all_results.append(result)
            
            if index < len(account_urls) - 1:
                print("=" * 50)
                print(f"⏳ 等待 2 秒后执行下一个账号...")
                time.sleep(2)
    else:
        # 并发执行模式
        print(f"🔄 使用并发模式执行，并发数: {CONCURRENT_NUM}")
        
        # 使用线程池执行
        with ThreadPoolExecutor(max_workers=CONCURRENT_NUM) as executor:
            # 提交所有任务
            future_to_index = {
                executor.submit(run_single_account, account_url, index, config): index 
                for index, account_url in enumerate(account_urls)
            }
            
            # 等待任务完成
            for future in as_completed(future_to_index):
                result = future.result()
                all_results.append(result)