1. import time
  2. import functools
  4. def with_timeout(seconds):
  5. """
  6. Decorator to add a timeout to a function.
  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. print(f"Function {func.__name__} raised an exception: {e}")
  16. raise
  18. elapsed_time = time.time() - start_time
  19. if elapsed_time > seconds:
  20. print(f"Function {func.__name__} timed out after {seconds} seconds.")
  21. raise TimeoutError(f"Function {func.__name__} timed out after {seconds} seconds.")
  23. return result
  24. return wrapper
  25. return decorator
  27. class TimeoutError(Exception):
  28. """Custom exception for timeout scenarios."""
  29. pass

Add your comment