在Rails中导入CSV - UTF-8中的非英文字符无效字节序列

时间:2011-09-08 22:31:41

标签: ruby-on-rails ruby csv import rubygems

我正在使用CSVMapper Gem将CSV文件中的某些记录导入到Rails 3模型中。 (我使用这个宝石,因为这是我发现最简单的方法)

无论如何,我用来导入记录的代码如下:

r = import('doc/socios_full.csv') do
    map_to Associate
    after_row lambda{|row, associate| associate.save }
    start_at_row 1
    [group,member,family_relationship_code,family_relationship_description,last_name,names,...]
#The previous line is actually longer, with more atts, but it's been cut to explain the example
end

除非解析器遇到一些非英语字符,例如ó, é, ñ, í, °...,否则它的效果非常好。那时我得到以下错误:

ArgumentError: invalid byte sequence in UTF-8
    from /home/bcb/.rvm/rubies/ruby-1.9.2-p136/lib/ruby/1.9.1/csv.rb:1831:in `sub!'
    from /home/bcb/.rvm/rubies/ruby-1.9.2-p136/lib/ruby/1.9.1/csv.rb:1831:in `block in shift'
    from /home/bcb/.rvm/rubies/ruby-1.9.2-p136/lib/ruby/1.9.1/csv.rb:1825:in `loop'
    from /home/bcb/.rvm/rubies/ruby-1.9.2-p136/lib/ruby/1.9.1/csv.rb:1825:in `shift'
    from /home/bcb/.rvm/rubies/ruby-1.9.2-p136/lib/ruby/1.9.1/csv.rb:1767:in `each'
    from /home/bcb/.rvm/gems/ruby-1.9.2-p136/gems/csv-mapper-0.5.1/lib/csv-mapper.rb:106:in `each_with_index'
    from /home/bcb/.rvm/gems/ruby-1.9.2-p136/gems/csv-mapper-0.5.1/lib/csv-mapper.rb:106:in `import'
    from (irb):63
    from /home/bcb/.rvm/gems/ruby-1.9.2-p136/gems/railties-3.0.9/lib/rails/commands/console.rb:44:in `start'
    from /home/bcb/.rvm/gems/ruby-1.9.2-p136/gems/railties-3.0.9/lib/rails/commands/console.rb:8:in `start'
    from /home/bcb/.rvm/gems/ruby-1.9.2-p136/gems/railties-3.0.9/lib/rails/commands.rb:23:in `<top (required)>'
    from script/rails:6:in `require'
    from script/rails:6:in `<main>'

我确实知道这一点,因为如果我替换所有这些字符,问题就会消失,直到解析器找到另一个非英文字符。问题是我有一个50k的记录文件,因此搜索我能想到的每个字符并尝试每次都导入所有这些记录非常耗时。

有没有办法忽略这些错误并允许解析器继续?或者是否有更简单的方法来导入此CSV文件?

4 个答案:

答案 0 :(得分:14)

这样做:

CSV.foreach(filename, :headers => true , :encoding => 'ISO-8859-1') do |row|

我在尝试读取通过MS Excel保存的CSV文件时遇到了同样的问题。您可以将编码指定为选项。我猜它默认采用UTF-8。

答案 1 :(得分:3)

使用不同的方法解决它,这是一个更容易的解决方案,用于将CSV文件导入Rails 3模型而不是使用外部gem:

    require 'csv'
    CSV.foreach('doc/socios_full.csv') do |row|
        record = Associate.new(
            :media_format   => row[0], 
            :group => row[0],
            :member => row[1],
            :family_relationship_code => row[2],
            :family_relationship_description => row[3],
            :last_name => row[4],
            :names => row[5],
            ...
        )
        record.save!
    end

即使使用非英文字符(只需尝试75k导入文件!),它也可以完美运行。希望它对某人有帮助。

答案 2 :(得分:1)

也许,你可以尝试这样的事情:

csv_string.force_encoding('ISO-8859-1')

答案 3 :(得分:0)

假设您确信CSV将包含正确的标题名称,则以下方法适用于任何模型:

  def self.import(file)
    CSV.foreach(file.path, headers: true) do |row|
      obj = self.new
      obj.attributes.each_key do |attribute|
        index = row.headers.index(attribute)
        obj.send("#{attribute}=",row[index]) if index
      end
      obj.save
    end
  end