mirror of
https://github.com/michivonah/bbzw-horizon.git
synced 2025-12-22 17:16:27 +01:00
59 lines
No EOL
2.2 KiB
Python
59 lines
No EOL
2.2 KiB
Python
import os
|
|
from sqlmodel import create_engine, Session
|
|
from dotenv import load_dotenv
|
|
from models import User, SensorData, Client, Session as SessionModel
|
|
from datetime import datetime
|
|
|
|
|
|
# Lade Umgebungsvariablen
|
|
load_dotenv()
|
|
DB_CONNECTION_STRING = os.getenv("DB_CONNECTION_STRING", "postgresql://user:password@localhost/sensordb")
|
|
|
|
# SQLAlchemy Setup
|
|
engine = create_engine(DB_CONNECTION_STRING, echo=True)
|
|
|
|
# Funktion zum Speichern von Sensordaten
|
|
def save_sensor_data(db: Session, sensor_data: SensorData):
|
|
db.add(sensor_data) # Das Pydantic-Modell kann direkt hinzugefügt werden
|
|
db.commit()
|
|
db.refresh(sensor_data) # Holt die letzten Informationen des hinzugefügten Eintrags
|
|
return sensor_data
|
|
|
|
def get_client_id_by_name(db: Session, client_name: str):
|
|
client = db.query(Client).filter(Client.name == client_name).first()
|
|
return client.id if client else None # Gibt die clientid zurück oder None, wenn nicht gefunden
|
|
|
|
def validate_token_with_access(db: Session, token: str) -> bool:
|
|
session = db.query(SessionModel).filter(SessionModel.token == token).first()
|
|
|
|
# Wenn die Sitzung nicht gefunden wird, ist der Zugriff verweigert
|
|
if not session:
|
|
return False
|
|
|
|
# Hole den Benutzer basierend auf userid
|
|
user = db.query(User).filter(User.id == session.userid).first()
|
|
|
|
# Überprüfe, ob das Token gültig ist und ob api_access True ist
|
|
if session.validuntil <= datetime.now().date() or not user.api_access:
|
|
return False
|
|
|
|
return True # Token ist gültig und User hat API-Zugriff
|
|
|
|
def check_api_access(db: Session, token: str) -> bool:
|
|
session = db.query(SessionModel).filter(SessionModel.token == token).first()
|
|
|
|
# Wenn die Sitzung nicht gefunden wird, ist der Zugriff verweigert
|
|
if not session:
|
|
return False
|
|
|
|
# Hole den Benutzer basierend auf userid
|
|
user = db.query(User).filter(User.id == session.userid).first()
|
|
|
|
# Überprüfe, ob api_access True ist
|
|
return user.api_access if user else False # Gibt True oder False zurück
|
|
|
|
def save_token_to_db(db: Session, token: str, user_id: int, valid_until: datetime):
|
|
new_session = SessionModel(token=token, validuntil=valid_until, userid=user_id)
|
|
db.add(new_session)
|
|
db.commit()
|
|
db.refresh(new_session) |