108000207
127 lines
4.0 KiB
Python
127 lines
4.0 KiB
Python
from json import loads as json_parse, dumps as _json_stringify
|
|
from socket import if_nametoindex
|
|
from typing import Dict, List, Optional, Any
|
|
from xml.dom.expatbuilder import parseString
|
|
import paho.mqtt.client as mqtt
|
|
from biopro.text import *
|
|
from .task import Task
|
|
|
|
_RUNTIME_COMPILE = False
|
|
|
|
def json_stringify(o: Any) -> str:
|
|
return _json_stringify(o, separators=(',', ':'))
|
|
|
|
class TaskManager():
|
|
def __init__(self, task_list):
|
|
self._task_list = []
|
|
self._next_task = []
|
|
self._prev_task = None
|
|
self._running_task = None
|
|
|
|
self.load_task_list(task_list)
|
|
|
|
# self.set_running_task(self._task_list[0])
|
|
# self._running_task.run()
|
|
self._next_task.append(self._task_list[0])
|
|
|
|
@property
|
|
def task_list(self):
|
|
return self._task_list
|
|
|
|
@property
|
|
def export_task_list(self):
|
|
return [d.as_json() for d in self._task_list]
|
|
|
|
@property
|
|
def prev_task(self):
|
|
return self._prev_task
|
|
|
|
@property
|
|
def running_task(self):
|
|
return self._running_task
|
|
|
|
@property
|
|
def next_task(self):
|
|
return self._next_task
|
|
|
|
@property
|
|
def check_list(self):
|
|
return [self._running_task, *self._next_task]
|
|
|
|
@property
|
|
def check_list(self):
|
|
return [self._running_task, *self._next_task]
|
|
|
|
def load_task_list(self, task_list):
|
|
for task in task_list:
|
|
task = Task(task)
|
|
self._task_list.append(task)
|
|
|
|
# original version
|
|
""" def set_running_task(self, task):
|
|
try:
|
|
# if there is task running & same task active ,then reject
|
|
if self._running_task != None and self._running_task.uuid == task.uuid:
|
|
return False
|
|
|
|
# save running task
|
|
self._prev_task = self._running_task
|
|
# clear next task list
|
|
self._next_task.clear()
|
|
self._running_task = task
|
|
self._running_task.run()
|
|
|
|
if self._prev_task != None:
|
|
# if previous task is still running, then need to close
|
|
if self._prev_task.status == 1:
|
|
self._prev_task.stop()
|
|
print('prev', 'run', self._prev_task.name, self._running_task.name)
|
|
|
|
for task_uuid in self._running_task.next:
|
|
_task = next((task for task in self._task_list if task.uuid == task_uuid), None)
|
|
if _task != None:
|
|
self._next_task.append(_task)
|
|
except RuntimeError as e:
|
|
print(e) """
|
|
|
|
def set_running_task(self, task):
|
|
try:
|
|
# if there is task running & same task active ,then reject
|
|
if self._running_task != None and self._running_task.uuid == task.uuid:
|
|
# print('set_running_task return false: ', self._running_task.name)
|
|
return False
|
|
|
|
# save running task
|
|
self._prev_task = self._running_task
|
|
# clear next task list
|
|
self._next_task.clear()
|
|
self._running_task = task
|
|
self._running_task.run()
|
|
|
|
if self._prev_task != None:
|
|
# if previous task is still running, then need to close
|
|
if self._prev_task.status == 1:
|
|
self._prev_task.stop()
|
|
print('prev', 'run', self._prev_task.name, self._running_task.name)
|
|
|
|
# print(f'!!@#!set_running_task\n\n\n{task._cycle_count}\n{task._cycle}\n\n\n')
|
|
if task._cycle_count < task._cycle - 1:
|
|
task._cycle_count += 1
|
|
# use uuid to get the cycle next task
|
|
cycle_target = [_task_next for _task_next in self._task_list if _task_next.uuid == task._cycle_next_uuid]
|
|
# cycle_target = self._task_list[task._cycle_next - 1] # original use index version
|
|
cycle_target[0].reset()
|
|
self._next_task.append(cycle_target[0])
|
|
else:
|
|
task._cycle_count = 0
|
|
for task_uuid in self._running_task.next:
|
|
_task = next((task for task in self._task_list if task.uuid == task_uuid), None)
|
|
if _task != None:
|
|
_task.reset()
|
|
self._next_task.append(_task)
|
|
except RuntimeError as e:
|
|
print(e)
|
|
|
|
def get_task(self, task_id):
|
|
return self._task_list[task_id]
|
|
|