~andreserl/maas/qa-lab-tests-bionic

import errno

from functools import wraps

import os

import signal

class TimeoutError(Exception):

    pass

# From StackOverflow question: "timeout function if it takes too long to

# finish": http://bit.ly/1g1wHpm

def timeout(seconds=10, error_message=os.strerror(errno.ETIME)):

    def decorator(func):

        def _handle_timeout(signum, frame):

            raise TimeoutError(error_message)

        def wrapper(*args, **kwargs):

            signal.signal(signal.SIGALRM, _handle_timeout)

            signal.alarm(seconds)

            try:

                result = func(*args, **kwargs)

            finally:

                signal.alarm(0)

            return result

        return wraps(func)(wrapper)

    return decorator