Update main.py

main.py

@@ -30,6 +30,37 @@ app.add_middleware(
 def read_root():
     return {"message": "Welcome to the Arduino sensor data API"}
 
+
+# Modelo para recibir consultas SQL
+class SQLQuery(BaseModel):
+    query: str
+
+# Conexión a la base de datos SQLite
+def get_db_connection():
+    conn = sqlite3.connect('database/anhelados.db')
+    conn.row_factory = sqlite3.Row
+    return conn
+
+@app.post("/execute", summary="Ejecutar consulta SQL", description="Ejecuta una consulta SQL en la base de datos SQLite")
+def execute_query(sql_query: SQLQuery):
+    conn = get_db_connection()
+    try:
+        cleaned_query = sql_query.query.replace("\nObservation", "").replace("\\'", "'")
+        print("Solicitud POST completa:", sql_query.json())
+        print("Consulta recibida en el servidor:", cleaned_query)
+        
+        cursor = conn.cursor()
+        cursor.execute(cleaned_query)
+        conn.commit()
+        if cursor.description:  # Verifica si la consulta tiene resultados
+            results = cursor.fetchall()
+            return {"results": [dict(row) for row in results]}
+        return {"message": "Consulta ejecutada exitosamente"}
+    except sqlite3.Error as e:
+        raise HTTPException(status_code=400, detail=str(e))
+    finally:
+        conn.close()
+
 @app.post("/sensor_data")
 async def create_sensor_data(sensor_data: SensorData):
     try: