mirror of
https://github.com/michivonah/bbzw-horizon.git
synced 2025-12-22 17:16:27 +01:00
54 lines
No EOL
1.8 KiB
Python
54 lines
No EOL
1.8 KiB
Python
# Webservice
|
|
# INP21b - Timo Weber & Michael von Ah
|
|
|
|
################ IMPORTS ################
|
|
from fastapi import FastAPI, Depends, HTTPException, Header
|
|
from sqlmodel import Session
|
|
from dbfunctions import save_sensor_data, get_client_id_by_name, validate_token_with_access, engine
|
|
from models import SensorDataIn, SensorData, MessageOnly
|
|
|
|
|
|
################ API ################
|
|
app = FastAPI(
|
|
title="M241-M245-BBZW-Horizon",
|
|
description="BBZW-Horizon",
|
|
summary="BBZW-Horizon",
|
|
version="0.0.1"
|
|
)
|
|
|
|
# DB Session
|
|
def get_db():
|
|
db = Session(bind=engine)
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
def authenticate_user(token: str = Header(...), db: Session = Depends(get_db)): # Token aus Header
|
|
if not validate_token_with_access(db, token):
|
|
raise HTTPException(status_code=401, detail="Invalid or expired token, or insufficient permissions")
|
|
|
|
|
|
@app.post("/sensors/push-data", response_model=MessageOnly, tags=["sensors"])
|
|
async def saveNewSensorData(
|
|
client: str,
|
|
data: SensorDataIn,
|
|
db: Session = Depends(get_db),
|
|
auth: bool = Depends(authenticate_user) # Hier wird das Token durch die Dependency validiert
|
|
):
|
|
try:
|
|
# Ermittle die clientid basierend auf dem Client-Namen
|
|
client_id = get_client_id_by_name(db, client)
|
|
if client_id is None:
|
|
raise HTTPException(status_code=404, detail="Client not found")
|
|
|
|
# Erstelle ein SensorData-Objekt für die Datenbank
|
|
sensor_data = SensorData(**data.dict())
|
|
sensor_data.clientid = client_id # Setze die clientid aus der DB
|
|
|
|
# Speichern der Sensordaten in der Datenbank
|
|
save_sensor_data(db, sensor_data)
|
|
|
|
return MessageOnly(message="Sensor data saved successfully.")
|
|
except Exception as error:
|
|
raise HTTPException(status_code=500, detail=str(error)) |