ruby-on-rails - 打开包含要在 rake 任务中使用的 csv 文件的 zip 文件

Question

我有一个看起来像这样的 rake 任务

require 'open-uri'
require 'csv'
namespace :Table do
   task reload: :environment do
       ActiveRecord::Base.connection.execute("TRUNCATE Table RESTART IDENTITY")
       csv_text = open('URL')
       csv = CSV.parse(csv_text, :headers=>true)
       csv.each do |row|
         Table.create(table columns)
       end
   end
end

现在我可以让文件以 gzip 或 zip 格式返回。它必须被压缩，但是 csv 是 zip 文件中的唯一文件。

有什么办法可以打开 zip 文件中的 csv 文件吗？

感谢帮助

山姆

Answer 1

Ruby 核心包括 Zlib 支持。如果你有一个指向压缩文件的 URL，你可以使用 Zlib::GzipReader膨胀 open 返回的 Tempfile。

gzipped = open('URL')
csv_text = Zlib::GzipReader.new(gzipped).read
csv = CSV.parse(csv_text, :headers=>true)