modify: 目录结构整理

This commit is contained in:
chakcy 2025-08-12 00:41:18 +08:00
parent 149855b867
commit 3f03290e4f
5 changed files with 3 additions and 45 deletions

View File

@ -1,3 +1,4 @@
from .task_mounter import TaskMounter
from .listen_mounter import ListenMounter
__all__ = ["TaskMounter"]
__all__ = ["TaskMounter", "ListenMounter"]

View File

@ -1,42 +0,0 @@
from typing import Callable, List
class BaseMounter:
@classmethod
def mount_(cls, function: Callable):
setattr(cls, function.__name__, function)
@staticmethod
def mount(meta: dict = {}):
"""带参数的装饰器"""
def decorator(function: Callable):
# 使用自定义名称或函数原名
function.meta = meta # type: ignore
setattr(BaseMounter, function.__name__, function)
return function
return decorator
@classmethod
def get_function(cls, name: str):
return getattr(cls, name, None)
@classmethod
def get_task_list(cls) -> List[str]:
"""获取所有挂载的任务函数名称列表"""
task_list = []
# 遍历类属性字典
for attr_name, attr_value in vars(cls).items():
# 过滤条件:
# 1. 必须是可调用对象(函数)
# 2. 不是类自带的特殊方法(非双下划线开头)
# 3. 不是类方法本身(如 mount_, get_task_list 等)
if (
callable(attr_value)
and not attr_name.startswith("__")
and attr_name not in ["mount_", "mount", "get_function", "get_task_list"]
):
task_list.append(attr_name)
return task_list

View File

@ -1,6 +1,5 @@
from ..mounter.listen_mounter import ListenMounter
import sqlite3
import threading
class ListenOperation:

View File

@ -10,7 +10,7 @@ import threading
class TaskScheduler:
def __init__(self, queue_operation: QueueOperation, task_thread_num: int = 1):
def __init__(self, queue_operation: QueueOperation, task_thread_num: int = 2):
self.task_thread_num = task_thread_num
self.is_running = False
self.executor = ThreadPoolExecutor(max_workers=task_thread_num) # 并行执行任务