Skip to content

Commit c58b614

Browse files
A better fix for python3 support
1 parent 5e603e5 commit c58b614

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

magicbus/plugins/loggers.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,14 @@
22
from magicbus.compat import ntob, unicodestr
33
import datetime
44
import sys
5-
import six
65
from magicbus.plugins import SimplePlugin
76

87

98
class 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

3839
class 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

4445
class 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

5051
class 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:

setup.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@
3838
),
3939
python_requires='>=2.7',
4040
install_requires=[
41-
'six',
4241
],
4342
extras_require={
4443
'testing': [

0 commit comments

Comments
 (0)