|
12 | 12 | from . import constant as constant |
13 | 13 |
|
14 | 14 |
|
15 | | -def init_log(log_file, create_file=True): |
| 15 | +def init_log(log_file, create_file=True, stream_log_level=logging.WARN, file_log_level=logging.INFO): |
16 | 16 |
|
17 | 17 | logger = logging.getLogger(constant.LOGGER_NAME) |
| 18 | + |
18 | 19 | if not logger.hasHandlers(): |
19 | | - log_level = logging.WARNING |
20 | | - formatter = logging.Formatter('%(message)s') |
21 | | - |
| 20 | + logger.setLevel(logging.DEBUG) |
22 | 21 | log_dir = os.path.dirname(log_file) |
23 | 22 | Path(log_dir).mkdir(parents=True, exist_ok=True) |
24 | 23 |
|
25 | 24 | if create_file: |
26 | | - file_hanlder = logging.FileHandler(log_file) |
27 | | - file_hanlder.setLevel(log_level) |
28 | | - file_hanlder.setFormatter(formatter) |
29 | | - file_hanlder.propagate = False |
30 | | - logger.addHandler(file_hanlder) |
31 | | - |
32 | | - console = logging.StreamHandler() |
33 | | - console.setLevel(log_level) |
34 | | - console.setFormatter(formatter) |
35 | | - console.propagate = False |
36 | | - logger.addHandler(console) |
| 25 | + |
| 26 | + file_handlder = logging.FileHandler(log_file) |
| 27 | + file_handlder.setLevel(file_log_level) |
| 28 | + file_formatter = logging.Formatter('[%(levelname)7s] %(message)s') |
| 29 | + file_handlder.setFormatter(file_formatter) |
| 30 | + file_handlder.propagate = False |
| 31 | + logger.addHandler(file_handlder) |
| 32 | + |
| 33 | + console_handler = logging.StreamHandler() |
| 34 | + console_handler.setLevel(stream_log_level) |
| 35 | + console_formatter = logging.Formatter('%(message)s') |
| 36 | + console_handler.setFormatter(console_formatter) |
| 37 | + console_handler.propagate = False |
| 38 | + logger.addHandler(console_handler) |
37 | 39 |
|
38 | 40 | logger.propagate = False |
39 | 41 |
|
|
0 commit comments