From 6aa3999af7a0614942f0a4cfbeb4d1a343e9c432 Mon Sep 17 00:00:00 2001 From: liujianjiang Date: Fri, 5 Dec 2025 09:39:59 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 16 ++-------------- 1 file changed, 2 insertions(+), 14 deletions(-) diff --git a/main.py b/main.py index 0a148b3..af1a418 100644 --- a/main.py +++ b/main.py @@ -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)):