-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlogger.py
More file actions
38 lines (31 loc) · 1.12 KB
/
logger.py
File metadata and controls
38 lines (31 loc) · 1.12 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
from datetime import datetime
import logging
from pathlib import Path
class Logger:
def get_logger(self):
logger = logging.getLogger("logger")
logger.propagate = False
formatter = logging.Formatter(
fmt="| %(asctime)s | %(levelname)s | %(message)s | %(filename)s | %(funcName)s() |",
datefmt="%Y-%m-%d %H:%M:%S"
)
stream_handler = logging.StreamHandler()
stream_handler.setFormatter(formatter)
logger.addHandler(stream_handler)
logger.setLevel(logging.DEBUG)
if self.out_dir:
if self.append:
file_handler = logging.FileHandler(
self.out_dir / f"logs_{datetime.now().strftime('%Y-%m-%d% %H:%M:%S')}.txt"
)
else:
file_handler = logging.FileHandler(
self.out_dir / "logs.txt"
)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
return logger
def __init__(self, out_dir=None, append=False):
out_dir = Path(out_dir)
self.out_dir = out_dir
self.append = append