22from magicbus .compat import ntob , unicodestr
33import datetime
44import sys
5- import six
65from magicbus .plugins import SimplePlugin
76
87
98class StreamLogger (SimplePlugin ):
109
1110 default_format = '[%(timestamp)s] (Bus %(bus)s) %(message)s\n '
1211
13- def __init__ (self , bus , stream , level = None , format = None , encoding = 'utf-8' ):
12+ def __init__ (self , bus , stream , level = None , format = None , encoding = None ):
1413 SimplePlugin .__init__ (self , bus )
1514 self .stream = stream
1615 self .level = level
@@ -30,27 +29,29 @@ def log(self, msg, level):
3029 if self .encoding is not None :
3130 if isinstance (complete_msg , unicodestr ):
3231 complete_msg = complete_msg .encode (self .encoding )
33- complete_msg = six .ensure_str (complete_msg )
32+ else :
33+ if not isinstance (complete_msg , unicodestr ):
34+ complete_msg = complete_msg .decode ("utf-8" )
3435 self .stream .write (complete_msg )
3536 self .stream .flush ()
3637
3738
3839class StdoutLogger (StreamLogger ):
3940
40- def __init__ (self , bus , level = None , format = None , encoding = 'utf-8' ):
41+ def __init__ (self , bus , level = None , format = None , encoding = None ):
4142 StreamLogger .__init__ (self , bus , sys .stdout , level , format , encoding )
4243
4344
4445class StderrLogger (StreamLogger ):
4546
46- def __init__ (self , bus , level = None , format = None , encoding = 'utf-8' ):
47+ def __init__ (self , bus , level = None , format = None , encoding = None ):
4748 StreamLogger .__init__ (self , bus , sys .stderr , level , format , encoding )
4849
4950
5051class FileLogger (StreamLogger ):
5152
5253 def __init__ (self , bus , filename = None , file = None ,
53- level = None , format = None , encoding = 'utf8' ):
54+ level = None , format = None , encoding = None ):
5455 self .filename = filename
5556 if file is None :
5657 if filename is None :
0 commit comments