在Linux/Mac下怎么為Python函數添加超時時間

本篇內容主要講解“在Linux/Mac下怎么為Python函數添加超時時間”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“在Linux/Mac下怎么為Python函數添加超時時間”吧!

首先我們來看看這個信號的使用方法：

import time import signal   def handler(signum, _):     print('定時到！')     raise Exception('定時到了！')  def clac_statistic(datas):     time.sleep(100)       signal.signal(signal.SIGALRM, handler) signal.alarm(5) clac_statistic('xxx')

運行效果如下圖所示：

首先綁定signal.SIGALRM事件到handler函數中，然后使用signal.alarm(10)延遲10秒發送一個信號。10秒到了以后，函數handler被運行。在函數中拋出了一個異常，導致程序結束。clac_statistic函數原本要運行100秒，但是在10秒以后就停止了，從而實現了函數的超時功能。

基于以上原理，我們實現一個裝飾器，來簡化為不同函數設置超時功能：

import time import signal   class FuncTimeoutException(Exception):     pass  def handler(signum, _):     raise FuncTimeoutException('函數定時到了！')  def func_timeout(times=0):     def decorator(func):         if not times:             return func         def wraps(*args, **kwargs):             signal.alarm(times)             result = func(*args, **kwargs)             signal.alarm(0)  # 函數提前運行完成，取消信號             return result         return wraps     return decorator  signal.signal(signal.SIGALRM, handler)

我們來試一試測試一下這個函數超時裝飾器。首先測試函數的運行時間小于超時時間時，程序正常運行沒有問題：

再來測試一下函數運行時間超過超時時間的情況：

正常拋出FuncTimeoutException異常。

那我們在實際使用中，可以使用try...except FuncTimeoutException捕獲這個異常，然后實現自定義的處理流程，例如：

try:     clac_statistic(100) except FuncTimeException:     print('該函數運行超時，運行自定義的處理流程')

到此，相信大家對“在Linux/Mac下怎么為Python函數添加超時時間”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！