158 lines
6.1 KiB
Python
158 lines
6.1 KiB
Python
import functools
|
||
import os,sys
|
||
import threading
|
||
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..'))
|
||
import thirdparty.py_trees as py_trees
|
||
from thirdparty.py_trees.trees import BehaviourTree
|
||
from core.behavior_tree import ReleaseFlowBuilder
|
||
from core.plugin import PluginManager
|
||
from core.defines import ReleasePhase, ProcessStep, ReleaseErr, Dict
|
||
from database import *
|
||
import core.logger as logger
|
||
log = logger.get_logger()
|
||
|
||
def draw_release_flow_status(snapshot_visitor, behaviour_tree):
|
||
"""保存行为树状态为HTML文件"""
|
||
mutex = threading.Lock()
|
||
with mutex:
|
||
# timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||
filename = f"alkaid_release_flow_status.html"
|
||
with open(filename, 'w') as f:
|
||
f.write('<html><head><title>SStar Alkaid Release Flow Status</title><body>')
|
||
f.write(py_trees.display.xhtml_tree(
|
||
behaviour_tree.root,
|
||
show_status=True,
|
||
visited=snapshot_visitor.visited,
|
||
previously_visited=snapshot_visitor.visited
|
||
))
|
||
f.write("</body></html>")
|
||
filename = f"alkaid_release_flow_status.dot"
|
||
with open(filename, 'w') as f:
|
||
f.write(str(py_trees.display.dot_tree(behaviour_tree.root, with_blackboard_variables=True)))
|
||
# log.info(f"Behavior tree saved to {filename}")
|
||
|
||
# 构建环境阶段 (验证配置、拉取code等基础环境的准备)
|
||
def _construct_env_phase(builder: ReleaseFlowBuilder, plugins: PluginManager):
|
||
with builder.sequence(ReleasePhase.ENV):
|
||
with builder.sequence(ProcessStep.PRE):
|
||
pass
|
||
with builder.sequence(ProcessStep.PROCESS):
|
||
if not ReleaseOptions.SkipSyncCode:
|
||
builder.action(plugins.BaseLine.PhaseEnv.Process())
|
||
with builder.sequence(ProcessStep.POST):
|
||
pass
|
||
|
||
# 配置编译文件阶段 （defconfig mkfile 等编译控制文件的配置）
|
||
def _construct_config_phase(builder: ReleaseFlowBuilder, plugins: PluginManager):
|
||
with builder.sequence(ReleasePhase.CONFIG):
|
||
with builder.sequence(ProcessStep.PRE):
|
||
pass
|
||
with builder.sequence(ProcessStep.PROCESS):
|
||
pass
|
||
with builder.sequence(ProcessStep.POST):
|
||
pass
|
||
|
||
# 构建阶段(编译构建sdk，根据上一步的配置执行编译构建)
|
||
def _construct_build_phase(builder: ReleaseFlowBuilder, plugins: PluginManager):
|
||
with builder.sequence(ReleasePhase.BUILD):
|
||
with builder.sequence(ProcessStep.PRE):
|
||
pass
|
||
with builder.sequence(ProcessStep.PROCESS):
|
||
pass
|
||
with builder.sequence(ProcessStep.POST):
|
||
pass
|
||
|
||
# 检查阶段（检查SDK编译结果是否符合预期）
|
||
def _construct_check_phase(builder: ReleaseFlowBuilder, plugins: PluginManager):
|
||
with builder.sequence(ReleasePhase.CHECK):
|
||
with builder.sequence(ProcessStep.PRE):
|
||
pass
|
||
with builder.sequence(ProcessStep.PROCESS):
|
||
pass
|
||
with builder.sequence(ProcessStep.POST):
|
||
pass
|
||
|
||
# 打包阶段 (将SDK其他文件诸如doc，tool等打包)
|
||
def _construct_package_phase(builder: ReleaseFlowBuilder, plugins: PluginManager):
|
||
with builder.sequence(ReleasePhase.PACKAGE):
|
||
with builder.sequence(ProcessStep.PRE):
|
||
pass
|
||
with builder.sequence(ProcessStep.PROCESS):
|
||
pass
|
||
with builder.sequence(ProcessStep.POST):
|
||
pass
|
||
|
||
# 验证阶段 (验证Release的SDK是否可以正常工作)
|
||
def _construct_verify_phase(builder: ReleaseFlowBuilder, plugins: PluginManager):
|
||
with builder.sequence(ReleasePhase.VERIFY):
|
||
with builder.sequence(ProcessStep.PRE):
|
||
pass
|
||
with builder.sequence(ProcessStep.PROCESS):
|
||
pass
|
||
with builder.sequence(ProcessStep.POST):
|
||
pass
|
||
|
||
class ReleaseFlow():
|
||
def __init__(self):
|
||
# 我们的Flow默认是一个串行的流程，所以根节点用Sequence
|
||
self.builder = ReleaseFlowBuilder()
|
||
self.tree = None
|
||
self.flow = None
|
||
self.timer = None
|
||
self.status_update_interval = 1 # 定时器间隔时间，单位秒
|
||
self.running = False
|
||
|
||
# 构建 Alkaid Release流程
|
||
def construct(self, plugins: PluginManager):
|
||
with self.builder.sequence("AlkaidReleaseFlow") as builder:
|
||
_construct_env_phase(builder, plugins)
|
||
_construct_config_phase(builder, plugins)
|
||
_construct_build_phase(builder, plugins)
|
||
_construct_check_phase(builder, plugins)
|
||
_construct_package_phase(builder, plugins)
|
||
_construct_verify_phase(builder, plugins)
|
||
self.flow = self.builder.build()
|
||
|
||
def _start_update_status(self, snapshot_visitor):
|
||
"""启动定时器"""
|
||
if self.timer is not None:
|
||
return
|
||
|
||
def timer_callback():
|
||
if self.tree is not None:
|
||
draw_release_flow_status(snapshot_visitor, self.tree)
|
||
# 重新设置定时器
|
||
self.timer = threading.Timer(self.status_update_interval, timer_callback)
|
||
self.timer.start()
|
||
|
||
# 启动第一个定时器
|
||
self.timer = threading.Timer(self.status_update_interval, timer_callback)
|
||
self.tree.add_post_tick_handler(functools.partial(draw_release_flow_status, snapshot_visitor))
|
||
self.timer.start()
|
||
|
||
# 运行Release 流程
|
||
def run(self):
|
||
if self.flow is None:
|
||
raise ReleaseErr("Flow is not built")
|
||
# 使用封装好的BehaviourTree去管理流程
|
||
self.tree = BehaviourTree(root=self.flow)
|
||
snapshot_visitor = py_trees.visitors.SnapshotVisitor()
|
||
self.tree.visitors.append(snapshot_visitor)
|
||
self.tree.setup() # 初始化树
|
||
# 启动定时器
|
||
self._start_update_status(snapshot_visitor)
|
||
# print(
|
||
# py_trees.display.dot_tree(
|
||
# self.tree.root
|
||
# )
|
||
# )
|
||
self.tree.tick() # 触发tick进行执行
|
||
return self.tree.root.status
|
||
|
||
def shutdown(self):
|
||
if self.timer is not None:
|
||
self.timer.cancel()
|
||
self.timer = None
|
||
if self.tree is not None:
|
||
self.tree.shutdown()
|