@@ -14,40 +14,34 @@ def flatten(lis):
14
14
new_lis .append (item )
15
15
return new_lis
16
16
17
- logger = None
17
+ _logger_level_set = False
18
18
19
- def get_logger (path = os .getcwd ()):
20
- global logger
21
- if logger is None :
22
- logger = logging .getLogger ("tornasole" )
23
- fh = logging .FileHandler (os .path .join (path , 'tornasole.log' ))
19
+ def get_logger ():
20
+ global _logger_level_set
21
+ name = 'tornasole'
22
+ if not _logger_level_set :
24
23
log_level = os .environ .get ('TORNASOLE_LOG_LEVEL' , default = 'info' )
25
24
log_level = log_level .lower ()
26
-
27
25
if log_level not in ['info' , 'debug' , 'warning' , 'error' , 'critical' , 'off' ]:
28
26
print ('Invalid log level for TORNASOLE_LOG_LEVEL' )
29
27
log_level = 'info'
30
28
31
29
if log_level == 'off' :
32
- logger .disabled = True
33
- elif log_level == 'critical' :
34
- logger .setLevel (logging .CRITICAL )
35
- elif log_level == 'error' :
36
- logger .setLevel (logging .ERROR )
37
- elif log_level == 'warning' :
38
- logger .setLevel (logging .WARNING )
39
- elif log_level == 'info' :
40
- logger .setLevel (logging .INFO )
41
- elif log_level == 'debug' :
42
- logger .setLevel (logging .DEBUG )
43
-
44
- fh .setFormatter (logging .Formatter ('%(asctime)s - %(name)s - %(levelname)s - %(message)s' ))
45
- logger .addHandler (fh )
46
- # logger.propagate = False
47
- return logger
48
- else :
49
- return logger
50
-
30
+ logging .getLogger (name ).disabled = True
31
+ else :
32
+ if log_level == 'critical' :
33
+ level = logging .CRITICAL
34
+ elif log_level == 'error' :
35
+ level = logging .ERROR
36
+ elif log_level == 'warning' :
37
+ level = logging .WARNING
38
+ elif log_level == 'info' :
39
+ level = logging .INFO
40
+ elif log_level == 'debug' :
41
+ level = logging .DEBUG
42
+ logging .getLogger (name ).setLevel (level )
43
+ _logger_level_set = True
44
+ return logging .getLogger (name )
51
45
52
46
def get_immediate_subdirectories (a_dir ):
53
47
return [name for name in os .listdir (a_dir )
0 commit comments