diff --git a/lib/data_structures.py b/lib/data_structures.py index 1c3cb6ff4..be95fcf65 100644 --- a/lib/data_structures.py +++ b/lib/data_structures.py @@ -17,22 +17,40 @@ ] def get_names(spicy_foods): - pass + return [food["name"] for food in spicy_foods] def get_spiciest_foods(spicy_foods): - pass + return [food for food in spicy_foods if food["heat_level"] >= 5] def print_spicy_foods(spicy_foods): - pass + for food in spicy_foods: + heat_level_emojis = "🌶" * food["heat_level"] + print(f"{food['name']} ({food['cuisine']}) | Heat Level: {heat_level_emojis}") def get_spicy_food_by_cuisine(spicy_foods, cuisine): - pass + for food in spicy_foods: + if food["cuisine"] == cuisine: + return food + return None + def print_spiciest_foods(spicy_foods): - pass + for food in spicy_foods: + if food["heat_level"] > 5: + heat_level_emojis = "🌶" * food["heat_level"] + print( + f"{food['name']} ({food['cuisine']}) | Heat Level: {heat_level_emojis}" + ) + def get_average_heat_level(spicy_foods): - pass + total_heat_level = sum(food["heat_level"] for food in spicy_foods) + num_spicy_food = len(spicy_foods) + if num_spicy_food == 0: + return 0 + return total_heat_level // num_spicy_food + def create_spicy_food(spicy_foods, spicy_food): - pass + spicy_foods.append(spicy_food) + return spicy_foods