如何强制 Ruby 的 CSV 输出中的一个字段用双引号引起来？

Question

我正在使用 Ruby 的内置 CSV 生成一些 CSV 输出。一切正常，但客户希望输出中的名称字段包含双引号，以便输出看起来像输入文件。例如，输入看起来像这样：

1,1.1.1.1,"Firstname Lastname",more,fields
2,2.2.2.2,"Firstname Lastname, Jr.",more,fields

CSV 的输出是正确的，看起来像：

1,1.1.1.1,Firstname Lastname,more,fields
2,2.2.2.2,"Firstname Lastname, Jr.",more,fields

我知道 CSV 正在做正确的事情，因为它嵌入了空白，所以不双引号第三个字段，并用双引号包裹该字段当它嵌入逗号时使用引号。为了帮助客户感到温暖和模糊，我想做的是告诉 CSV 始终对第三个字段加双引号。

我尝试在我的 to_a 方法中将该字段用双引号括起来，这会创建一个传递给 CSV 的 "Firstname Lastname" 字段，但 CSV 嘲笑了我微不足道的人类尝试并输出 """Firstname Lastname"""。这是正确的做法，因为它转义了双引号，所以这是行不通的。

然后我尝试设置 CSV 的 :force_quotes ==> true 在 open 方法中，它按预期输出双引号包裹所有字段，但客户不喜欢这样，这也是我所期望的。所以，这也不起作用。

我浏览了 Table 和 Row 文档，似乎没有任何东西可以让我访问“生成字符串字段”方法，或设置“for field n 始终使用引用”标志的方法。

我正要深入研究源代码，看看是否有一些超级秘密的调整，或者是否有一种方法可以对 CSV 进行猴子修补并弯曲它来执行我的意愿，但想知道是否有人有一些特殊知识或遇到过这个前。

是的，我知道我可以推出自己的 CSV 输出，但我不喜欢重新发明经过充分测试的轮子。而且，我还知道 FasterCSV；现在它是我正在使用的 Ruby 1.9.2 的一部分，因此明确使用 FasterCSV 并没有给我带来什么特别的。另外，我没有使用 Rails，也无意在 Rails 中重写它，所以除非您有一种使用 Rails 的一小部分来实现它的可爱方法，否则不要打扰。我会否决任何使用这些方法的建议，只是因为您没有费心读到这里。

Answer 1

嗯，有一种方法可以做到这一点，但它并不像我希望的 CSV 代码那样干净。

我必须对 CSV 进行子类化，然后重写 CSV::Row.<<= 方法并添加另一个方法 forced_quote_fields= 才能定义我想要的字段强制引用，并从其他方法中提取两个 lambda。至少它适用于我想要的：

require 'csv'

class MyCSV < CSV
    def <<(row)
      # make sure headers have been assigned
      if header_row? and [Array, String].include? @use_headers.class
        parse_headers  # won't read data for Array or String
        self << @headers if @write_headers
      end

      # handle CSV::Row objects and Hashes
      row = case row
        when self.class::Row then row.fields
        when Hash            then @headers.map { |header| row[header] }
        else                      row
      end

      @headers = row if header_row?
      @lineno  += 1

      @do_quote ||= lambda do |field|
        field         = String(field)
        encoded_quote = @quote_char.encode(field.encoding)
        encoded_quote                                +
        field.gsub(encoded_quote, encoded_quote * 2) +
        encoded_quote
      end

      @quotable_chars      ||= encode_str("\r\n", @col_sep, @quote_char)
      @forced_quote_fields ||= []

      @my_quote_lambda ||= lambda do |field, index|
        if field.nil?  # represent +nil+ fields as empty unquoted fields
          ""
        else
          field = String(field)  # Stringify fields
          # represent empty fields as empty quoted fields
          if (
            field.empty?                          or
            field.count(@quotable_chars).nonzero? or
            @forced_quote_fields.include?(index)
          )
            @do_quote.call(field)
          else
            field  # unquoted field
          end
        end
      end

      output = row.map.with_index(&@my_quote_lambda).join(@col_sep) + @row_sep  # quote and separate
      if (
        @io.is_a?(StringIO)             and
        output.encoding != raw_encoding and
        (compatible_encoding = Encoding.compatible?(@io.string, output))
      )
        @io = StringIO.new(@io.string.force_encoding(compatible_encoding))
        @io.seek(0, IO::SEEK_END)
      end
      @io << output

      self  # for chaining
    end
    alias_method :add_row, :<<
    alias_method :puts,    :<<

    def forced_quote_fields=(indexes=[])
      @forced_quote_fields = indexes
    end
