Как ограничить время выполнения функции в Python?

Чтобы ограничить время выполнения функции в Python, можно использовать модуль signal . Он представляет собой таймер, который будет считаться в фоновом режиме параллельно с программой. Когда таймер отсчитает нужное количество секунд, он сгенерирует исключение TimeoutError.

Пример кода:

import signal

def raise_timeout(signum, frame):
    raise TimeoutError

signal.signal(signal.SIGALRM, raise_timeout)

def my_func():
    signal.alarm(5)

try:
    my_func()
finally:
    signal.alarm(0)

Этот подход подходит для Unix-подобных систем. Если нужно обеспечить работу таймаута на различных платформах, можно рассмотреть модули concurrent.futures или multiprocessing.

0 0
Аватар пользователя Elena Gromova
Elena Gromova
31 октября 2024

Для ограничения времени выполнения функции в Python можно воспользоваться модулем signal, который позволяет устанавливать таймаут для выполнения определенного участка кода. Ниже приведен пример кода, который демонстрирует, как можно ограничить время выполнения функции:

import signal

def handler(signum, frame):
    raise TimeoutError("Execution time limit exceeded")

def limited_execution_time(func, time_limit):
    signal.signal(signal.SIGALRM, handler)
    signal.alarm(time_limit)
    try:
        result = func()
    except TimeoutError:
        result = None
    finally:
        signal.alarm(0)
    return result

def my_function():
    # Код вашей функции здесь
    pass

result = limited_execution_time(my_function, 5) # Устанавливаем время ожидания в 5 секунд
if result is None:
    print("Функция превысила ограничение времени выполнения")
else:
    print("Результат выполнения функции:", result)

В этом примере мы определяем функцию-обработчик handler, которая будет вызываться в случае превышения времени выполнения функции. Затем мы создаем функцию limited_execution_time, которая принимает функцию и время ожидания в секундах. Внутри этой функции мы устанавливаем сигнал SIGALRM с помощью signal.signal, чтобы обработчик handler вызывался по истечении времени исполнения. Затем мы выполняем функцию func в блоке try...except, чтобы захватить исключение TimeoutError, которое выбрасывается при превышении времени выполнения. Если исключение произошло, возвращаем None, иначе возвращаем результат выполнения функции.

0 0