From d08b4e203b68b3e6e83a38e511ae1eb1d13553f1 Mon Sep 17 00:00:00 2001
From: liujianjiang <liujianjiang@chongdaads.com>
Date: Fri, 28 Nov 2025 15:38:58 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E8=8E=B7=E5=8F=96=E4=BB=BB?=
 =?UTF-8?q?=E5=8A=A1=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index ac22180..f393e93 100644
--- a/main.py
+++ b/main.py
@@ -38,18 +38,18 @@ def get_account_manager():
     return None
 
 
-def get_task_manager():
+def get_crawler_manager():
     """任务管理器实例"""
     config = get_config()
     try:
         from crawler_management.crawler_management import CrawlerManagement
-        return AllTask(config)
+        return CrawlerManagement(config)
     except ImportError:
         print(f"未找到AllTask类，返回模拟实例")
     return None
 
 
-def get_crawler_manager():
+def get_task_manager():
     """任务获取管理器实例"""
     config = get_config()
     try:
@@ -133,13 +133,22 @@ async def update_account(account_data: AccountUpdate, account_manager: Any = Dep
         raise HTTPException(status_code=500, detail="账号失败，请重试,失败原因：{}".format(e))
 
 
-@app.get("/get_crawler_task", summary="获取任务")
+@app.post("/get_crawler_task", summary="获取任务")
 async def get_crawler_task(task_data: CrawlerItem, task_manager: Any = Depends(get_crawler_manager)):
     """
     获取指定应用的可用账号
     - **app_name**: 应用名称
     - **country**: 应用名称
     """
+    try:
+        params = task_data.model_dump()
+        result = task_manager.get_task_item(params)
+        if result:
+            return {"code": 200, "message": "任务获取成功", "data": result}
+        raise HTTPException(status_code=404, detail="队列暂时没有任务，请等待一段时间后重新尝试")
+    except Exception as e:
+        print(f"获取任务失败，失败原因: {e}")
+        raise HTTPException(status_code=500, detail="获取任务失败；失败原因{}".format(e))
 
 
 @app.post("/receive_data", summary="接收抓取得数据")