96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
from __future__ import annotations
|
|
from copy import deepcopy
|
|
|
|
from enum import Enum
|
|
|
|
from ..draw_utils import draw_clock
|
|
|
|
class Recipe:
|
|
def __init__(self, name: str, steps: list[Step]):
|
|
self.name = name
|
|
self.steps = steps
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def copy(self) -> Recipe:
|
|
return deepcopy(self)
|
|
|
|
class StepType(Enum):
|
|
SECTION = 0
|
|
WEIGH = 1
|
|
WEIGH_WITH_TIMER = 2
|
|
TARE = 3
|
|
|
|
def render(self, draw, position, fill='black', **kwargs) -> str:
|
|
font_size = kwargs.get('font_size', 10)
|
|
if self == StepType.SECTION:
|
|
draw.text(position, "T", fill=fill, **kwargs)
|
|
elif self == StepType.WEIGH:
|
|
draw.text(position, "W", fill=fill, **kwargs)
|
|
elif self == StepType.WEIGH_WITH_TIMER:
|
|
draw.text(position, "W", fill=fill, **kwargs)
|
|
r = max(3, font_size // 4)
|
|
draw_clock(draw, (position[0] + font_size + 2, position[1] + font_size // 2), radius=r, color=fill)
|
|
elif self == StepType.TARE:
|
|
draw.text(position, "0.0g", fill=fill, **kwargs)
|
|
|
|
class Step:
|
|
|
|
def __init__(self,
|
|
step_type: StepType,
|
|
value: float | str = None,
|
|
goal_time: float = 0.0):
|
|
self.step_type = step_type
|
|
self.value = value
|
|
self.goal_time = goal_time
|
|
|
|
@property
|
|
def value_str(self) -> str:
|
|
if self.step_type == StepType.WEIGH:
|
|
return f"{self.value}g"
|
|
elif self.step_type == StepType.WEIGH_WITH_TIMER:
|
|
s = f"{self.value}g"
|
|
if self.goal_time > 0:
|
|
minutes = int(self.goal_time) // 60
|
|
seconds = int(self.goal_time) % 60
|
|
if minutes == 0:
|
|
s += f" {seconds}s"
|
|
else:
|
|
s += f" {minutes}:{seconds:02d}"
|
|
return s
|
|
elif self.step_type == StepType.SECTION:
|
|
return str(self.value)
|
|
elif self.step_type == StepType.TARE:
|
|
return "Tare"
|
|
else:
|
|
return ""
|
|
|
|
|
|
###### example recipes #########
|
|
V60 = Recipe(
|
|
"V60",
|
|
[
|
|
Step(StepType.SECTION, "Grind"),
|
|
Step(StepType.TARE),
|
|
Step(StepType.WEIGH, 15),
|
|
|
|
Step(StepType.SECTION, "Brew"),
|
|
Step(StepType.TARE),
|
|
Step(StepType.WEIGH_WITH_TIMER, 50, goal_time=45),
|
|
Step(StepType.WEIGH, 250),
|
|
]
|
|
)
|
|
|
|
ESPRESSO = Recipe(
|
|
"Espresso",
|
|
[
|
|
Step(StepType.SECTION, "Grind"),
|
|
Step(StepType.TARE),
|
|
Step(StepType.WEIGH, 18),
|
|
|
|
Step(StepType.SECTION, "Brew"),
|
|
Step(StepType.TARE),
|
|
Step(StepType.WEIGH_WITH_TIMER, 40, goal_time=0.0),
|
|
]
|
|
) |