загрузить CSV-файл с отложенной работой - PullRequest
1 голос
/ 15 июня 2010

У меня есть грабли loaddata.rake.

require File.join(File.dirname(__FILE__), 'load_test_data.rb')
namespace :test do
  desc "Insert test data into the database"
  task(:loadtest => :environment) do
    puts "environment = #{RAILS_ENV}"
    puts "load_test_data            ";   load_test_data

    puts "DONE!"
  end
end

У меня есть еще один файл load_test_data.rb, у меня есть некоторые функции для чтения CSV-файла и хранения данных в базе данных. И я вызываю этот файл из моего рейка.

This is my load_test_data.rb
require "faster_csv"

def load_test_data
  puts "test: Balance"
  directory = "test/data"
  name = "balances.csv"
  lines = get_lines_from_csv(directory,name,"snapshot_errors")
  unless lines.nil?
    row_number = 1
    lines.each do |row|
      unless row.header_row?
        begin
          attributes = row.to_hash
          attributes[:balance_name_id] = BalanceName.find(:first, :conditions =>            ["name = ?", attributes[:balance_name]]).id
          attributes.delete(:balance_name)
          Balance.create! attributes
        rescue
          store_error_in_table(row.fields.to_csv,name,row_number,"snapshot_errors")
        end
        row_number += 1
      end
    end
  end
end

Rake задача работает нормально с моим этим кодом ..

Теперь я хочу вызвать эту функцию из моей грабли с отложенной работой. В любом случае, я могу справиться с этим ??

1 Ответ

1 голос
/ 16 июня 2010

Вы можете вызвать его системной командой

system 'rake test:loadtest'

...