mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-12-23 14:04:58 +01:00
123 lines
No EOL
2.9 KiB
Python
123 lines
No EOL
2.9 KiB
Python
import asyncio
|
|
import logging
|
|
from abc import abstractmethod
|
|
|
|
from cbpi.api.base import CBPiBase
|
|
|
|
__all__ = ["StepResult", "StepState", "StepMove", "CBPiStep"]
|
|
|
|
from enum import Enum
|
|
|
|
logging.basicConfig(format='%(asctime)s,%(msecs)d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s',
|
|
datefmt='%Y-%m-%d:%H:%M:%S',
|
|
level=logging.INFO)
|
|
|
|
|
|
|
|
class StepResult(Enum):
|
|
STOP = 1
|
|
NEXT = 2
|
|
DONE = 3
|
|
ERROR = 4
|
|
|
|
|
|
class StepState(Enum):
|
|
INITIAL = "I"
|
|
DONE = "D"
|
|
ACTIVE = "A"
|
|
ERROR = "E"
|
|
STOP = "S"
|
|
|
|
|
|
class StepMove(Enum):
|
|
UP = -1
|
|
DOWN = 1
|
|
|
|
|
|
class CBPiStep(CBPiBase):
|
|
|
|
def __init__(self, cbpi, id, name, props, on_done) -> None:
|
|
self.name = name
|
|
self.cbpi = cbpi
|
|
self.id = id
|
|
self.timer = None
|
|
self._done_callback = on_done
|
|
self.props = props
|
|
self.cancel_reason: StepResult = None
|
|
self.summary = ""
|
|
self.task = None
|
|
self.running: bool = False
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
def _done(self, task):
|
|
if self._done_callback is not None:
|
|
try:
|
|
result = task.result()
|
|
self._done_callback(self, result)
|
|
except Exception as e:
|
|
self.logger.error(e)
|
|
|
|
async def start(self):
|
|
self.logger.info("Start {}".format(self.name))
|
|
self.running = True
|
|
self.task = asyncio.create_task(self._run())
|
|
self.task.add_done_callback(self._done)
|
|
|
|
async def next(self):
|
|
self.running = False
|
|
self.cancel_reason = StepResult.NEXT
|
|
self.task.cancel()
|
|
await self.task
|
|
|
|
async def stop(self):
|
|
try:
|
|
self.running = False
|
|
if self.task is not None and self.task.done() is False:
|
|
self.cancel_reason = StepResult.STOP
|
|
self.task.cancel()
|
|
await self.task
|
|
except Exception as e:
|
|
self.logger.error(e)
|
|
|
|
async def reset(self):
|
|
pass
|
|
|
|
async def on_props_update(self, props):
|
|
self.props = {**self.props, **props}
|
|
|
|
async def save_props(self):
|
|
await self.cbpi.step.save()
|
|
|
|
async def push_update(self):
|
|
self.cbpi.step.push_udpate()
|
|
|
|
async def on_start(self):
|
|
pass
|
|
|
|
async def on_stop(self):
|
|
pass
|
|
|
|
async def _run(self):
|
|
try:
|
|
await self.on_start()
|
|
await self.run()
|
|
self.cancel_reason = StepResult.DONE
|
|
except asyncio.CancelledError as e:
|
|
pass
|
|
finally:
|
|
await self.on_stop()
|
|
|
|
return self.cancel_reason
|
|
|
|
@abstractmethod
|
|
async def run(self):
|
|
pass
|
|
|
|
def __str__(self):
|
|
return "name={} props={}, type={}".format(self.name, self.props, self.__class__.__name__)
|
|
|
|
class CBPiFermentationStep(CBPiStep):
|
|
|
|
def __init__(self, cbpi, fermenter, step, props, on_done) -> None:
|
|
self.fermenter = fermenter
|
|
super().__init__(cbpi, step.id, step.name, props, on_done) |