Meal-Picker/src/MealBuilder.py

from FileHandler import read_json, update_json

import flet as ft

meal_json = read_json()

def get_meal_names():
    meal_list = list(meal_json.keys())
    meal_list.sort()
    return meal_list

def is_number(s):
    try:
        float(s)
        return True
    except ValueError:
        return False


def builder(page):
    class Meal():
        
        def get_meal_radios(self):
            self.meal_radios = []
            for name in get_meal_names():
                self.meal_radios.append(
                        ft.Radio(value=name, label=name),
                    )

        def get_meal_radios_group(self):
            return ft.RadioGroup(
                    content=ft.Column(
                    controls=self.meal_radios,
                    )
                )
    
        def update_ingredient(self, e):
            self.new_ingredient = e.control.value
    
        def update_quantity(self, e):
            self.new_quantity = e.control.value

        def update_units(self, e):
            self.new_units = e.control.value

        def create_ingredient_row(self, ingredient, details):
            row = []
            row.append(ft.TextField(label="Ingredient", value=ingredient, on_change=self.update_ingredient))
            row.append(ft.TextField(label="Quantity", value=details['quantity'], on_change=self.update_quantity))
            row.append(ft.TextField(label="Units (optional)", value=details['units'], on_change=self.update_units))
            return row

        def create_new_ingredient_row(self):
            row = []
            row.append(ft.TextField(label="Ingredient", value=""))
            row.append(ft.TextField(label="Quantity", value=""))
            row.append(ft.TextField(label="Units (optional)", value=""))
            return row
    
        def update_ingredients(self, name):
            pass
            # Doesn't work yet

            # if not self.new_ingredient or not self.new_quantity or not is_number(self.new_quantity):
            #     return

            # ingredient = {
            #     self.new_ingredient: 
            #         {
            #             "quantity":float(self.new_quantity), 
            #             "units": self.new_units if self.new_units != "" else None
            #         }
            #     }
            # meal_json[name] = ingredient
            # update_json(meal_json)
            # page.update()

        def show_new_meal(self):
            self.new_meal = {}
            selector_body.controls = [
                ft.TextField(label="Meal", value="", autofocus=True),
                ft.ListView(controls=[ft.Row(self.create_new_ingredient_row(), 
                            alignment=ft.MainAxisAlignment.SPACE_EVENLY,
                            width=300,
                            height=100,
                            )],
                            expand=True,
                            spacing=5,
                            padding=5,
                            auto_scroll=False,
                        ),
                ft.ElevatedButton(text="Add Ingredient", on_click=lambda e: self.append_new_ingredient_row(selector_body)),
                ft.ElevatedButton(text="Add Meal", on_click=lambda e: self.add_new_meal(selector_body)),
                ft.ElevatedButton(text="Back", on_click=lambda e: self.show_meal_selection(selector_body, page))
            ]
            
            page.title = "Add New Meal"
            page.update()
        
        def append_new_ingredient_row(self, selector_body):
            selector_body.controls[1].controls.append(ft.Row(self.create_new_ingredient_row(), 
                            alignment=ft.MainAxisAlignment.SPACE_EVENLY,
                            width=300,
                            height=100,))
            
            page.update()

        def add_new_meal(self, selector_body):
            meal = selector_body.controls[0].value
            ing, qua, uni = [], [], []
            
            for row in selector_body.controls[1].controls:
                #skip blank row
                if row.controls[0].value == "" or row.controls[1].value == "":
                    continue
                ing.append(row.controls[0].value)
                qua.append(row.controls[1].value)
                uni.append(row.controls[2].value if row.controls[2].value != "" else None)

            meal_json[meal] = {}
            for i in range(len(ing)):
                meal_json[meal][ing[i]] = { 'quantity': qua[i], 'units': uni[i] }

            #write changes to the ingredients.json file
            update_json(meal_json)
            self.show_new_meal()

        def show_meal_details(self, selector_body, page):
            expanded_meal = []
            
            meal_name = selector_body.controls[0].controls[0].value

            for details in meal_json[meal_name].items():
                expanded_meal.append(
                    ft.Row(self.create_ingredient_row(details[0], details[1]), 
                            alignment=ft.MainAxisAlignment.SPACE_EVENLY,
                            width=300,
                            height=100,
                        )
                    )
                
            selector_body.controls = [
                        ft.TextField(label="Meal", value=meal_name),
                        ft.ListView(
                            controls=expanded_meal,
                            expand=True,
                            spacing=5,
                            padding=5,
                            auto_scroll=False,
                            ),
                        ft.ElevatedButton(text="Update Meal", on_click=lambda e: self.update_ingredients(selector_body.controls[0].value)),
                        ft.ElevatedButton(text="Back", on_click=lambda e: self.show_meal_selection(selector_body, page)),
                        ]
            if page.title == "Create and Edit Meals!":
                page.title = f"Editing {meal_name}"
                page.update()

            return selector_body
    
        def show_meal_selection(self, selector_body, page):
            self.get_meal_radios()
            selector_body.controls = [ft.ListView(
                        controls=[
                                self.get_meal_radios_group(),
                            ],
                        expand=True,
                        spacing=5,
                        padding=5,
                        auto_scroll=False,
                        ),
                        ft.ElevatedButton(text="Update Meal", 
                                          on_click=lambda e: self.show_meal_details(selector_body, page)),
                        ft.ElevatedButton(text="Add Meal", on_click=lambda e: self.show_new_meal())
                        ]

            if page.title != "Create and Edit Meals!":
                page.title = f"Create and Edit Meals!"
                page.update()

            return selector_body

        

    meal = Meal()
    page.title = "Create and Edit Meals!"
   
    selector_body = ft.Column(
        controls = [],
        height = 500,
        expand = False,
    )

    page.controls[0].content = meal.show_meal_selection(selector_body, page)
    page.update()
    
    # TODO add functionality to add ingredients to a meal
    # TODO (Possible) add functionality to delete meals and/or ingredients

if __name__ == "__main__":
    ft.app(builder)