在相對於rake任務的位置處理CSV文件

Question

我有一個rake任務和需要處理的CSV文件。 它們位於lib\\tasks目錄的子目錄中：

\foo
  one.csv
  two.csv
  ...
  foo.rake

任務：

task foo: :environment do

  # for each file in directory
  Dir.foreach("./*.csv") do |file|   # not valid

    # process CSV file's content
    CSV.foreach(file, {:headers => true, :col_sep => ";"}) do |row|
      ...
    end

  end # Dir

end # task

如何引用與rake任務相關的文件？

Answer 1

我需要考慮更多，我認為結合File.join和Dir.glob將允許您處理所有的csv文件：

require "csv"

foo_dir = File.join(Rails.root, "lib", "tasks", "foo")

task foo: :environment do

  # for each file in directory
  Dir.glob(foo_dir + "/*.csv") do |csv_file|

    # process CSV file's content
    CSV.foreach(csv_file, {:headers => true, :col_sep => ";"}) do |row|
      #...
    end

  end # Dir

end # task

---

編輯：正如@craig在下面的注釋中指出的那樣，可以通過使用File.dirname和__FILE__更加簡潔地完成此操作：

require "csv"

task foo: :environment do

  # for each file in directory
  Dir.glob(File.dirname(__FILE__) + "/*.csv").each do |file|

    # process CSV file's content
    CSV.foreach(csv_file, {:headers => true, :col_sep => ";"}) do |row|
      #...
    end

  end # Dir

end # task