3232del get_versions
3333
3434
35- def start_logger ():
35+ def start_logger (color = True ):
3636 FORMAT_LOG = "%(levelname)-8s %(asctime)s %(module)s.%(funcName)s :\n \t %(message)s"
3737 logger = logging .getLogger ("pet" )
3838 if len (logger .handlers ) == 0 :
3939 # set up logging to CONSOLE
4040 console = logging .StreamHandler ()
41- console .setFormatter (ColoredFormatter (FORMAT_LOG ))
41+ console .setFormatter (ColoredFormatter (FORMAT_LOG , color = color ))
4242 # add the handler to the root logger
4343 logger .addHandler (console )
4444 return logger
@@ -53,13 +53,14 @@ class ColoredFormatter(logging.Formatter):
5353 DEBUG = "\033 [34m\t " ,
5454 )
5555
56- def __init__ (self , message ):
56+ def __init__ (self , message , color = True ):
5757 super ().__init__ (message )
58+ self .with_color = color
5859
5960 def format (self , record ):
6061 color = self .COLOR_LEVEL .get (record .levelname , "" )
6162 color_reset = "\033 [0m"
62- model = color + "%s" + color_reset
63+ model = ( color + "%s" + color_reset ) if self . with_color else "%s"
6364 record .msg = model % record .msg
6465 record .funcName = model % record .funcName
6566 record .module = model % record .module
@@ -696,3 +697,6 @@ def identify_time(str_date):
696697 VAR_DESCR_inv [VAR_DESCR [key ]["nc_name" ]] = key
697698 for key_old in VAR_DESCR [key ].get ("old_nc_name" , list ()):
698699 VAR_DESCR_inv [key_old ] = key
700+
701+ from . import _version
702+ __version__ = _version .get_versions ()['version' ]
0 commit comments