32
32
del get_versions
33
33
34
34
35
- def start_logger ():
35
+ def start_logger (color = True ):
36
36
FORMAT_LOG = "%(levelname)-8s %(asctime)s %(module)s.%(funcName)s :\n \t %(message)s"
37
37
logger = logging .getLogger ("pet" )
38
38
if len (logger .handlers ) == 0 :
39
39
# set up logging to CONSOLE
40
40
console = logging .StreamHandler ()
41
- console .setFormatter (ColoredFormatter (FORMAT_LOG ))
41
+ console .setFormatter (ColoredFormatter (FORMAT_LOG , color = color ))
42
42
# add the handler to the root logger
43
43
logger .addHandler (console )
44
44
return logger
@@ -53,13 +53,14 @@ class ColoredFormatter(logging.Formatter):
53
53
DEBUG = "\033 [34m\t " ,
54
54
)
55
55
56
- def __init__ (self , message ):
56
+ def __init__ (self , message , color = True ):
57
57
super ().__init__ (message )
58
+ self .with_color = color
58
59
59
60
def format (self , record ):
60
61
color = self .COLOR_LEVEL .get (record .levelname , "" )
61
62
color_reset = "\033 [0m"
62
- model = color + "%s" + color_reset
63
+ model = ( color + "%s" + color_reset ) if self . with_color else "%s"
63
64
record .msg = model % record .msg
64
65
record .funcName = model % record .funcName
65
66
record .module = model % record .module
@@ -696,3 +697,6 @@ def identify_time(str_date):
696
697
VAR_DESCR_inv [VAR_DESCR [key ]["nc_name" ]] = key
697
698
for key_old in VAR_DESCR [key ].get ("old_nc_name" , list ()):
698
699
VAR_DESCR_inv [key_old ] = key
700
+
701
+ from . import _version
702
+ __version__ = _version .get_versions ()['version' ]
0 commit comments