15

我正在寻找一种在 Python中定义自定义quoting的方法。csv.writer有 4 种内置方式来 qoute 值：

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

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

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

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

4

2 回答 2

14

禁用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')
于 2013-10-11T10:15:11.030 回答
8

我已经编写了自己的 csv 编写器，它完全符合我的要求：

class PostgresCSVWriter(object):
    def __init__(self, stream, quotechar="\"", delimiter=",", escapechar="\\"):
        self.stream = stream
        self.quotechar = quotechar
        self.delimiter = delimiter
        self.escapechar = escapechar
        self.buffer_size = 16384

    def _convert_value(self, obj):
        if obj is None:
            return ""
        value = str(obj)
        value = value.replace(self.quotechar, self.quotechar+self.quotechar)
        value = value.replace(self.delimiter, self.escapechar+self.delimiter)
        return self.quotechar+value+self.quotechar

    def _convert_row(self, row):
        return self.delimiter.join(self._convert_value(v) for v in row) + "\r\n"

    def writerow(self, row):
        self.stream.write(self._convert_row(row))

    def writerows(self, rows):
        data = ""
        counter = 0
        for row in rows:
            buf = self._convert_row(row)
            data += buf
            counter += len(buf)
            if counter >= self.buffer_size:
                self.stream.write(data)
                data = ""
                counter = 0

        if data:
            self.stream.write(data)

如果有人看到它有任何问题，请告诉我。不过，我仍在寻找带有csv模块的解决方案。

于 2013-10-11T10:40:58.097 回答