menu resto

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script para administrar un restoran. Implementa dos tablas:
* categoria (primeros, segundos, postres)
* platos (milanesa c/pure, ravioles c /estofado, ensalada mixta, flan casero)
"""

import sqlite3

def mostrar_opciones():
    return """
        Menu de administracion del restaurante
    *--------------------------------------------*
        1. Agregar categoría
        2. Agregar plato
        3. Mostrar menú
        4. Salir
    *--------------------------------------------*
    """

def agregar_categoria(nombre):
    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():
    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 = []
            print("Categorias")
            print("----------")
            print("ID | nombre")
            print("-----------")
            for ID, nombre in categorias:
                id_categoria.append(ID)
                print(f"{ID} | {nombre}")
            while True:
                try:
                    ID = int(input("\nSeleccione un id: "))
                except ValueError:
                    print("El id debe ser nro entero")
                else:
                    if ID in id_categoria:
                        break
                    print("ID de categoria inexistente")
            nombre_plato = input("Ingrese el nombre del plato: ")
            try:
                conn.execute("INSERT INTO platos VALUES (null,?,?)", (nombre_plato,ID))
            except sqlite3.OperationalError:
                print(f"Error. Ya existe el plato {nombre_plato}")
            else:
                conn.commit()
        else:
            print("No existen categorias de platos")





def mostrar_menu():
    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:
        print("\nID  categoria  ID  plato   ")
        print("-----------------------------")
        for plato in platos:
            print()
            for elemento in plato:
                if isinstance(elemento, str):
                    print("{:<10}".format(elemento), end=" ")
                else:
                    if elemento is not None:
                        print("{:^3}".format(elemento), end=" ")
        print()
    else:
        print("No hay platos")