loggingで出力した内容を変数に格納したい

Question

前提・実現したいこと

AWS LambdaにてあるAPIを作成しています
Lambda内で何らかのエラーが発生したとき、

1. loggingを使ってAWS CloudWatchにログを吐き出す
2. 吐き出したログと同じ内容をAPIのレスポンスとして返却する

といったことが行いたいです。
ログの出力は

FORMAT = '%(asctime)s.%(msecs)03d [%(levelname)s] %(message)s'

でやっているので、１と２で出力する文字列を別の処理で作成しようとすると、時間がずれる可能性があります

何か良い手段をご教示願えますでしょうか。

Answer 1

ジャストアイデアですが。

import logging

class SendLogHandler(logging.StreamHandler):
    def __init__(self):
        StreamHandler.__init__(self)

    def emit(self, record):
        msg = self.format(record)
        #ここでAPIのレスポンスとして返却する
        print(f"SEND {msg}")

のように StreamHandler を継承したオリジナルの LoggingHandlerを作成し、

logger = logging.getLogger('my_application')
logger.setLevel(logging.DEBUG)

fh = logging.FileHandler('out.log')
sh = SendLogHandler()

fh.setLevel(logging.DEBUG)
sh.setLevel(logging.DEBUG)

FORMAT = '%(asctime)s.%(msecs)03d [%(levelname)s] %(message)s'
formatter = logging.Formatter(FORMAT)
fh.setFormatter(formatter)
sh.setFormatter(formatter)

logger.addHandler(fh)
logger.addHandler(sh)

のようにファイル書き出し用のFileHandlerと共に登録するというのはどうでしょうか。

Comments

[Takahiro310]

回答有り難うございます！
今試せる環境に居ないので、後ほど試させていただきます！