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

import sqlite3

def mostrar_opciones():
	return """
		Menu de administración del menu del restaurante
	*---------------------------------------------------*
		1. Agregar categoria
		2. Agregar plato
		3. Mostrar menu
		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}'")
	except sqlite3.OperationalError:
		print("La consulta no se ejecutó correctamente")
	else:
		conn.commit()
	

def agregar_plato():
	try:
		cursor.execute("SELECT * FROM categoria ORDER BY id")
	except sqlite3.OperationalError:
		print("La consulta no se ejecutó correctamente")
	else:
		categorias = cursor.fetchall() # vuelco todos los datos
		if categorias:
			id_categoria = [] # creo una lista vacía para almacenar id de categorías
			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 un numero entero...")
				else:
					if ID in id_categoria:
						break
					print("ID de categoría inexistente..")
					
			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}'")
			except sqlite3.OperationalError:
				print("La consulta no se ejecutó correctamente")
			else:
				conn.commit()
			
		else:
			print("No hay categorías 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("{:<12}".format(elemento), end=" ")
				elif isinstance(elemento, int):
					print("{:^3}".format(elemento), end=" ")
		print()
	else:
		print("No hay ningún plato en el menú")



conn = sqlite3.connect("restaurante.db")
cursor = conn.cursor()

while True:
	print(mostrar_opciones())
	opcion = input("Seleccione su opción: ")
	
	if opcion == "1":
		agregar_categoria(input("Ingrese el nombre: "))
		
	elif opcion == "2":
		agregar_plato()
		
	elif opcion == "3":
		mostrar_menu()
	
	elif opcion == "4":
		conn.close()
		print("Cerrando conexiones...")
		break
		
	else:
		print("Opción incorrecta")