endpoints.py

from flask import Flask, jsonify
import json
import urllib.parse

app = Flask(__name__)

with open("mcdonalds_menu_data.json", "r", encoding="utf-8") as f:
    data = json.load(f)

def normalize(s):
    return s.lower().replace('"', '').strip()

@app.route("/all_products/", methods=["GET"])
def get_all_products():
    return jsonify(data)

@app.route("/all_products/<product_name>", methods=["GET"])
def search_products(product_name):
    decoded = urllib.parse.unquote(product_name)
    norm = normalize(decoded)

    matched = {k: v for k, v in data.items() if norm in normalize(k)}

    if not matched:
        return jsonify({"error": "Продукт не знайдено"}), 404

    if len(matched) == 1:
        # Якщо один збіг — повертаємо повну інформацію про продукт
        return jsonify(next(iter(matched.values())))

    # Якщо кілька збігів — повертаємо тільки назви продуктів
    return jsonify({"matches": list(matched.keys())})

@app.route("/all_products/<product_name>/<product_field>", methods=["GET"])
def get_product_field(product_name, product_field):
    decoded_name = urllib.parse.unquote(product_name)
    decoded_field = urllib.parse.unquote(product_field)
    norm_name = normalize(decoded_name)
    norm_field = normalize(decoded_field)

    # Частковий збіг по назві продукту (v - value, k - key)
    matched = {k: v for k, v in data.items() if norm_name in normalize(k)} 


    if not matched:
        return jsonify({"error": "Продукт не знайдено"}), 404

    if len(matched) > 1:
        return jsonify({
            "error": "Знайдено кілька продуктів, уточніть назву",
            "matches": list(matched.keys())
        }), 400

    product = next(iter(matched.values()))
    nutrients = product.get("нутрієнти", {})
    if not nutrients:
        return jsonify({"error": "Немає інформації про нутрієнти"}), 404

    # Точний збіг по назві поля всередині нутрієнтів
    for field_key, field_val in nutrients.items():
        if normalize(field_key) == norm_field:
            return jsonify({field_key: field_val})

    return jsonify({
        "error": f"Поле '{decoded_field}' не знайдено у нутрієнтах продукту '{decoded_name}'",
        "available_fields": list(nutrients.keys())
    }), 404


if __name__ == "__main__":
    app.run(debug=True)