1. import time
  2. import functools
  4. def token_timeout(timeout):
  5. """
  6. Decorator to add a timeout to authentication token validation.
  7. """
  8. def decorator(func):
  9. @functools.wraps(func)
  10. def wrapper(*args, **kwargs):
  11. start_time = time.time()
  12. try:
  13. result = func(*args, **kwargs)
  14. except Exception as e:
  15. if time.time() - start_time > timeout:
  16. raise TimeoutError(f"Authentication token expired after {timeout} seconds.") from e
  17. else:
  18. raise # Re-raise other exceptions
  19. return result
  20. return wrapper
  21. return decorator
  23. if __name__ == '__main__':
  24. @token_timeout(5) # Timeout after 5 seconds
  25. def authenticate(token):
  26. """Simulates authentication with a token."""
  27. time.sleep(2) # Simulate some processing time
  28. if token == "valid_token":
  29. return "Authentication successful!"
  30. else:
  31. raise ValueError("Invalid token")
  33. try:
  34. print(authenticate("valid_token"))
  35. except TimeoutError as e:
  36. print(e)
  37. except ValueError as e:
  38. print(e)
  40. try:
  41. print(authenticate("invalid_token"))
  42. except TimeoutError as e:
  43. print(e)
  44. except ValueError as e:
  45. print(e)

Add your comment