iot.py 504 B

1234567891011121314151617181920
  1. from functools import wraps
  2. from fastapi import HTTPException
  3. class MissingIOTDataError(Exception):
  4. def __init__(self):
  5. super().__init__(f"Missing IOT data error.")
  6. def missing_iot_data_handler(func):
  7. @wraps(func)
  8. async def inner_function(*args, **kwargs):
  9. try:
  10. results = await func(*args, **kwargs)
  11. except MissingIOTDataError:
  12. raise HTTPException(status_code=400, detail="Missing data.")
  13. return results
  14. return inner_function