Как найти код выхода или причину, когда обратный вызов atexit вызывается в Python? - PullRequest
Новая
       30

Как найти код выхода или причину, когда обратный вызов atexit вызывается в Python?

16 голосов
/ 16 марта 2012

Я хочу знать, правильно ли завершается скрипт Python или нет.Для этого я использую atexit, но проблема в том, что я не знаю, как различить, если atexit был вызван с sys.exit (0) или ненулевым или исключением.

Причина: если программа завершается правильно, она ничего не будет делать, но если программа завершится с исключением или вернет код ошибки (состояние выхода), отличный от нуля, я хочу запустить какое-либо действие.

На случай, если вы удивитесь, почемуЯ не использую try / finally, потому что я хочу добавить такое же поведение для дюжины сценариев, которые импортируют общий модуль.Вместо того, чтобы модифицировать их все, я хочу добавить хак atexit () к импортируемому модулю и получить это поведение бесплатно во всех них.

1 Ответ

16 голосов
/ 16 марта 2012

Вы можете решить эту проблему, используя sys.excepthook и мартышки-патчи 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)
...