#!/usr/bin/env python3
"""测试重试机制脚本"""
from tasks import daily_morning_task
import time

def test_retry_mechanism():
    """测试重试机制"""
    print("=== 测试API重试机制 ===")
    print("启动任务测试...")
    
    # 手动触发任务
    result = daily_morning_task.delay()
    
    print(f"任务ID: {result.id}")
    print("等待任务执行完成...")
    
    # 监控任务状态
    for i in range(30):  # 最多等待30秒
        try:
            status = result.status
            if status in ['SUCCESS', 'FAILURE']:
                print(f"\n任务最终状态: {status}")
                if status == 'SUCCESS':
                    print(f"任务结果: {result.get()}")
                else:
                    print(f"任务失败信息: {result.result}")
                break
            else:
                print(f"任务状态: {status}... (等待中)")
                time.sleep(1)
        except Exception as e:
            print(f"检查任务状态时出错: {e}")
            time.sleep(1)
    
    print("测试完成")

if __name__ == '__main__':
    test_retry_mechanism()