feat: ローカル実行用のログハンドラを追加

This commit is contained in:
shimoda.m@nds-tyo.co.jp 2022-07-01 14:57:43 +09:00
parent af34d6a3d5
commit 3337a0abff

View File

@ -1,5 +1,6 @@
import datetime
import logging
import sys
from zoneinfo import ZoneInfo
from constants import DEFAULT_TIMEZONE, LOG_DATE_FORMAT, LOG_FORMAT
@ -17,10 +18,14 @@ class SingletonLogger:
LOG_DATE_FORMAT
)
formatter.converter = lambda *arg: datetime.datetime.now(ZoneInfo(DEFAULT_TIMEZONE)).timetuple()
# ローカル環境で動かす場合、標準出力ハンドラーを追加する
# AWS Lambda上では`LambdaLoggerHandler`がデフォルトでセットされている
if len(sys.argv) == 2 and sys.argv[1] == 'local':
localHandler = logging.StreamHandler()
logger.addHandler(localHandler)
for handler in logger.handlers:
handler.setFormatter(formatter)
level = logging.getLevelName(LOG_LEVEL)
print(level)
logger.setLevel(level)
return logger