start implementing edit_recipe

2025-10-18 23:28:55 +02:00
parent 06df2e0e9b
commit eb32f089ad
12 changed files with 241 additions and 70 deletions
--- a/frontend/views/recipes/edit_recipe.py
+++ b/frontend/views/recipes/edit_recipe.py
@@ -0,0 +1,104 @@
+from typing import Tuple
+from PIL import ImageDraw, Image
+
+from ..base import View
+from ..button_interface import ButtonInterface
+
+from .recipe_manager import RecipeManager
+from .recipe import Recipe
+
+class EditRecipe(View, ButtonInterface):
+    def __init__(self, parent, im_size, center,
+                 recipe_manager: RecipeManager,
+                 deactivate_command,
+                 recipe: Recipe = None):
+        self.deactivate_command = deactivate_command
+        self.recipe_manager = recipe_manager
+        self.recipe = recipe
+        if recipe is None:
+            self.recipe = Recipe("New", [])
+        
+        self.confirm_view = False
+        self.selected_field = 0  # 0: name, 1+: steps
+
+        super().__init__(parent, im_size, center)
+    
+    def _get_visual_steps(self):
+        steps = self.recipe.steps + ['+']
+        if len(steps) < 4:
+            return steps, 0
+        
+        start = max(0, self.selected_field - 2)
+        end = min(len(steps), start + 4)
+
+        steps = steps[start:end]
+
+        return steps, start
+
+    def update_weight(self, weight: float) -> Image.Image:
+        im = self.bkg_im.copy()
+        draw = ImageDraw.Draw(im)
+
+        x = 40
+
+        draw.text((x, 10), "Name:", fill='black')
+        if self.selected_field == 0:
+            r = 10
+            offset = 35
+            for i in range(0, 90, r // 2):
+                draw.circle((x + i, offset), r, fill='black')
+            draw.text((x, 30), self.recipe.name, fill='white')
+        else:
+            draw.text((x, 30), self.recipe.name, fill='black')
+
+        visual_steps, start = self._get_visual_steps()
+
+        for idx, step in enumerate(visual_steps):
+            y_pos = 60 + idx * 20
+            if start + idx + 1 == self.selected_field:
+                r = 10
+                offset = 15
+                for i in range(0, 90, r // 2):
+                    draw.circle((x + i, y_pos), r, fill='black')
+                if str(step) != '+':
+                    step.step_type.render(draw, (x, y_pos - 5), fill='white')
+                    draw.text((x + 30, y_pos - 5), step.value_str, fill='white')
+                else:
+                    draw.text((x, y_pos - 5), '+', fill='white')
+            elif str(step) == '+':
+                draw.text((x, y_pos - 5), '+', fill='black')
+            else:
+                step.step_type.render(draw, (x, y_pos - 5), fill='black')
+                draw.text((x + 30, y_pos - 5), step.value_str, fill='black')
+
+        return im
+
+    def left_press(self):
+        # edit entry
+        pass
+    
+    def left_long_press(self):
+        # save
+        self.recipe_manager.add_recipe(self.recipe)
+        self.deactivate_command()
+
+    def right_press(self):
+        self.selected_field += 1
+        if self.selected_field > len(self.recipe.steps) + 1:
+            self.selected_field = 0
+
+    def has_button(self) -> Tuple[bool, bool, bool, bool]:
+        return True, True, True, True
+
+
+    def render_left_press(self, draw, x, y):
+        draw.regular_polygon((x, y, 5), 3, fill='black', rotation=270)
+
+    def render_left_long_press(self, draw, x, y):
+        draw.text((x - 3, y - 5), 'Save', fill='black')
+
+    def render_right_press(self, draw, x, y):
+        draw.regular_polygon((x, y + 6, 5), 3, fill='black', rotation=180)
+
+    def render_right_long_press(self, draw, x, y):
+        draw.text((x - 30, y - 5), 'Cancel', fill='black')