end

这就是代码。调用它：

data = [ 
  %w[1 2 3], 
  [ 2, 'two too',  3 ], 
  [ 3, 'two, too', 3 ] 
]

quote_fields = [1]

puts "Ruby version:   #{ RUBY_VERSION }"
puts "Quoting fields: #{ quote_fields.join(', ') }", "\n"

csv = MyCSV.generate do |_csv|
  _csv.forced_quote_fields = quote_fields
  data.each do |d| 
    _csv << d
  end
end

puts csv

结果：

# >> Ruby version:   1.9.2
# >> Quoting fields: 1
# >> 
# >> 1,"2",3
# >> 2,"two too",3
# >> 3,"two, too",3

Answer 2

这篇文章很旧，但我不敢相信没有人想到这一点。

为什么不这样做：

csv = CSV.generate :quote_char => "\0" do |csv|

其中 \0 是空字符，然后只需在需要的每个字段中添加引号：

csv << [product.upc, "\"" + product.name + "\"" # ...

然后最后您可以执行

csv.gsub!(/\0/, '')

Answer 3

我怀疑这是否会帮助顾客在这么长时间后感到温暖和模糊，但这似乎有效：

require 'csv'
#prepare a lambda which converts field with index 2 
quote_col2 = lambda do |field, fieldinfo|
  # fieldinfo has a line- ,header- and index-method
  if fieldinfo.index == 2 && !field.start_with?('"') then 
    '"' + field + '"'
  else
    field
  end
end

# specify above lambda as one of the converters
csv =  CSV.read("test1.csv", :converters => [quote_col2])
p csv 
# => [["aaa", "bbb", "\"ccc\"", "ddd"], ["fff", "ggg", "\"hhh\"", "iii"]]
File.open("test1.txt","w"){|out| csv.each{|line|out.puts line.join(",")}}

Answer 4

CSV 有一个 force_quotes 选项，该选项将强制它引用所有字段（当您最初发布此内容时它可能不存在）。我意识到这并不完全是你所提议的，但它不是猴子修补。

2.1.0 :008 > puts CSV.generate_line [1,'1.1.1.1','Firstname Lastname','more','fields']
1,1.1.1.1,Firstname Lastname,more,fields
2.1.0 :009 > puts CSV.generate_line [1,'1.1.1.1','Firstname Lastname','more','fields'], force_quotes: true
"1","1.1.1.1","Firstname Lastname","more","fields"

缺点是第一个整数值最终以字符串形式列出，这在导入 Excel 时会发生变化。

Answer 5

已经过去很长一段时间了，但由于 CSV 库 已修补，这可能会对某人有所帮助，如果他们'现在面临这个问题：

require 'csv'

# puts CSV::VERSION # this should be 3.1.9+

headers = ['id', 'ip', 'name', 'foo', 'bar']
data = [
[1, '1.1.1.1','Firstname Lastname','more','fields'],
[2, '2.2.2.2','Firstname Lastname, Jr.','more','fields']
]

quoter = Proc.new do |field, field_meta|
 # the index starts at zero, that's why the third field would be 2:
 field = '"' + field + '"' if field_meta.index == 2 && fields_meta.index > 1
 field = '"' + field + '"' if field.is_a?(String) && field.include?(',')
 # ^ CSV format needs to escape fields containing comma(s): ,
 field
end

file = CSV.generate(headers: true, quote_char: '', write_converters: quoter) do |csv|
    csv << headers
    data.each { |row| csv << row }
end

puts file

输出将是：

id,ip,name,foo,bar
1,1.1.1.1,"Firstname Lastname",more,fields
2,2.2.2.2,"Firstname Lastname, Jr.",more,fields

Answer 6

看起来除了猴子修补/重写之外，现有的 CSV 实现没有任何方法可以做到这一点。

但是，假设您对源数据具有完全控制权，则可以执行以下操作：

1. 将自定义字符串包括逗号（即永远不会在数据中自然找到的字符串）附加到字段末尾每行都有问题；也许类似于“FORCE_COMMAS，”。
2. 生成 CSV 输出。
3. 现在您的字段的每一行都有引号的 CSV 输出，请删除自定义字符串：csv.gsub!(/FORCE_COMMAS,/, "")
4. 客户感到温暖而模糊。

Answer 7

正如 @jwadsa​​ck 提到的，CSV 在 Ruby 2.1 中发生了一些变化，但是这里有 @the-tin-man 的 MyCSV 的工作版本。进行了一点修改，您可以通过选项设置forced_quote_fields。

MyCSV.generate(forced_quote_fields: [1]) do |_csv|...

修改后的代码

require 'csv'

class MyCSV < CSV

  def <<(row)
    # make sure headers have been assigned
    if header_row? and [Array, String].include? @use_headers.class
      parse_headers  # won't read data for Array or String
      self << @headers if @write_headers
    end

    # handle CSV::Row objects and Hashes
    row = case row
          when self.class::Row then row.fields
          when Hash            then @headers.map { |header| row[header] }
          else                      row
          end

    @headers =  row if header_row?
    @lineno  += 1

    output = row.map.with_index(&@quote).join(@col_sep) + @row_sep  # quote and separate
    if @io.is_a?(StringIO)             and
       output.encoding != (encoding = raw_encoding)
      if @force_encoding
        output = output.encode(encoding)
      elsif (compatible_encoding = Encoding.compatible?(@io.string, output))
        @io.set_encoding(compatible_encoding)
        @io.seek(0, IO::SEEK_END)
      end
    end
    @io << output

    self  # for chaining
  end

  def init_separators(options)
    # store the selected separators
    @col_sep    = options.delete(:col_sep).to_s.encode(@encoding)
    @row_sep    = options.delete(:row_sep)  # encode after resolving :auto
    @quote_char = options.delete(:quote_char).to_s.encode(@encoding)
    @forced_quote_fields = options.delete(:forced_quote_fields) || []

    if @quote_char.length != 1
      raise ArgumentError, ":quote_char has to be a single character String"
    end

    #
    # automatically discover row separator when requested
    # (not fully encoding safe)
    #
    if @row_sep == :auto
      if [ARGF, STDIN, STDOUT, STDERR].include?(@io) or
         (defined?(Zlib) and @io.class == Zlib::GzipWriter)
        @row_sep = $INPUT_RECORD_SEPARATOR
      else
        begin
          #
          # remember where we were (pos() will raise an exception if @io is pipe
          # or not opened for reading)
          #
          saved_pos = @io.pos
          while @row_sep == :auto
            #
            # if we run out of data, it's probably a single line
            # (ensure will set default value)
            #
            break unless sample = @io.gets(nil, 1024)
            # extend sample if we're unsure of the line ending
            if sample.end_with? encode_str("\r")
              sample << (@io.gets(nil, 1) || "")
            end

            # try to find a standard separator
            if sample =~ encode_re("\r\n?|\n")
              @row_sep = 
amp;
              break
            end
          end

          # tricky seek() clone to work around GzipReader's lack of seek()
          @io.rewind
          # reset back to the remembered position
          while saved_pos > 1024  # avoid loading a lot of data into memory
            @io.read(1024)
            saved_pos -= 1024
          end
          @io.read(saved_pos) if saved_pos.nonzero?
        rescue IOError         # not opened for reading
          # do nothing:  ensure will set default
        rescue NoMethodError   # Zlib::GzipWriter doesn't have some IO methods
          # do nothing:  ensure will set default
        rescue SystemCallError # pipe
          # do nothing:  ensure will set default
        ensure
          #
          # set default if we failed to detect
          # (stream not opened for reading, a pipe, or a single line of data)
          #
          @row_sep = $INPUT_RECORD_SEPARATOR if @row_sep == :auto
        end
      end
    end
    @row_sep = @row_sep.to_s.encode(@encoding)

    # establish quoting rules
    @force_quotes   = options.delete(:force_quotes)
    do_quote        = lambda do |field|
      field         = String(field)
      encoded_quote = @quote_char.encode(field.encoding)
      encoded_quote                                +
      field.gsub(encoded_quote, encoded_quote * 2) +
      encoded_quote
    end
    quotable_chars = encode_str("\r\n", @col_sep, @quote_char)

    @quote         = if @force_quotes
      do_quote
    else
      lambda do |field, index|
        if field.nil?  # represent +nil+ fields as empty unquoted fields
          ""
        else
          field = String(field)  # Stringify fields
          # represent empty fields as empty quoted fields
          if field.empty? or
             field.count(quotable_chars).nonzero? or
             @forced_quote_fields.include?(index)
            do_quote.call(field)
          else
            field  # unquoted field
          end
        end
      end
    end
  end
end