File size: 1,452 Bytes
9901342 |
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 |
import os
import json
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import psycopg2
import redis
from datetime import datetime
from dotenv import load_dotenv
load_dotenv()
DATABASE_URL = os.getenv("DATABASE_URL")
REDIS_URL = os.getenv("REDIS_URL")
app = FastAPI()
r = redis.from_url(REDIS_URL)
def get_conn():
return psycopg2.connect(DATABASE_URL)
class Detection(BaseModel):
namespace: str
type: str
timestamp: str = None
location: dict
severity: int = 1
images: list = []
provenance: dict = {}
@app.post("/detect")
def detect(d: Detection):
ts = d.timestamp or datetime.utcnow().isoformat()
try:
conn = get_conn()
cur = conn.cursor()
cur.execute(
"INSERT INTO objects (namespace,type,timestamp,location,severity,images,provenance) VALUES (%s,%s,%s,%s,%s,%s,%s) RETURNING id",
(d.namespace, d.type, ts, json.dumps(d.location), d.severity, json.dumps(d.images), json.dumps(d.provenance))
)
new_id = cur.fetchone()[0]
conn.commit()
cur.close()
conn.close()
# publish event
ev = {"id": str(new_id), "namespace": d.namespace, "type": d.type, "location": d.location, "severity": d.severity}
r.publish("infra:faults:created", json.dumps(ev))
return {"id": str(new_id)}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
|