routers/pedidos_proveedor.py

from library.librerias import *
from models.pedidos_proveedor import PedidosProveedor

router = APIRouter(
    prefix="/pedidos_proveedor",
    tags=["Pedidos_proveedor"],
    responses={404: {"description": "No encontrado"}},
)
"""CREATE TABLE pedidos_proveedor (
  ID_Pedido INTEGER PRIMARY KEY AUTOINCREMENT,
  ID_Proveedor INTEGER,
  Fecha DATE,
  Total REAL,
  FOREIGN KEY (ID_Proveedor) REFERENCES clientes (ID_Proveedor)
);
"""


@router.get("/")
def get_pedidos_proveedor():
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute("SELECT * FROM pedidos_proveedor")
            pedidos_proveedor = cursor.fetchall()
            return pedidos_proveedor
    except Exception as e:
        print(e)
        return []


@router.post("/")
def post_pedido_proveedor(pedido_proveedor: PedidosProveedor):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute(
                "INSERT INTO pedidos_proveedor (ID_Proveedor, Fecha, Total) VALUES (?, ?, ?)",
                (
                    pedido_proveedor.ID_Proveedor,
                    pedido_proveedor.Fecha,
                    pedido_proveedor.Total,
                ),
            )
            conn.commit()
            return {"message": "Pedido_proveedor creado"}
    except Exception as e:
        print(e)
        return []


@router.put("/")
def put_pedido_proveedor(pedido_proveedor: PedidosProveedor):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute(
                "UPDATE pedidos_proveedor SET ID_Proveedor = ?, Fecha = ?, Total = ? WHERE ID_Pedido = ?",
                (
                    pedido_proveedor.ID_Proveedor,
                    pedido_proveedor.Fecha,
                    pedido_proveedor.Total,
                    pedido_proveedor.ID_Pedido,
                ),
            )
            conn.commit()
            return {"message": "Pedido_proveedor actualizado"}
    except Exception as e:
        print(e)
        return []


@router.delete("/{ID_Pedido}")
def delete_pedido_proveedor(ID_Pedido: int):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute(
                "DELETE FROM pedidos_proveedor WHERE ID_Pedido = ?", (ID_Pedido,)
            )
            conn.commit()
            return {"message": "Pedido_proveedor eliminado"}
    except Exception as e:
        print(e)
        return []


@router.get("/{ID_Pedido}")
def get_pedido_proveedor(ID_Pedido: int):
    try:
        with DatabaseConnection().get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute(
                "SELECT * FROM pedidos_proveedor WHERE ID_Pedido = ?", (ID_Pedido,)
            )
            pedido_proveedor = cursor.fetchall()
            return pedido_proveedor
    except Exception as e:
        print(e)
        return []