Timeout(): приклад з реального світу

Написання функцій у Python

Shayne Miel

Software Architect @ Duo Security

Timeout


def function1():
  # Ця функція інколи
  # працює ду-у-уже довго
  ...


def function2():
  # Ця функція інколи
  # зависає й не повертає результат
  ...

Timeout

@timeout
def function1():
  # Ця функція інколи
  # працює ду-у-уже довго
  ...

@timeout
def function2():
  # Ця функція інколи
  # зависає й не повертає результат
  ...

секундомір

Timeout — довідка

import signal

def raise_timeout(*args, **kwargs):
  raise TimeoutError()

# Коли спрацьовує сигнал "alarm", викликати raise_timeout()
signal.signal(signalnum=signal.SIGALRM, handler=raise_timeout)

# Запустити сигнал через 5 секунд
signal.alarm(5)

# Скасувати сигнал
signal.alarm(0)

def timeout_in_5s(func):

  @wraps(func)
  def wrapper(*args, **kwargs):

    # Установити сигнал на 5 секунд
    signal.alarm(5)

    try:
      # Викликати декоровану функцію
      return func(*args, **kwargs)

    finally:
      # Скасувати сигнал
      signal.alarm(0)

  return wrapper

@timeout_in_5s
def foo():
  time.sleep(10)
  print('foo!')

foo()

TimeoutError

def timeout(n_seconds):

  def decorator(func):

    @wraps(func)
    def wrapper(*args, **kwargs):

      # Установити сигнал на n секунд
      signal.alarm(n_seconds)

      try:
        # Викликати декоровану функцію
        return func(*args, **kwargs)
      finally:
        # Скасувати сигнал
        signal.alarm(0)

    return wrapper

  return decorator

@timeout(5)
def foo():
  time.sleep(10)
  print('foo!')

@timeout(20)
def bar():
  time.sleep(10)
  print('bar!')

foo()

TimeoutError

bar()

bar!

Давайте потренуємось!

Написання функцій у Python