@@ -24,7 +24,7 @@ def __init__(self, root_name, *args, **kwargs):
2424 def _alog_fn (self , fn ):
2525 if 'ipython-input-' in fn : # pragma: no cover
2626 return "<IPython"
27- elif fn == '<stdin>' : # pragma: no cover
27+ if fn == '<stdin>' : # pragma: no cover
2828 return '<stdin'
2929
3030 paths = []
@@ -143,19 +143,31 @@ def get_level(self, logger=None):
143143 return handler .level
144144
145145 def get_format (self , logger = None ):
146- logger = logger or self
147- for handler in self .handlers :
146+ from warnings import warn
147+ msg = "`get_format()` actually return a Formatter. " \
148+ "Use `get_formatter()` instead."
149+ warn (msg )
150+ alogger = logger or self
151+ return alogger .get_formatter ()
152+
153+ def get_formatter (self , logger = None ):
154+ alogger = logger or self
155+ for handler in alogger .handlers :
148156 if handler .formatter :
149157 return handler .formatter
150158
159+ def set_formatter (self , formatter , alogger = None ):
160+ alogger = alogger or self
161+ for handler in alogger .handlers :
162+ handler .setFormatter (formatter )
163+
151164 def set_format (self , fs , alogger = None , is_default = False ,
152165 time_strfmt = "%Y-%m-%d %H:%M:%S" ):
153166 alogger = alogger or self
154167 formatter = Formatter (fs , time_strfmt ) \
155168 if in_python2_runtime \
156169 else Formatter (fs , time_strfmt , "%" )
157- for handler in alogger .handlers :
158- handler .setFormatter (formatter )
170+ alogger .set_formatter (formatter , alogger = alogger )
159171 if not is_default :
160172 alogger .alog_config ['custom_format' ] = fs
161173
0 commit comments