first commit

wechat_auto/core/task_scheduler.py

@@ -0,0 +1,97 @@
+import asyncio
+import uuid
+from datetime import datetime
+from typing import Dict, Any, Optional
+from wechat_auto.models.activity import ActivityModel, TaskStatus
+from wechat_auto.core.executor.pyautogui_executor import PyAutoGUIExecutor
+from wechat_auto.core.executor.qwen_ai_executor import QwenAIExecutor
+from wechat_auto.utils.logger import logger
+from wechat_auto.config import settings
+
+
+class TaskScheduler:
+    def __init__(self):
+        self.primary = PyAutoGUIExecutor()
+        self.secondary = QwenAIExecutor()
+        self.max_retries = settings.max_retries
+        self.tasks: Dict[str, TaskStatus] = {}
+    
+    async def publish_activity(self, activity: ActivityModel) -> Dict[str, Any]:
+        task_id = str(uuid.uuid4())
+        logger.info(f"创建任务 {task_id}，发布活动: {activity.title}")
+        
+        task_status = TaskStatus(
+            task_id=task_id,
+            status="running",
+            created_at=datetime.now(),
+            updated_at=datetime.now()
+        )
+        self.tasks[task_id] = task_status
+        
+        result = await self._execute_with_fallback(activity)
+        
+        task_status.status = result.get("status", "failed")
+        task_status.method = result.get("method")
+        task_status.error = result.get("error")
+        task_status.updated_at = datetime.now()
+        
+        return {
+            "task_id": task_id,
+            "status": task_status.status,
+            "method": task_status.method,
+            "error": task_status.error
+        }
+    
+    async def _execute_with_fallback(self, activity: ActivityModel) -> Dict[str, Any]:
+        logger.info("=" * 50)
+        logger.info("开始执行方案1: pyautogui")
+        logger.info("=" * 50)
+        
+        for attempt in range(1, self.max_retries + 1):
+            try:
+                result = await asyncio.to_thread(self.primary.execute, activity)
+                if result.get("status") == "success":
+                    logger.info(f"pyautogui方案成功")
+                    return result
+            except Exception as e:
+                logger.warning(f"pyautogui方案第{attempt}次失败: {e}")
+            
+            if attempt < self.max_retries:
+                delay = settings.retry_base_delay * (2 ** (attempt - 1))
+                logger.info(f"{delay}秒后重试...")
+                await asyncio.sleep(delay)
+        
+        logger.warning("pyautogui方案全部失败，切换到备选方案")
+        
+        logger.info("=" * 50)
+        logger.info("开始执行方案2: Qwen AI")
+        logger.info("=" * 50)
+        
+        for attempt in range(1, self.max_retries + 1):
+            try:
+                result = await self.secondary.execute(activity)
+                if result.get("status") == "success":
+                    logger.info(f"Qwen AI方案成功")
+                    return result
+            except Exception as e:
+                logger.warning(f"Qwen AI方案第{attempt}次失败: {e}")
+            
+            if attempt < self.max_retries:
+                delay = settings.retry_base_delay * (2 ** (attempt - 1))
+                logger.info(f"{delay}秒后重试...")
+                await asyncio.sleep(delay)
+        
+        logger.error("所有方案均失败")
+        return {
+            "status": "failed",
+            "error": "pyautogui和Qwen AI方案均失败"
+        }
+    
+    def get_task_status(self, task_id: str) -> Optional[TaskStatus]:
+        return self.tasks.get(task_id)
+    
+    def list_tasks(self) -> list[TaskStatus]:
+        return list(self.tasks.values())
+
+
+task_scheduler = TaskScheduler()