Spaces:
Running
Running
File size: 2,300 Bytes
224e50f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
from library.librerias import *
from models.maquinas import Maquinas
router = APIRouter(
prefix="/maquinas",
tags=["Maquinas"],
responses={404: {"description": "No encontrado"}},
)
"""CREATE TABLE maquinas (
ID_Maquina INTEGER PRIMARY KEY AUTOINCREMENT,
Tipo TEXT,
Capacidad INTEGER,
Consumo_Energetico REAL
);"""
@router.get("/")
def get_maquinas():
try:
with DatabaseConnection().get_connection() as conn:
cursor = conn.cursor()
cursor.execute("SELECT * FROM maquinas")
maquinas = cursor.fetchall()
return maquinas
except Exception as e:
print(e)
return []
# @router.post("/")
def post_maquina(maquina: Maquinas):
try:
with DatabaseConnection().get_connection() as conn:
cursor = conn.cursor()
cursor.execute(
"INSERT INTO maquinas (Tipo, Capacidad, Consumo_Energetico) VALUES (?, ?, ?)",
(
maquina.Tipo,
maquina.Capacidad,
maquina.Consumo_Energetico,
),
)
conn.commit()
return {"message": "Maquina creada"}
except Exception as e:
print(e)
return []
# @router.put("/")
def put_maquina(maquina: Maquinas):
try:
with DatabaseConnection().get_connection() as conn:
cursor = conn.cursor()
cursor.execute(
"UPDATE maquinas SET Tipo = ?, Capacidad = ?, Consumo_Energetico = ? WHERE ID_Maquina = ?",
(
maquina.Tipo,
maquina.Capacidad,
maquina.Consumo_Energetico,
maquina.ID_Maquina,
),
)
conn.commit()
return {"message": "Maquina actualizada"}
except Exception as e:
print(e)
return []
# search by tipo de maquina
@router.get("/{tipo}")
def get_maquina(tipo: str):
try:
with DatabaseConnection().get_connection() as conn:
cursor = conn.cursor()
cursor.execute("SELECT * FROM maquinas WHERE Tipo = ?", (tipo,))
maquina = cursor.fetchall()
return maquina
except Exception as e:
print(e)
return []
|