pythonで修飾器で異常ログを記録する

1493 ワード

スクリプトで必要とされる同じ例外操作が多い場合は、修飾器を使用してコードを簡略化できます.たとえば、放出された例外を記録する必要があります.
ロゴ_exception.pyファイルで、
import functools
import logging

def create_logger():
    logger = logging.getLogger("test_log")
    logger.setLevel(logging.INFO)
    fh = logging.FileHandler("test.log")
    fmt = "[%(asctime)s-%(name)s-%(levelname)s]: %(message)s"
    formatter = logging.Formatter(fmt)
    fh.setFormatter(formatter)
    logger.addHandler(fh) 
    return logger

def log_exception(fn):
    @functools.wraps(fn)
    def wrapper(*args, **kwargs):
        logger = create_logger()
        try:
            fn(*args, **kwargs)
        except Exception as e:
            logger.exception("[Error in {}] msg: {}".format(__name__, str(e)))
            raise
    return wrapper

testでpyファイル:
from log_exception import log_exception

@log_exception
def reciprocal(x):
    return 1/x

if __name__ == "__main__":
    reciprocal(0)

testでlogファイルには、次のエラーメッセージが表示されます.
[2017-11-26 23:37:41,012-test_log-ERROR]: [Error in __main__] msg: integer division or modulo by zero
Traceback (most recent call last):
  File "", line 16, in wrapper
    fn(*args, **kwargs)
  File "", line 3, in reciprocal
    return 1/x
ZeroDivisionError: integer division or modulo by zero