Python: Windows эквивалент SIGALRM - PullRequest
Новая
       41

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

11 голосов
/ 07 декабря 2011

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

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 ]

8 голосов
/ 07 декабря 2011

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

Использование этого класса может быть заключено в декоратор или превращено в обработчик контекста 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()
0 голосов
/ 17 ноября 2016

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

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

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

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

alias python=python3

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

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

...