-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathlogs.py
More file actions
45 lines (30 loc) · 1.24 KB
/
logs.py
File metadata and controls
45 lines (30 loc) · 1.24 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# INspired from Son Nguyen Kim gist here: https://gist.github.com/nguyenkims/e92df0f8bd49973f0c94bddf36ed7fd0
import logging
import sys, os
from logging.handlers import TimedRotatingFileHandler
from datetime import datetime as dt
FORMATTER = logging.Formatter(
"%(asctime)s - %(filename)s:%(lineno)s - %(funcName)s - %(levelname)s - %(message)s"
)
LOG_FILE = os.environ.get("LOG_OUTPUT_DIR", "") + dt.now().strftime(
"logs_%Y-%m-%dT%H-%M-%S.log"
)
LOG_LEVEL = os.environ.get("LOG_LEVEL", "DEBUG").upper()
def get_console_handler():
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setFormatter(FORMATTER)
return console_handler
def get_file_handler():
file_handler = TimedRotatingFileHandler(LOG_FILE, when="midnight")
file_handler.setFormatter(FORMATTER)
return file_handler
def get_logger(logger_name):
logger = logging.getLogger(logger_name)
if logger.hasHandlers():
logger.handlers.clear()
logger.setLevel(LOG_LEVEL) # better to have too much log than not enough
logger.addHandler(get_console_handler())
logger.addHandler(get_file_handler())
# with this pattern, it's rarely necessary to propagate the error up to parent
logger.propagate = False
return logger