multiplatform timeout for loops in Python

itimeout.py

from threading import Timer
from multiprocessing import TimeoutError
    
def itimeout(iterable,timeout):
    """timeout for loops
    :param iterable: any iterable
    :param timeout: float max running time in seconds 
    :yield: items in iterator until timeout occurs
    :raise: multiprocessing.TimeoutError if timeout occured
    """
    timer=Timer(timeout,lambda:None)
    timer.start()
    for i in iterable:
        yield i
        if timer.finished.is_set(): 
            raise TimeoutError

usage:

  try:
    for i in itimeout(itertools.count(),10): # count as fast as possible for 10 seconds
      pass
  except TimeoutError:
    print('could count till',i)

can anybody turn this into a decorator that could be used as:

  try:
    @itimeout(10)
    for i in itertools.count(): pass
  except TimeoutError:
    print('could count till',i)