[英]Process CSV files in location relative to rake task
我有一个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任务相关的文件?
我需要考虑更多,我认为结合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
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.