1. ホーム
  2. python

[解決済み] 関数呼び出し時のタイムアウト

2022-03-23 08:12:21

質問

Pythonで関数を呼び出しているのですが、途中で止まってしまい、スクリプトを再起動しなければならなくなる可能性があることが分かっています。

もし5秒以上かかったら、スクリプトがそれをキャンセルして他のことをするようにするには、どのように関数を呼び出すか、または何を包むか?

どのように解決するのですか?

を使用することができます。 シグナル パッケージを使用します。

In [1]: import signal

# Register an handler for the timeout
In [2]: def handler(signum, frame):
   ...:     print("Forever is over!")
   ...:     raise Exception("end of time")
   ...: 

# This function *may* run for an indetermined time...
In [3]: def loop_forever():
   ...:     import time
   ...:     while 1:
   ...:         print("sec")
   ...:         time.sleep(1)
   ...:         
   ...:         

# Register the signal function handler
In [4]: signal.signal(signal.SIGALRM, handler)
Out[4]: 0

# Define a timeout for your function
In [5]: signal.alarm(10)
Out[5]: 0

In [6]: try:
   ...:     loop_forever()
   ...: except Exception, exc: 
   ...:     print(exc)
   ....: 
sec
sec
sec
sec
sec
sec
sec
sec
Forever is over!
end of time

# Cancel the timer if the function returned before timeout
# (ok, mine won't but yours maybe will :)
In [7]: signal.alarm(0)
Out[7]: 0

呼び出しから10秒後 signal.alarm(10) が、ハンドラが呼ばれる。これは例外を発生させ、通常のPythonのコードから傍受することができます。

このモジュールはスレッドとうまく付き合えません(でも、誰がそうするか?)

注意点 タイムアウトが発生すると例外を発生させるので、そのような関数の例では、関数内部でキャッチされて無視されてしまう可能性があります。

def loop_forever():
    while 1:
        print('sec')
        try:
            time.sleep(10)
        except:
            continue