超時功能，如果它需要太長時間才能完成

Question

我有一個shell腳本，通過包含URL的文本文件進行循環，該文件包含我想要訪問並截取的URL。

所有這一切都很簡單。該腳本初始化一個類，該類在運行時創建列表中每個站點的屏幕截圖。有些網站需要很長時間才能加載，有些網站可能根本無法加載。所以我想在一個超時腳本中包裝screengrabber函數，如果函數在10秒內無法完成，則返回False。

我滿足於最簡單的解決方案，也許設置一個異步計時器，在10秒後返回False，而不管函數內部發生了什麼？

Answer 1

在signal的文檔中描述了超時操作的過程。

其基本思想是使用信號處理程序在某個時間間隔內設置警報，並在該定時器到期後引發異常。

請注意，這隻適用於UNIX。

以下是創建裝飾器的實現（將以下代碼保存爲timeout.py）。

from functools import wraps 
import errno 
import os 
import signal 

class TimeoutError(Exception): 
    pass 

def timeout(seconds=10, error_message=os.strerror(errno.ETIME)): 
    def decorator(func): 
     def _handle_timeout(signum, frame): 
      raise TimeoutError(error_message) 

     def wrapper(*args, **kwargs): 
      signal.signal(signal.SIGALRM, _handle_timeout) 
      signal.alarm(seconds) 
      try: 
       result = func(*args, **kwargs) 
      finally: 
       signal.alarm(0) 
      return result 

     return wraps(func)(wrapper) 

    return decorator 

這創建了一個名爲@timeout的裝飾器，可以應用於任何長時間運行的函數。

因此，在應用程序代碼，您可以使用像這樣的裝飾：

from timeout import timeout 

# Timeout a long running function with the default expiry of 10 seconds. 
@timeout 
def long_running_function1(): 
    ... 

# Timeout after 5 seconds 
@timeout(5) 
def long_running_function2(): 
    ... 

# Timeout after 30 seconds, with the error "Connection timed out" 
@timeout(30, os.strerror(errno.ETIMEDOUT)) 
def long_running_function3(): 
    ... 

Answer 2

我重寫使用with聲明大衛的回答，它可以讓你做這樣做：

with timeout(seconds=3): 
    time.sleep(4) 

哪會引發TimeoutError。

的代碼仍然使用signal，因此僅適用於UNIX：

import signal 

class timeout: 
    def __init__(self, seconds=1, error_message='Timeout'): 
     self.seconds = seconds 
     self.error_message = error_message 
    def handle_timeout(self, signum, frame): 
     raise TimeoutError(self.error_message) 
    def __enter__(self): 
     signal.signal(signal.SIGALRM, self.handle_timeout) 
     signal.alarm(self.seconds) 
    def __exit__(self, type, value, traceback): 
     signal.alarm(0)