diff --git a/main.py b/main.py index 8ec7e82..3403b21 100644 --- a/main.py +++ b/main.py @@ -8,7 +8,8 @@ from typing import Dict, Any, Optional, List from fastapi import FastAPI, HTTPException, Depends from public_function.public_func import read_config -from model.model import AccountCreate, AccountUpdate, CrawlerTask, AccountDelete, DataReceive +from model.model import CrawlerTask, AccountDelete, DataReceive +from model.model import AccountCreate, AccountUpdate, DeviceResetData app = FastAPI() @@ -135,5 +136,21 @@ async def crawler_task(task_data: CrawlerTask, task_manager: Any = Depends(get_t raise HTTPException(status_code=500, detail="获取数据失败;失败原因{}".format(e)) +@app.get("/device_reset", summary="爬虫任务调用接口") +async def device_reset(task_data: CrawlerTask, reset_manager: Any = Depends(get_reset_manager)): + """设备重置接口""" + try: + params = task_data.dict() + params['task_id'] = uuid.uuid4().hex + # 将任务记录到mysql + # result = await task_manager.task_distribution(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)) + + if __name__ == '__main__': uvicorn.run(app)