File tree Expand file tree Collapse file tree 1 file changed +21
-19
lines changed
src/codegen/shared/logging Expand file tree Collapse file tree 1 file changed +21
-19
lines changed Original file line number Diff line number Diff line change 4
4
5
5
6
6
def get_colored_logger (name : str ) -> logging .Logger :
7
- handler = colorlog .StreamHandler ()
8
- handler .setFormatter (
9
- colorlog .ColoredFormatter (
10
- "%(white)s%(asctime)s - %(name)s - %(log_color)s%(levelname)s%(reset)s%(white)s - %(message_log_color)s%(message)s" ,
11
- log_colors = {
7
+ formatter = colorlog .ColoredFormatter (
8
+ "%(white)s%(asctime)s - %(name)s - %(log_color)s%(levelname)s%(reset)s%(white)s - %(message_log_color)s%(message)s" ,
9
+ log_colors = {
10
+ "DEBUG" : "cyan" ,
11
+ "INFO" : "green" ,
12
+ "WARNING" : "yellow" ,
13
+ "ERROR" : "red" ,
14
+ "CRITICAL" : "red,bg_white" ,
15
+ },
16
+ secondary_log_colors = {
17
+ "message" : {
12
18
"DEBUG" : "cyan" ,
13
- "INFO" : "green " ,
19
+ "INFO" : "blue " ,
14
20
"WARNING" : "yellow" ,
15
21
"ERROR" : "red" ,
16
22
"CRITICAL" : "red,bg_white" ,
17
- },
18
- secondary_log_colors = {
19
- "message" : {
20
- "DEBUG" : "cyan" ,
21
- "INFO" : "blue" ,
22
- "WARNING" : "yellow" ,
23
- "ERROR" : "red" ,
24
- "CRITICAL" : "red,bg_white" ,
25
- }
26
- },
27
- )
23
+ }
24
+ },
28
25
)
29
-
30
26
logger = logging .getLogger (name )
31
- logger .addHandler (handler )
27
+ if logger .hasHandlers ():
28
+ for h in logger .handlers :
29
+ h .setFormatter (formatter )
30
+ else :
31
+ handler = colorlog .StreamHandler ()
32
+ handler .setFormatter (formatter )
33
+ logger .addHandler (handler )
32
34
return logger
You can’t perform that action at this time.
0 commit comments