Ruby - 检查文件是否为 CSV

Question

我刚刚编写了一个代码，其中我得到了一个传入参数的 csv 文件并逐行处理它；到目前为止，一切都很好。现在，我想通过确保我们在参数中收到的是一个 .csv 文件来保护我的代码。

我在 Ruby 文档中看到它存在一个 == "--file" 选项，但使用它会产生错误：按照我的理解，这个选项似乎只适用于 txt 文件。

是否有特定的方法可以检查我的文件是否为 csv ？这是我的一些代码：

    if ARGV.empty?
       puts "j'ai rien reçu"
    # option to check, don't work 
    elsif ARGV[0].shift == "--file"

    # my code so far, whithout checking  
    else  CSV.foreach(ARGV.shift) do |row|

等等等等……

Answer 1

我认为在没有额外信息的情况下进行真正安全的测试是不可能的。

请注意您可以做什么： 您在变量文件名中获得一个文件名。

首先，检查是否是文件：

File.exist?

然后你可以检查一下，编码是否正确：

raise "Wrong encoding" unless content.valid_encoding?

您的 csv 是否始终具有相同的列数？你只有一个班轮吗？ 这可能是进行下一次检查的可能性：

content.each_line{|line|
  return false if line.count(sep) < columns - 1
}

可以根据您的情况修改此检查，例如如果你总是有一个确切的行数。

总的来说，您可以定义如下内容：

require 'csv'
#columns defines the expected numer of columns per line
def csv?(filename, sep: ';', columns: 3)
  return false unless File.exist?(filename) #"No file" 
  content = File.read(filename, :encoding => 'utf-8')
  return false unless content.valid_encoding? #"Wrong encoding" 

  content.each_line{|line|
    return false if line.count(sep) < columns - 1
  }
  CSV.parse(content, :col_sep => sep)

end

if csv = csv?('test.csv')
  csv.each do |row|
    p row
  end
end

Answer 2

你可以使用ruby-filemagicgem

gem install ruby-filemagic

用法：

$ irb 
irb(main):001:0> require 'filemagic' 
=> true
irb(main):002:0> fm = FileMagic.new
=> #<FileMagic:0x7fd4afb0>
irb(main):003:0> fm.file('foo.zip') 
=> "Zip archive data, at least v2.0 to extract"
irb(main):004:0>

https://github.com/ricardochimal/ruby-filemagic

Answer 3

使用File.extname()检查源文件

File.extname("test.rb")         #=> ".rb"