resto

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
Script que implemente un menu para administrar un restaurante. Vamos a
implementar dos tablas: categorias y platos
EJ: categorias: ensaladas, entradas, guarniciones, plato principal, postre
platos: milanesa, ravioles, fideos, flan, empanada, ensalada cesar, etc.
Vamos a hacer un menu que implemente por ejemplo:
 1. Agregar categoria
 2. Agregar plato
 3. Mostrar menu
 4. Salir

"""
import sqlite3
from tabulate import tabulate

def menu_de_opciones():
    return """
        Menú de categorías y platos
    *--------------------------------*
        1. Agregar categoría
        2. Agregar plato
        3. Mostrar menú
        4. Salir
    *--------------------------------*
    """

def agregar_categoria():
    nombre = input("Ingrese el nombre de la categoría: ")
    try:
        conn.execute("INSERT INTO categoria VALUES(null,?)",(nombre,))
    except sqlite3.IntegrityError:
        print(f"Error: ya existe la categoría '{nombre}'")
    else:
        conn.commit()



def agregar_plato():
    tabla = []
    try:
        cursor.execute("SELECT * FROM categoria ORDER BY id")
    except sqlite3.OperationalError:
        print("No se pudo ejecutar la consulta")
    else:
        categorias = cursor.fetchall()
        if categorias:
            id_categoria = [] # almacenará los id de las categorías YA CREADAS
            for categoria in categorias:
                tabla.append(categoria)
                id_categoria.append(categoria[0])
            print(tabulate(
                tabla,
                headers = ["ID de categoría", "categoría"],
                tablefmt = "grid",
                colalign = ["center", "left"]
                )
            )
            # valido que el usuario ingrese un id de categoría válido
            while True:
                try:
                    ID = int(input("\nSeleccione un id: "))
                except ValueError:
                    print("El ID debe ser un nro entero")
                else:
                    if ID in id_categoria:
                        break
                    print(f"No existe el id '{ID}'")

            # Creo el plato
            nombre_plato = input("Ingrese el nombre del plato: ")
            try:
                conn.execute("INSERT INTO platos VALUES(null,?,?)",(nombre_plato,ID))
            except sqlite3.IntegrityError:
                print(f"Error: ya existe el plato '{nombre_plato}'")
            else:
                conn.commit()
        else:
            print("No existen categorías de platos")



def menu():
    tabla = []
    query = "SELECT categoria.id, categoria.nombre, platos.id, platos.nombre FROM categoria\
            LEFT JOIN platos\
            ON platos.id_categoria = categoria.id\
            ORDER BY categoria.id, platos.id"

    cursor.execute(query)
    platos = cursor.fetchall()
    if platos:
        for plato in platos:
            tabla.append(plato)
        return tabulate(
                    tabla,
                    headers = ["ID de categoría", "Categoría", "ID del plato", "Plato"],
                    tablefmt = "grid",
                    colalign = ["center", "left", "center", "left"]
                    )
    else:
        return "No existe ningún plato en el menú"



while True:
    conn = sqlite3.connect("restaurante.db")
    cursor = conn.cursor()
    print(menu_de_opciones())

    opcion = input("Ingrese su opción: ")

    if opcion == "1":
        agregar_categoria()

    elif opcion == "2":
        agregar_plato()

    elif opcion == "3":
        print(menu())

    elif opcion == "4":
        conn.close()
        print("Hasta luego...")
        break

    else:
        print("Opción incorrecta")