Files
Meal-Picker/src/FileHandler.py
2025-12-17 23:16:29 -05:00

88 lines
3.2 KiB
Python

from pathlib import Path
import json
working_directory = Path(__file__).parent
try:
with open(str(working_directory / "settings.json"), "rt") as settings_file:
settings = json.load(settings_file)
checklist_file_path = settings.get("checklist_storage_path", str(working_directory / "json")) + "/checklists.json"
ingredients_file_path = settings.get("meal_storage_path", str(working_directory / "json")) + "/ingredients.json"
except Exception as e:
print(e)
with open(str(working_directory / "settings.json"), "w") as settings_file:
settings = {
"meal_storage_path": str(working_directory / "json"),
"checklist_storage_path": str(working_directory / "json")
}
json.dump(settings, settings_file, indent=4)
checklist_file_path = settings["checklist_storage_path"] + "/checklists.json"
ingredients_file_path = settings["meal_storage_path"] + "/ingredients.json"
def read_ingredient_json() -> dict:
try:
with open(ingredients_file_path, "rt") as file:
return json.load(file)
except:
print("Could not find or read ingredients file.")
return {}
def read_json_file(filename: str) -> dict:
try:
with open(filename, "rt") as file:
return json.load(file)
except:
print(f"Could not find or read file.\n{filename=}")
return {}
def update_json(meals: dict) -> None:
try:
with open(ingredients_file_path, "w") as file:
json.dump(meals, file, indent=4)
except Exception as e:
print(f"Unable to write data to file. {e}")
def update_file(filename: str, content: dict) -> None:
try:
with open(filename, "w") as file:
json.dump(content, file, indent=4)
except Exception as e:
print(f"Unable to write data to file. {e}")
def get_selected_meals(meals: dict) -> dict:
meal_json = read_ingredient_json()
selected_meals = {}
if meal_json:
for meal in meals:
if meal in meal_json:
selected_meals[meal] = meal_json[meal]
return selected_meals
def combine_ingredients(meals: dict) -> dict:
combined_ingredients = {}
for meal, ingredients in meals.items():
for ingredient, detail in ingredients.items():
if ingredient in combined_ingredients:
combined_ingredients[ingredient]['quantity'] = str(float(combined_ingredients[ingredient]['quantity']) + float(detail['quantity']))
else:
combined_ingredients[ingredient] = detail
return combined_ingredients
def write_checklist(data: dict) -> None:
try:
with open(checklist_file_path, 'w') as checklist:
for ingredient, details in data.items():
s = ingredient + " - " + str(details['quantity'])
if details['units'] != None:
s += " " + details['units']
checklist.write(f"- [ ] {s}\n")
except Exception as e:
print(f"Unable to write data to file. {e}")
if __name__ == "__main__":
meals = read_ingredient_json()
if meals:
combined_ingredients = combine_ingredients(meals)
write_checklist(combined_ingredients)