ljj
/
crawler_task_management
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

代码优化

This commit is contained in:
liujianjiang 2025-11-27 11:39:28 +08:00
parent b135ba1a9c
commit 10e8d1d4e9
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
main.py
View File

@ -79,9 +79,9 @@ async def add_account(account_data: AccountCreate, account_manager: Any = Depend
- **app_name**: 应用名称 - **app_name**: 应用名称
""" """
try: try:
print(account_data.dict()) print(account_data.model_dump())
# 这里应该调用实际的添加账号方法 # 这里应该调用实际的添加账号方法
result = await account_manager.add_account([account_data.dict()]) result = await account_manager.add_account([account_data.model_dump()])
if result: if result:
return {"code": 200, "message": "新增账号成功", "data": result} return {"code": 200, "message": "新增账号成功", "data": result}
raise HTTPException(status_code=404, detail="新增账号失败") raise HTTPException(status_code=404, detail="新增账号失败")
@ -99,7 +99,7 @@ async def update_account(account_data: AccountUpdate, account_manager: Any = Dep
- **app_name**: 应用名称 - **app_name**: 应用名称
""" """
try: try:
data = account_data.dict() data = account_data.model_dump()
set_param = {"status": data["status"]} set_param = {"status": data["status"]}
# params = {"account_id": data["account_id"], "app_name": data["app_name"]} # params = {"account_id": data["account_id"], "app_name": data["app_name"]}
params = (data["account_id"], data["app_name"]) params = (data["account_id"], data["app_name"])
@ -118,7 +118,7 @@ async def update_account(account_data: AccountUpdate, account_manager: Any = Dep
async def receive_data(task_data: DataReceive, task_manager: Any = Depends(get_task_manager)): async def receive_data(task_data: DataReceive, task_manager: Any = Depends(get_task_manager)):
"""数据接收接口""" """数据接收接口"""
try: try:
params = task_data.dict() params = task_data.model_dump()
print(params) print(params)
result = await task_manager.deal_receive_data(params) result = await task_manager.deal_receive_data(params)
if result: if result:
@ -133,7 +133,7 @@ async def receive_data(task_data: DataReceive, task_manager: Any = Depends(get_t
async def crawler_task(task_data: CrawlerTask, task_manager: Any = Depends(get_task_manager)): async def crawler_task(task_data: CrawlerTask, task_manager: Any = Depends(get_task_manager)):
"""爬虫任务接口""" """爬虫任务接口"""
try: try:
params = task_data.dict() params = task_data.model_dump()
params['task_id'] = uuid.uuid4().hex params['task_id'] = uuid.uuid4().hex
result = await task_manager.task_distribution(params) result = await task_manager.task_distribution(params)
if result: if result:
@ -148,7 +148,7 @@ async def crawler_task(task_data: CrawlerTask, task_manager: Any = Depends(get_t
# async def device_reset(task_data: CrawlerTask, reset_manager: Any = Depends(get_reset_manager)): # async def device_reset(task_data: CrawlerTask, reset_manager: Any = Depends(get_reset_manager)):
# """设备重置接口""" # """设备重置接口"""
# try: # try:
# params = task_data.dict() # params = task_data.model_dump()
# params['task_id'] = uuid.uuid4().hex # params['task_id'] = uuid.uuid4().hex
# # 将任务记录到mysql # # 将任务记录到mysql
# # result = await task_manager.task_distribution(params) # # result = await task_manager.task_distribution(params)