mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-09-19 22:07:28 +02:00
230 lines
6.6 KiB
Python
230 lines
6.6 KiB
Python
from cbpi.api.config import ConfigType
|
|
from enum import Enum
|
|
from typing import Any
|
|
from cbpi.api.step import StepState
|
|
from dataclasses import dataclass, field
|
|
from typing import List
|
|
|
|
class Props:
|
|
|
|
def __init__(self, data={}):
|
|
super(Props, self).__setattr__('__data__', {})
|
|
for key, value in data.items():
|
|
self.__setattr__(key, value)
|
|
def __getattr__(self, name):
|
|
return self.__data__.get(name)
|
|
|
|
def __setattr__(self, name, value):
|
|
self.__data__[name] = value
|
|
|
|
def __str__(self):
|
|
return self.__data__.__str__()
|
|
|
|
|
|
def __getitem__(self, key):
|
|
return self.__data__[key]
|
|
|
|
def __setitem__(self, key, value):
|
|
self.__data__[key] = value
|
|
|
|
def __contains__(self, key):
|
|
return key in self.__data__
|
|
|
|
def get(self, key, d=None):
|
|
if key in self.__data__ and self.__data__[key] != "":
|
|
return self.__data__[key]
|
|
else:
|
|
return d
|
|
|
|
|
|
def to_dict(self):
|
|
def parse_object(value):
|
|
if isinstance(value, Props):
|
|
return value.to_dict()
|
|
elif isinstance(value, list):
|
|
return list(map(parse_object, value))
|
|
else:
|
|
return value
|
|
|
|
return dict((key, parse_object(value)) for (key, value) in self.__data__.items())
|
|
|
|
|
|
@dataclass
|
|
class Actor:
|
|
id: str = None
|
|
name: str = None
|
|
props: Props = Props()
|
|
state: bool = False
|
|
power: int = 100
|
|
maxoutput: int = 100
|
|
output: int = maxoutput
|
|
timer: int = 0
|
|
type: str = None
|
|
instance: str = None
|
|
|
|
def __str__(self):
|
|
return "name={} props={}, state={}, type={}, power={}, output={}, maxoutput={}, timer={}".format(self.name, self.props, self.state, self.type, self.power, self.output, self.maxoutput, self.timer)
|
|
def to_dict(self):
|
|
return dict(id=self.id, name=self.name, type=self.type, props=self.props.to_dict(), state=self.instance.get_state(), power=self.power, output=self.output, maxoutput=self.maxoutput, timer=self.timer)
|
|
|
|
class DataType(Enum):
|
|
VALUE="value"
|
|
DATETIME="datetime"
|
|
STRING="string"
|
|
|
|
@dataclass
|
|
class Sensor:
|
|
id: str = None
|
|
name: str = None
|
|
props: Props = Props()
|
|
state: bool = False
|
|
type: str = None
|
|
instance: str = None
|
|
datatype: DataType = DataType.VALUE
|
|
inrange: bool = True
|
|
temp_range: float = 0
|
|
|
|
def __str__(self):
|
|
return "name={} props={}, state={}".format(self.name, self.props, self.state)
|
|
def to_dict(self):
|
|
return dict(id=self.id, name=self.name, type=self.type, props=self.props.to_dict(), state=self.state)
|
|
|
|
@dataclass
|
|
class Kettle:
|
|
id: str = None
|
|
name: str = None
|
|
props: Props = Props()
|
|
instance: str = None
|
|
agitator: Actor = None
|
|
heater: Actor = None
|
|
sensor: Sensor = None
|
|
type: str = None
|
|
target_temp: float = 0
|
|
|
|
def __str__(self):
|
|
return "name={} props={} temp={}".format(self.name, self.props, self.target_temp)
|
|
def to_dict(self):
|
|
|
|
if self.instance is not None:
|
|
|
|
state = self.instance.state
|
|
|
|
else:
|
|
state = False
|
|
return dict(id=self.id, name=self.name, state=state, target_temp=self.target_temp, heater=self.heater, agitator=self.agitator, sensor=self.sensor, type=self.type, props=self.props.to_dict())
|
|
|
|
|
|
|
|
@dataclass
|
|
class Step:
|
|
id: str = None
|
|
name: str = None
|
|
props: Props = Props()
|
|
type: str = None
|
|
status: StepState = StepState.INITIAL
|
|
instance: str = None
|
|
|
|
def __str__(self):
|
|
return "name={} props={}, type={}, instance={}".format(self.name, self.props, self.type, self.instance)
|
|
def to_dict(self):
|
|
|
|
msg = self.instance.summary if self.instance is not None else ""
|
|
return dict(id=self.id, name=self.name, state_text=msg, type=self.type, status=self.status.value, props=self.props.to_dict())
|
|
|
|
@dataclass
|
|
class Fermenter:
|
|
id: str = None
|
|
name: str = None
|
|
sensor: Sensor = None
|
|
pressure_sensor : Sensor = None
|
|
heater: Actor = None
|
|
cooler: Actor = None
|
|
valve: Actor = None
|
|
brewname: str = None
|
|
description : str = None
|
|
props: Props = Props()
|
|
target_temp: float = 0
|
|
target_pressure: float = 0
|
|
type: str = None
|
|
steps: List[Step]= field(default_factory=list)
|
|
instance: str = None
|
|
|
|
|
|
def __str__(self):
|
|
return "name={} props={} temp={}".format(self.name, self.props, self.target_temp, self.target_pressure)
|
|
|
|
def to_dict(self):
|
|
|
|
if self.instance is not None:
|
|
|
|
state = self.instance.state
|
|
|
|
else:
|
|
state = False
|
|
|
|
steps = list(map(lambda item: item.to_dict(), self.steps))
|
|
return dict(id=self.id, name=self.name, state=state, sensor=self.sensor, pressure_sensor=self.pressure_sensor, heater=self.heater, cooler=self.cooler, valve=self.valve, brewname=self.brewname, description=self.description, props=self.props.to_dict() if self.props is not None else None, target_temp=self.target_temp, target_pressure=self.target_pressure, type=self.type, steps=steps)
|
|
|
|
|
|
@dataclass
|
|
class FermenterStep:
|
|
id: str = None
|
|
name: str = None
|
|
fermenter: Fermenter = None
|
|
props: Props = Props()
|
|
type: str = None
|
|
status: StepState = StepState.INITIAL
|
|
endtime: int = 0 # endtime if step is active and timer is running
|
|
instance: str = None
|
|
step: dict = None
|
|
|
|
def __str__(self):
|
|
return "name={} props={}, type={}, instance={}".format(self.name, self.props, self.type, self.instance)
|
|
def to_dict(self):
|
|
msg = self.instance.summary if self.instance is not None else ""
|
|
return dict(id=self.id, name=self.name, state_text=msg, type=self.type, status=self.status.value, endtime=self.endtime, props=self.props.to_dict())
|
|
|
|
|
|
|
|
class ConfigType(Enum):
|
|
STRING="string"
|
|
ACTOR="actor"
|
|
SENSOR="sensor"
|
|
KETTLE="kettle"
|
|
NUMBER="number"
|
|
SELECT="select"
|
|
STEP="step"
|
|
FERMENTER="fermenter"
|
|
|
|
@dataclass
|
|
class Config:
|
|
|
|
name: str = None
|
|
value: Any = None
|
|
description: str = None
|
|
type: ConfigType = ConfigType.STRING
|
|
source: str = None
|
|
options: Any = None
|
|
|
|
def __str__(self):
|
|
return "....name={} value={}".format(self.name, self.value)
|
|
def to_dict(self):
|
|
return dict(name=self.name, value=self.value, type=self.type.value, description=self.description, source=self.source, options=self.options)
|
|
|
|
@dataclass
|
|
class NotificationAction:
|
|
label: str
|
|
method: Any = None
|
|
id: str = None
|
|
|
|
def to_dict(self):
|
|
return dict(id=self.id, label=self.label)
|
|
|
|
class NotificationType(Enum):
|
|
INFO="info"
|
|
WARNING="warning"
|
|
ERROR="error"
|
|
SUCCESS="success"
|
|
|
|
def __str__(self):
|
|
return self.value
|