21

我有这个装饰器:

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 中也能工作的最简单方法是什么?

4

2 回答 2

12

它不是很漂亮,但我必须以跨平台的方式做类似的事情,我想出了使用单独的线程。基于信号的系统不能在所有平台上可靠地工作。

这个类的使用可以包装在一个装饰器中,或者做成一个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()
于 2011-12-07T18:26:28.803 回答
-1

此答案仅适用于 Cygwin

我发现这个超时装饰器代码也非常方便。(我最初在这个问题的答案中找到它:如何限制函数调用的执行时间?

为了让它在 Windows 上运行,我使用了与 Cygwin 一起安装的 Python。

我运行 setup-x86_64.exe,然后python3从 Python 文件夹中选择包。(或者,如果您更喜欢 Python 2,请使用该python软件包。)

要将 python3 重命名为 python2,我定义了别名

alias python=python3

从 Cygwin 命令提示符。由于我不经常使用此功能,因此我可能不会将其放入 .bashrc 或任何东西中。

相关问题: Python 信号在 Cygwin 上也不起作用?

于 2016-11-17T00:18:12.313 回答