294 lines
13 KiB
Python
294 lines
13 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
Task Action 配置计划
|
||
─────────────────────────────────────────────────────────────────
|
||
描述 Task 中 Action 的挂载方式与执行策略：
|
||
|
||
挂载层级：
|
||
RouteActionPlan —— 整条航线级别的 Action 配置
|
||
SegmentActionPlan —— 单个航段（RouteSegment）级别的 Action 配置
|
||
|
||
执行策略：
|
||
PARALLEL —— 同组 Action 并行执行（无顺序约束）
|
||
SEQUENTIAL—— 同组 Action 按 order 字段串行执行
|
||
|
||
触发时机：
|
||
ON_ENTER —— 进入航线/航段时触发
|
||
ON_EXIT —— 离开航线/航段时触发
|
||
ON_WAYPOINT—— 到达指定航路点时触发
|
||
CONTINUOUS —— 在航线/航段内持续触发（按 interval_s 间隔）
|
||
SCHEDULED —— 在指定偏移时刻触发
|
||
─────────────────────────────────────────────────────────────────
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import uuid
|
||
from dataclasses import dataclass, field
|
||
from enum import Enum
|
||
from typing import Dict, List, Optional
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 执行策略
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
class ExecutionPolicy(Enum):
|
||
PARALLEL = "parallel" # 并行执行（无顺序约束的 Action 组）
|
||
SEQUENTIAL = "sequential" # 串行执行（按 order 字段顺序）
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 触发时机
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
class TriggerMoment(Enum):
|
||
ON_ENTER = "on_enter" # 进入航线/航段时触发（一次）
|
||
ON_EXIT = "on_exit" # 离开航线/航段时触发（一次）
|
||
ON_WAYPOINT = "on_waypoint" # 到达指定航路点时触发
|
||
CONTINUOUS = "continuous" # 持续触发（按 interval_s）
|
||
SCHEDULED = "scheduled" # 在指定时间偏移处触发
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# Action 挂载项
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
@dataclass
|
||
class ActionEntry:
|
||
"""
|
||
单个 Action 挂载项
|
||
封装 Action 对象 + 执行顺序 + 触发时机 + 调度参数
|
||
"""
|
||
action: object # Action 实例（来自 action_module）
|
||
order: Optional[int] = None # 串行顺序编号（None=不参与串行，归入并行组）
|
||
trigger: TriggerMoment = TriggerMoment.ON_ENTER
|
||
interval_s: float = 10.0 # CONTINUOUS 模式触发间隔（秒）
|
||
time_offset_s: float = 0.0 # SCHEDULED 模式时间偏移（秒，相对于段起始）
|
||
waypoint_id: Optional[str] = None # ON_WAYPOINT 模式目标航路点标识符
|
||
enabled: bool = True # 是否启用
|
||
entry_id: str = field(default_factory=lambda: str(uuid.uuid4())[:8])
|
||
description: str = ""
|
||
|
||
@property
|
||
def is_sequential(self) -> bool:
|
||
"""是否参与串行执行"""
|
||
return self.order is not None
|
||
|
||
@property
|
||
def is_parallel(self) -> bool:
|
||
"""是否归入并行执行组"""
|
||
return self.order is None
|
||
|
||
def __repr__(self):
|
||
policy = f"seq[{self.order}]" if self.is_sequential else "parallel"
|
||
return (f"ActionEntry(id={self.entry_id}, "
|
||
f"action={self.action.type_key}, "
|
||
f"policy={policy}, trigger={self.trigger.value})")
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 航段 Action 配置计划
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
@dataclass
|
||
class SegmentActionPlan:
|
||
"""
|
||
单个航段（RouteSegment）的 Action 配置计划
|
||
|
||
segment_index: 对应 FlightRoute.segments 的下标
|
||
entries: 该航段挂载的所有 ActionEntry
|
||
|
||
执行规则：
|
||
- entries 中 order=None 的 ActionEntry → 并行执行组
|
||
- entries 中 order!=None 的 ActionEntry → 按 order 升序串行执行
|
||
- 串行组与并行组的相对执行顺序由 serial_first 控制
|
||
"""
|
||
segment_index: int
|
||
entries: List[ActionEntry] = field(default_factory=list)
|
||
serial_first: bool = True # True=先串行后并行，False=先并行后串行
|
||
description: str = ""
|
||
|
||
# ── 添加 Action ───────────────────────────────────────────────
|
||
|
||
def add(
|
||
self,
|
||
action,
|
||
order: Optional[int] = None,
|
||
trigger: TriggerMoment = TriggerMoment.ON_ENTER,
|
||
interval_s: float = 10.0,
|
||
time_offset_s: float = 0.0,
|
||
waypoint_id: Optional[str] = None,
|
||
description: str = "",
|
||
) -> "SegmentActionPlan":
|
||
"""
|
||
添加 Action 到本航段计划
|
||
:param order: None=并行，整数=串行顺序（从0或1开始均可）
|
||
:return: self（支持链式调用）
|
||
"""
|
||
entry = ActionEntry(
|
||
action = action,
|
||
order = order,
|
||
trigger = trigger,
|
||
interval_s = interval_s,
|
||
time_offset_s = time_offset_s,
|
||
waypoint_id = waypoint_id,
|
||
description = description,
|
||
)
|
||
self.entries.append(entry)
|
||
return self
|
||
|
||
# ── 分组查询 ──────────────────────────────────────────────────
|
||
|
||
@property
|
||
def parallel_entries(self) -> List[ActionEntry]:
|
||
"""获取并行执行组（order=None）"""
|
||
return [e for e in self.entries if e.is_parallel and e.enabled]
|
||
|
||
@property
|
||
def sequential_entries(self) -> List[ActionEntry]:
|
||
"""获取串行执行组（order!=None），按 order 升序排列"""
|
||
return sorted(
|
||
[e for e in self.entries if e.is_sequential and e.enabled],
|
||
key=lambda e: e.order
|
||
)
|
||
|
||
def entries_by_trigger(self, trigger: TriggerMoment) -> List[ActionEntry]:
|
||
"""按触发时机过滤"""
|
||
return [e for e in self.entries if e.trigger == trigger and e.enabled]
|
||
|
||
def __repr__(self):
|
||
return (f"SegmentActionPlan(seg={self.segment_index}, "
|
||
f"entries={len(self.entries)}, "
|
||
f"parallel={len(self.parallel_entries)}, "
|
||
f"sequential={len(self.sequential_entries)})")
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 航线 Action 配置计划
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
@dataclass
|
||
class RouteActionPlan:
|
||
"""
|
||
整条航线的 Action 配置计划
|
||
|
||
包含：
|
||
route_entries —— 航线级别的 ActionEntry（整条航线触发）
|
||
segment_plans —— 各航段的 SegmentActionPlan（按段索引索引）
|
||
|
||
执行规则与 SegmentActionPlan 相同：
|
||
- order=None → 并行组
|
||
- order!=None → 串行组（按 order 升序）
|
||
"""
|
||
route_entries: List[ActionEntry] = field(default_factory=list)
|
||
segment_plans: Dict[int, SegmentActionPlan] = field(default_factory=dict)
|
||
serial_first: bool = True
|
||
description: str = ""
|
||
|
||
# ── 航线级 Action ─────────────────────────────────────────────
|
||
|
||
def add_route_action(
|
||
self,
|
||
action,
|
||
order: Optional[int] = None,
|
||
trigger: TriggerMoment = TriggerMoment.ON_ENTER,
|
||
interval_s: float = 10.0,
|
||
time_offset_s: float = 0.0,
|
||
waypoint_id: Optional[str] = None,
|
||
description: str = "",
|
||
) -> "RouteActionPlan":
|
||
"""
|
||
添加航线级 Action（整条航线范围内触发）
|
||
:param order: None=并行，整数=串行顺序
|
||
:return: self（链式调用）
|
||
"""
|
||
entry = ActionEntry(
|
||
action = action,
|
||
order = order,
|
||
trigger = trigger,
|
||
interval_s = interval_s,
|
||
time_offset_s = time_offset_s,
|
||
waypoint_id = waypoint_id,
|
||
description = description,
|
||
)
|
||
self.route_entries.append(entry)
|
||
return self
|
||
|
||
# ── 航段级 Action ─────────────────────────────────────────────
|
||
|
||
def get_segment_plan(self, segment_index: int) -> SegmentActionPlan:
|
||
"""获取指定航段的计划（不存在则自动创建）"""
|
||
if segment_index not in self.segment_plans:
|
||
self.segment_plans[segment_index] = SegmentActionPlan(
|
||
segment_index=segment_index
|
||
)
|
||
return self.segment_plans[segment_index]
|
||
|
||
def add_segment_action(
|
||
self,
|
||
segment_index: int,
|
||
action,
|
||
order: Optional[int] = None,
|
||
trigger: TriggerMoment = TriggerMoment.ON_ENTER,
|
||
interval_s: float = 10.0,
|
||
time_offset_s: float = 0.0,
|
||
waypoint_id: Optional[str] = None,
|
||
description: str = "",
|
||
) -> "RouteActionPlan":
|
||
"""
|
||
为指定航段添加 Action
|
||
:param segment_index: FlightRoute.segments 的下标
|
||
:param order: None=并行，整数=串行顺序
|
||
:return: self（链式调用）
|
||
"""
|
||
plan = self.get_segment_plan(segment_index)
|
||
plan.add(
|
||
action = action,
|
||
order = order,
|
||
trigger = trigger,
|
||
interval_s = interval_s,
|
||
time_offset_s = time_offset_s,
|
||
waypoint_id = waypoint_id,
|
||
description = description,
|
||
)
|
||
return self
|
||
|
||
# ── 分组查询（航线级）────────────────────────────────────────
|
||
|
||
@property
|
||
def parallel_route_entries(self) -> List[ActionEntry]:
|
||
return [e for e in self.route_entries if e.is_parallel and e.enabled]
|
||
|
||
@property
|
||
def sequential_route_entries(self) -> List[ActionEntry]:
|
||
return sorted(
|
||
[e for e in self.route_entries if e.is_sequential and e.enabled],
|
||
key=lambda e: e.order
|
||
)
|
||
|
||
def route_entries_by_trigger(self, trigger: TriggerMoment) -> List[ActionEntry]:
|
||
return [e for e in self.route_entries
|
||
if e.trigger == trigger and e.enabled]
|
||
|
||
# ── 汇总信息 ──────────────────────────────────────────────────
|
||
|
||
def summary(self) -> str:
|
||
lines = [
|
||
f"RouteActionPlan:",
|
||
f" 航线级 Actions : {len(self.route_entries)} "
|
||
f"(并行={len(self.parallel_route_entries)}, "
|
||
f"串行={len(self.sequential_route_entries)})",
|
||
f" 已配置航段数 : {len(self.segment_plans)}",
|
||
]
|
||
for idx, plan in sorted(self.segment_plans.items()):
|
||
lines.append(
|
||
f" Seg[{idx}]: {len(plan.entries)} actions "
|
||
f"(并行={len(plan.parallel_entries)}, "
|
||
f"串行={len(plan.sequential_entries)})"
|
||
)
|
||
return "\n".join(lines)
|
||
|
||
def __repr__(self):
|
||
return (f"RouteActionPlan(route_actions={len(self.route_entries)}, "
|
||
f"segment_plans={len(self.segment_plans)})") |