update_firestore.py

from google.cloud import firestore
from locations import locations
from menu import menu

def upload_menu_to_firestore():
    # Initialize Firestore client
    db = firestore.Client()

    # reference to location collection
    location_collection = db.collection("locations")
    # create empty locations list
    location_list = []

    # start batch
    batch = db.batch()
    locations_requests = locations()
    for loc in locations_requests:
        curr_mealperiods = []

        # Handle both single-location and multi-location venues
        location_data = locations_requests[loc]
        if isinstance(location_data, tuple):
            # Multi-location venue: (request, category_filter)
            request_obj, category_filter = location_data
            dishes = menu(request_obj, category_filter=category_filter)
        else:
            # Single-location venue: just the request
            dishes = menu(location_data)

        location_list.append(loc)

        # Reference to the collection
        collection_ref = db.collection(loc)

        # Delete existing documents in the collection
        docs = collection_ref.stream()
        for doc in docs:
            doc.reference.delete()

        # Prepare data for bulk insert
        data_to_insert = []
        for mealperiod in dishes:
            for dish in dishes[mealperiod]:
                dish_data = dishes[mealperiod][dish]
                data_to_insert.append({
                    "mealperiod": mealperiod,
                    "category": dish_data["category"],
                    "name": dish.strip(),
                    "calories": dish_data["Calories (kcal)"],
                    "fat": dish_data["Total Lipid/Fat (g)"],
                    "carbs": dish_data["Carbohydrate (g)"],
                    "protein": dish_data["Protein (g)"],
                    "sugar": dish_data["Sugar (g)"],
                    "servingSize": dish_data["servingSize"]
                })
            curr_mealperiods.append(mealperiod)
        
        # Batch upload data to Firestore
        for data in data_to_insert:
            doc_ref = collection_ref.document()
            batch.set(doc_ref, data)
        batch.set(collection_ref.document("mealperiods"), {"periods": curr_mealperiods})
    # update locations collection
    batch.set(location_collection.document("locations"), {"names": location_list})
    batch.commit()

#upload_menu_to_firestore()