27

Pythonスクリプトが正しく終了しているかどうかを知りたいです。これには私が使用してatexitいますが、問題は、atexitがsys.exit（0）で呼び出されたか、ゼロ以外で呼び出されたか、例外で呼び出されたかを区別する方法がわからないことです。

理由：プログラムが適切に終了した場合、プログラムは例外によって終了した場合、またはゼロ以外のエラーコード（終了ステータス）を返した場合以外は何もしません。何らかのアクションをトリガーしたいと思います。

なぜtry/finalを使用していないのか不思議に思うかもしれませんが、これは、共通のモジュールをインポートしている多数のスクリプトに同じ動作を追加したいためです。それらすべてを変更する代わりに、インポートされるモジュールにatexit（）ハックを追加し、それらすべてでこの動作を無料で取得したいと思います。

4

1 に答える 1

22

sys.excepthookこれは、monkey-patchingを使用して解決できますsys.exit()。

import atexit
import sys

class ExitHooks(object):
    def __init__(self):
        self.exit_code = None
        self.exception = None

    def hook(self):
        self._orig_exit = sys.exit
        sys.exit = self.exit
        sys.excepthook = self.exc_handler

    def exit(self, code=0):
        self.exit_code = code
        self._orig_exit(code)

    def exc_handler(self, exc_type, exc, *args):
        self.exception = exc

hooks = ExitHooks()
hooks.hook()

def foo():
    if hooks.exit_code is not None:
        print("death by sys.exit(%d)" % hooks.exit_code)
    elif hooks.exception is not None:
        print("death by exception: %s" % hooks.exception)
    else:
        print("natural death")
atexit.register(foo)

# test
sys.exit(1)
于 2012-03-16T17:24:33.070 に答える