Python 2 和 3 csv 模块文本二进制模式向后兼容

Question

我想创建一个与 Python 2.7-3.6 兼容的代码 我正在尝试解决 csv 模块的问题，我最初在 Python 2.7 中使用 outfile=open('./test.csv','wb') 现在我必须使用 outfile=open('./test.csv','w') like in this question 否则我会招致一个 TypeError: a bytes-like object is required, not 'str'.

我正在使用此代码修复它的那一刻：

import sys
w = 'w'
if sys.version_info[0] < 3:
   w = 'wb'
# Where needed
outfile=open('./test.csv',w)

不是很好，如果我使用 Python 2.7 和 w 如果我使用 Python 3.x，是否有更好的解决方案可以在“wb”中打开文件？为了澄清，我必须在 Python 2.7 中使用 wb，否则，每次向文件添加新行时都会有一个空行。

Answer 1

当在 python 3 上打开要与模块 csv 一起使用的文件时，总是应该在 open 语句中添加 newline=""：

import sys
mode = 'w'
if sys.version_info[0] < 3:
   mode  = 'wb'

# python 3 write 
with open("somefile.txt", mode, newline="") as f:
    pass  # do something with f

newline 参数 在 python 2 中不存在 - 但如果你在 python 3 中跳过它，你会在窗口中得到形状不正确的 csv 输出，其中包含额外的空行.

见csv.writer (python 3):

如果 csvfile 是一个文件对象，它应该用newline='' 打开。如果未指定newline=''，则嵌入在引用字段中的换行符将不会被正确解释，并且在使用\r\n linendings 的平台上，将添加额外的\r。指定newline='' 应该始终是安全的，因为csv 模块 会自己处理（通用）换行符。

---

您也应该使用上下文管理with：

with open("somefile.txt", mode) as f:  # works in 2 and 3
    pass  # do something with f

即使遇到某种异常，也可以关闭文件句柄。这是 python 2 安全的 - 请参阅 Methods of file objects:

处理文件对象时最好使用with 关键字。这样做的好处是文件在其套件完成后正确关闭，即使在途中引发异常也是如此。它也比编写等效的 try-finally 块要短得多。

---

您的解决方案 - 丑陋但有效：

import sys
python3 = sys.version_info[0] >= 3 

if python3:
    with open("somefile.txt","w",newline="") as f:
        pass
else:
    with open("somefile.txt","wb") as f:
        pass

问题是 参数 newline 在 python 2 中不存在。要解决此问题，您必须 wrap/monkypath open(..) 包括上下文管理。