from library.librerias import *
from models.detalle_pedidos import DetallePedidos

router = APIRouter(
    prefix="/detalle_pedidos",
    tags=["Detalle_pedidos"],
    responses={404: {"description": "No encontrado"}},
)

"""
CREATE TABLE detalle_pedidos (
  ID_Pedido INTEGER,
  ID_Insumo INTEGER,
  Cantidad INTEGER,
  Costo_Unitario REAL,
  PRIMARY KEY (ID_Pedido, ID_Insumo),
  FOREIGN KEY (ID_Pedido) REFERENCES pedidos_proveedor (ID_Pedido),
  FOREIGN KEY (ID_Insumo) REFERENCES insumos (ID_Insumo)
);

"""


# get/detalle_pedidos -> obtener todos los detalle_pedidos con with connection as conn, manejo de errores
@router.get("/")
def get_detalle_pedidos():
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute("SELECT * FROM detalle_pedidos")
            detalle_pedidos = cursor.fetchall()
            return detalle_pedidos
    except Exception as e:
        print(e)
        return []


# post/detalle_pedidos -> crear un detalle_pedido con with connection as conn, manejo de errores
@router.post("/")
def post_detalle_pedido(detalle_pedido: DetallePedidos):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute(
                "INSERT INTO detalle_pedidos (ID_Pedido, ID_Insumo, Cantidad, Costo_Unitario) VALUES (?, ?, ?, ?)",
                (
                    detalle_pedido.ID_Pedido,
                    detalle_pedido.ID_Insumo,
                    detalle_pedido.Cantidad,
                    detalle_pedido.Costo_Unitario,
                ),
            )
            conn.commit()
            return {"message": "Detalle_pedido creado"}
    except Exception as e:
        print(e)
        return []