使用自定义引号的Python中的csv编写器

Question

我正在寻找一种方式来定义自定义quoting用csv.writerPython编写的。有4种内置的方法来对值进行量化：

csv.QUOTE_ALL, csv.QUOTE_MINIMAL, csv.QUOTE_NONNUMERIC, csv.QUOTE_NONE

但是，我需要一种引用机制来模仿Postgres FORCE QUOTE *，即它将引用所有非值。使用csv.QUOTE_ALLPython会将None变成，''但我想改用空字符串。

可以使用内置csv模块来做到这一点（我对黑客不感兴趣，我已经在这样做：P）？还是我被迫编写/获取一些自定义的csv解析器？

通常：是否可以为csv模块编写自定义报价机制？

Answer 1

禁用csv报价并自己添加报价：

def quote(col):
    if col is None:
        return ''
    # uses double-quoting style to escape existing quotes
    return '"{}"'.format(str(col).replace('"', '""'))

writer = csv.writer(fileobj, quoting=csv.QUOTE_NONE, escapechar='', quotechar='')

for row in rows:
    writer.writerow(map(quote, row))

通过将escapechar和都设置quotechar为空字符串，可以避免模块引用已引用的值。

只要您在csv值中不使用 定界符 ，上述方法就可以工作。

请注意，此时您自己编写用逗号分隔的行会更容易：

with open(filename, 'w'), fd:
    for row in rows:
        fd.write(','.join(map(quote, row)) + '\r\n')