controller-wisetopdataserver/python/biopro/project/task_manager.py

from json import loads as json_parse, dumps as _json_stringify
from typing import Dict, 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._current_task = None
      self._task_list = []
      self._need_to_check_task_list = []

      for task in task_list:
        task = Task(task) 
        self._task_list.append(task)
      
      self.set_current_task(self._task_list[0])
      self._current_task.run()
    
    def remove(self, index):
      pass

    def get_task(self, task_id):
      return self._task_list[task_id]

    def get_check_task_list(self):
      return self._need_to_check_task_list

    def get_current_task(self) -> Task:
      return self._current_task

    def get_next_task(self):
      return  self._current_task.next

    def set_current_task(self, task):
      try:
        if self._current_task != None and self._current_task.id != task.id:
          self._current_task.stop()

        self._current_task = task
        self._need_to_check_task_list = []
        # append current task
        self._need_to_check_task_list.append(self._current_task)

        # append next task
        for task_uuid in self._current_task.next:
          task = next((task for task in self._task_list if task.uuid == task_uuid), None)
          self._need_to_check_task_list.append(task)
      except Exception as e:
        print(e)