mirror of
https://github.com/michivonah/bbzw-horizon.git
synced 2025-12-22 17:16:27 +01:00
add check for api access to authentication
This commit is contained in:
parent
ac66e3d2b5
commit
7f24ba04da
2 changed files with 36 additions and 14 deletions
|
|
@ -4,7 +4,7 @@
|
|||
################ 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, engine
|
||||
from dbfunctions import save_sensor_data, get_client_id_by_name, validate_token_with_access, engine
|
||||
from models import SensorDataIn, SensorData, MessageOnly
|
||||
|
||||
|
||||
|
|
@ -24,18 +24,19 @@ def get_db():
|
|||
finally:
|
||||
db.close()
|
||||
|
||||
def authenticate_user(token: str, db: Session = Depends(get_db)):
|
||||
if not validate_token(db, token):
|
||||
raise HTTPException(status_code=401, detail="Invalid or expired token")
|
||||
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, token: str = Header(...), db: Session = Depends(get_db)):
|
||||
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:
|
||||
# Token-Validierung
|
||||
if not validate_token(db, token):
|
||||
raise HTTPException(status_code=401, detail="Invalid or expired token")
|
||||
|
||||
# Ermittle die clientid basierend auf dem Client-Namen
|
||||
client_id = get_client_id_by_name(db, client)
|
||||
if client_id is None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue