Update main.py

main.py

@@ -1,8 +1,17 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel, Field
 import sqlite3
 import uvicorn
 from fastapi.middleware.cors import CORSMiddleware
 
+class SensorData(BaseModel):
+    timestamp: str
+    humedad_suelo: float
+    luz: float
+    turbidez: float
+    voltaje: float
+    estado: str
+
 app = FastAPI(
     title="API para Arduino Sensor Data",
     description="API para almacenar datos de sensores de Arduino en una base de datos SQLite",
@@ -17,30 +26,32 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
 @app.get("/")
 def read_root():
     return {"message": "Welcome to the Arduino sensor data API"}
 
-
 @app.post("/sensor_data")
-async def create_sensor_data(
-    timestamp: str,
-    humedad_suelo: float,
-    luz: float,
-    turbidez: float,
-    voltaje: float,
-    estado: str,
-):
-    conn = sqlite3.connect("arduino_data.db")
-    cursor = conn.cursor()
-    cursor.execute(
-        "INSERT INTO sensor_data (timestamp, humedad_suelo, luz, turbidez, voltaje, estado) VALUES (?, ?, ?, ?, ?, ?)",
-        (timestamp, humedad_suelo, luz, turbidez, voltaje, estado),
-    )
-    conn.commit()
-    conn.close()
+async def create_sensor_data(sensor_data: SensorData):
+    try:
+        conn = sqlite3.connect("arduino_data.db")
+        cursor = conn.cursor()
+        cursor.execute(
+            "INSERT INTO sensor_data (timestamp, humedad_suelo, luz, turbidez, voltaje, estado) VALUES (?, ?, ?, ?, ?, ?)",
+            (
+                sensor_data.timestamp,
+                sensor_data.humedad_suelo,
+                sensor_data.luz,
+                sensor_data.turbidez,
+                sensor_data.voltaje,
+                sensor_data.estado,
+            ),
+        )
+        conn.commit()
+        conn.close()
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error inserting data: {e}")
 
     return {"message": "Data inserted successfully"}
 
-
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)