Python: Windows эквивалент SIGALRM

У меня есть этот декоратор:

def timed_out(timeout):
    def decorate(f):
        if not hasattr(signal, "SIGALRM"):
            return f

        def handler(signum, frame):
            raise TimedOutExc()

        @functools.wraps(f)
        def new_f(*args, **kwargs):
            old = signal.signal(signal.SIGALRM, handler)
            signal.alarm(timeout)
            try:
                result = f(*args, **kwargs)
            finally:
                signal.signal(signal.SIGALRM, old)
            signal.alarm(0)
            return result

        new_f.func_name = f.func_name
        return new_f

    return decorate

Код только делает что-то на Linux, хотя, как и на Windows, нет SIGALRM, Какой самый простой способ заставить этот код работать и в Windows?

Источник

2 ответа

Решение

Это не очень красиво, но я должен был сделать нечто подобное кроссплатформенным способом, и я придумал использовать отдельный поток. Системы на основе сигналов работают не на всех платформах надежно.

Использование этого класса может быть заключено в декоратор или превращено в with обработчик контекста

YMMV.

#!/usr/bin/env python2.7
import time, threading

class Ticker(threading.Thread):
  """A very simple thread that merely blocks for :attr:`interval` and sets a
  :class:`threading.Event` when the :attr:`interval` has elapsed. It then waits
  for the caller to unset this event before looping again.

  Example use::

    t = Ticker(1.0) # make a ticker
    t.start() # start the ticker in a new thread
    try:
      while t.evt.wait(): # hang out til the time has elapsed
        t.evt.clear() # tell the ticker to loop again
        print time.time(), "FIRING!"
    except:
      t.stop() # tell the thread to stop
      t.join() # wait til the thread actually dies

  """
  # SIGALRM based timing proved to be unreliable on various python installs,
  # so we use a simple thread that blocks on sleep and sets a threading.Event
  # when the timer expires, it does this forever.
  def __init__(self, interval):
    super(Ticker, self).__init__()
    self.interval = interval
    self.evt = threading.Event()
    self.evt.clear()
    self.should_run = threading.Event()
    self.should_run.set()

  def stop(self):
    """Stop the this thread. You probably want to call :meth:`join` immediately
    afterwards
    """
    self.should_run.clear()

  def consume(self):
    was_set = self.evt.is_set()
    if was_set:
      self.evt.clear()
    return was_set

  def run(self):
    """The internal main method of this thread. Block for :attr:`interval`
    seconds before setting :attr:`Ticker.evt`

    .. warning::
      Do not call this directly!  Instead call :meth:`start`.
    """
    while self.should_run.is_set():
      time.sleep(self.interval)
      self.evt.set()
Источник

Я считаю этот код тайм-декоратора очень удобным. (Первоначально я нашел это в ответе на этот вопрос: как ограничить время выполнения вызова функции в Python)

Чтобы заставить его работать в Windows, я использую Python, который устанавливается вместе с Cygwin.

Я запускаю setup-x86_64.exe, затем выбираю python3 пакет из папки Python. (Или, если вы предпочитаете Python 2, python пакет.)

Чтобы переименовать python3 в python2, я определяю псевдоним

alias python=python3

из командной строки Cygwin. Так как я не использую эту функцию очень часто, я, вероятно, не буду помещать ее в.bashrc или что-то еще.

Смежный вопрос: сигнал Python не работает даже на Cygwin?

Источник
Другие вопросы по тегам