butubb
98cdc39224
tasks/douyin/（task_type=douyin_nurture）整体删除，唯一任务类型是 generic_steps。 所有默认值/文档/接口示例同步改成 generic_steps： - tasks/：删 douyin 包；__init__ 只注册 generic；base.py/generic 注释改为照 generic 抄 - 默认值：core/models.py（列默认 + 旧 JSON 迁移默认）、core/task_manager.py TaskJob、 web/tasks_api.py 建任务默认、static/admin/tasks.js 新建任务默认 - core/task_manager.py：去掉 douyin 专属的"清理废弃 comment 参数"迁移块，改为**启动时告警** 仍残留已删类型的任务（只告警不改数据）；run_job_now 对已删类型直接返回明确错误， 不再"报已触发、线程里静默失败" - 清理残留：douyin_running 状态位（无任何读取方）、core/__init__、core/actions/*、 core/logger.py 注释里的抖音示例 - 文档：README（特性/目录树/类型表/参数表/示例）、TASK_DEV（目录树/注册说明/模板引用）、 ARCHITECTURE（注册示例/action 注册表示例）、API.md（task_types 与任务 JSON 示例）、 AI_TASK_GEN（P1 去掉 douyin 预设）、DEVELOPMENT - 注：示例里"抖音"作为**App 名**（MCP 列应用、AI 建任务的需求举例）保留，与任务类型无关 自测（全部通过）：类型列表只剩 generic_steps；建任务不传类型默认 generic_steps；传 douyin_nurture 被 400 拒；库里塞残留旧类型任务 → 启动日志告警 + 执行返回明确错误 + 不自动删用户数据；前端新建任务下拉 1 项且默认选中、界面建任务成功；监控页卡片两个按钮 + 覆盖设备正常。临时任务/数据验完已清理，任务集合复原。
121 lines
4.4 KiB
Python
121 lines
4.4 KiB
Python
"""任务类型基类与注册机制。
|
||
|
||
目录约定：
|
||
config.py — 项目根配置（STF/adb/web 等，核心程序用，不放任务参数）
|
||
core/ — 核心程序（STF、adb、BaseWorker、任务管理器）
|
||
tasks/ — 任务包，每个任务一个子文件夹，自包含任务+专属操作+参数
|
||
tasks/<name>/
|
||
__init__.py — import task 触发注册
|
||
task.py — BaseTask 子类 + Worker + DEFAULT_PARAMS
|
||
actions/ — 该任务专属操作（xpath 等只适用于本任务，不污染全局）
|
||
__init__.py
|
||
base.py — 本任务的操作基类 + 注册器
|
||
xxx.py — 具体操作
|
||
|
||
新增任务步骤（照着 tasks/generic/ 抄即可）：
|
||
|
||
1. 在 tasks/ 下新建 my_task/ 子包（文件夹 + __init__.py）
|
||
2. 在 my_task/task.py 顶部写默认参数：
|
||
|
||
DEFAULT_PARAMS = {"loop": 10, "actions": {...}}
|
||
|
||
3. 写一个 BaseWorker 子类，实现 run_task(d)：
|
||
|
||
from core.device_worker import BaseWorker, _update_status
|
||
|
||
class MyWorker(BaseWorker):
|
||
def run_task(self, d): # d 是 u2.Device，基类已连好
|
||
d.app_start("com.xxx", wait=True)
|
||
while not self.stopped():
|
||
self.set_action("干活中")
|
||
self.set_progress(done=1)
|
||
|
||
4. 写一个 BaseTask 子类，用 @register_task 装饰，实现 create_worker：
|
||
|
||
@register_task
|
||
class MyTask(BaseTask):
|
||
task_type = "my_task"
|
||
name = "我的任务"
|
||
default_params = dict(DEFAULT_PARAMS)
|
||
|
||
# 有专属操作时重写这两个方法，从 actions 包取
|
||
@classmethod
|
||
def list_action_types(cls):
|
||
from .actions import list_action_types
|
||
return list_action_types()
|
||
|
||
@classmethod
|
||
def get_action_class(cls, action_type):
|
||
from .actions import get_action_class
|
||
return get_action_class(action_type)
|
||
|
||
def create_worker(self, serial, params):
|
||
return MyWorker(serial, params=params)
|
||
|
||
5. 在 my_task/__init__.py 加：from . import task （触发注册）
|
||
6. 在 tasks/__init__.py 加：from .my_task import task （触发注册）
|
||
|
||
做完前端自动出现新任务。有专属操作时前端"操作"区自动列出。
|
||
|
||
BaseWorker 已封装好的（子类不用管）：
|
||
- STF 占用/释放（try/finally 保证释放）
|
||
- adb 连接 / u2.connect
|
||
- 状态上报（status/current_action/last_error/model/remote_adb_url）
|
||
- 异常捕获（run_task 里抛异常 → status=error，设备仍会释放）
|
||
- stop 信号（self.stopped() 检查）
|
||
|
||
子类可用的工具：
|
||
self.stopped() — 是否被要求停止
|
||
self.set_action(s) — 上报当前动作
|
||
self.set_progress() — 上报进度字段
|
||
self.d — u2.Device
|
||
self.params — 任务参数 dict
|
||
core.u2_helper — 通用 u2 辅助函数（ensure_app_running/safe_click 等）
|
||
"""
|
||
|
||
|
||
class BaseTask:
|
||
"""任务类型基类。子类实现 create_worker。
|
||
|
||
有专属操作的任务重写 list_action_types / get_action_class，
|
||
从自己的 actions 子包取（操作 xpath 等只适用于本任务）。
|
||
"""
|
||
task_type = "base"
|
||
name = "基础任务"
|
||
description = ""
|
||
default_params = {}
|
||
|
||
def create_worker(self, serial, params):
|
||
"""返回一个 threading.Thread（已启动或待启动），执行实际任务。"""
|
||
raise NotImplementedError
|
||
|
||
@classmethod
|
||
def list_action_types(cls):
|
||
"""返回本任务支持的操作列表。默认无操作，子类按需重写。"""
|
||
return []
|
||
|
||
@classmethod
|
||
def get_action_class(cls, action_type):
|
||
"""按 action_type 取操作类。默认无操作，子类按需重写。"""
|
||
return None
|
||
|
||
|
||
_TASK_TYPES = {}
|
||
|
||
|
||
def register_task(task_cls):
|
||
"""注册新任务类型。用作装饰器。"""
|
||
_TASK_TYPES[task_cls.task_type] = task_cls
|
||
return task_cls
|
||
|
||
|
||
def list_task_types():
|
||
"""返回所有已注册任务类型的元信息（供前端展示）。"""
|
||
return [{"task_type": t.task_type, "name": t.name, "description": t.description,
|
||
"default_params": t.default_params} for t in _TASK_TYPES.values()]
|
||
|
||
|
||
def get_task_class(task_type):
|
||
"""按 task_type 取任务类。"""
|
||
return _TASK_TYPES.get(task_type)
|