兼容Python 2和3的字符串文件编写器

我创建了一个python Context Manager,它捕获所有输出sys.stdout(例如,使用print()),并将其写入文件.

问题是我无法同时使用python 2.7和3.6.

上下文管理器在内部使用

self.file_writer = open(self.log_file, 'w', encoding='utf8')

但是当我在Python 2.7中运行它时

print(u"a test string")

导致错误消息：

write() argument 1 must be unicode, not str

即使该字符串显然是unicode.

如果我将文件更改为

self.file_writer = open(self.log_file, 'wb')

那么它就可以在Python 2.7中工作,但在3.6中不行.

我需要怎么做才能使其适用于任何python版本？

以下是管理器的摘录：

PATH_PREFIX = "some/path/"
class manager:
    def __init__(self):
        self.log_file = os.path.join(PATH_PREFIX, 'log.txt')
    def __enter__(self):
        # create a file for logging
        self.log_file_stream = open(self.log_file, 'w', encoding='utf8')
        self.log_file_stream.__enter__()
        # redirect stdout to this file
        self.previous_stdout = sys.stdout
        sys.stdout = self.log_file_stream
        return self
    def __exit__(self, etype, value, exception_traceback):
        # stop redirecting stdout to the log file
        sys.stdout = self.previous_stdout
        # close the log file
        self.log_file_stream.__exit__()

解决方法:

sys.stdout在Python 2中应该是字节流,但在Python 3中是Unicode流.Python2的print在写入stdout之前将Unicode字符串编码为字节字符串,但是您已经覆盖了sys.stdout为Unicode.在Python 2和Python 3中都流式传输.

覆盖sys.stdout时,需要为Python 2提供一个字节流,但为Python 3提供一个Unicode流.可以使用sys.version_info.major来确定要支持的字节流.

标签：python-unicode,python-3-x,python-2-7,stdout,python
来源： https://codeday.me/bug/20191025/1926813.html