代码优化

This commit is contained in:
liujianjiang 2025-12-05 09:39:59 +08:00
parent 980f6245d1
commit 6aa3999af7
1 changed files with 2 additions and 14 deletions

16
main.py
View File

@ -7,6 +7,8 @@ from pathlib import Path
from typing import Dict, Any, Optional, List
from fastapi import FastAPI, HTTPException, Depends, Header
from public_function.auth import verify_tk_token
from task_management.all_task_management import AllTask
from account_management.deal_account import DealAccount
from public_function.public_func import read_config, create_logger
from model.model import GoodsInfo, DataReceive, AccountStatus, AccountUpdate, TokenItem, CrawlerItem, ResetTask, AccountObtain
@ -20,7 +22,6 @@ def get_config():
config_path = os.path.join(Path(__file__).resolve().parent, 'public_function/config.yaml')
try:
# 这里假设read_config函数存在
from public_function.public_func import read_config
return read_config(config_path)
except ImportError:
print(f"未找到read_config函数使用默认配置")
@ -31,7 +32,6 @@ def get_account_manager():
"""获取账号管理器实例"""
config = get_config()
try:
from account_management.deal_account import DealAccount
return DealAccount(config)
except ImportError:
print(f"未找到DealAccount类返回模拟实例")
@ -42,24 +42,12 @@ def get_task_manager():
"""任务获取管理器实例"""
config = get_config()
try:
from task_management.all_task_management import AllTask
return AllTask(config)
except ImportError:
print(f"未找到AllTask类返回模拟实例")
return None
def get_reset_manager():
"""设备重置管理器"""
config = get_config()
try:
from task_management.all_task_management import AllTask
return AllTask(config)
except ImportError:
print("未找到 类,返回模拟实例")
return None
# 账号处理相关
@app.post("/obtain_account", summary="获取可用账号")
async def obtain_account(account_data: AccountObtain, account_manager: Any = Depends(get_account_manager)):