code-snipplet|| put table content into yaml

原创于 2009-08-22 12:20:17 发布 · 180 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Rails #ActiveRecord #SQL

code snipplets 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种使用 Ruby on Rails 从现有数据库中创建 YAML 格式的测试固定装置的方法。默认针对开发数据库，可通过设置 RAILS_ENV 来指定其他环境。该过程涉及遍历数据库表并导出其内容为 YAML 文件。

namespace :db do
  namespace :fixtures do

    desc 'Create YAML test fixtures from data in an existing database.  
    Defaults to development database.  Set RAILS_ENV to override.'
    task :dump => :environment do
      sql  = "SELECT * FROM %s"
      skip_tables = ["schema_info"]
      ActiveRecord::Base.establish_connection(RAILS_ENV)
      (ActiveRecord::Base.connection.tables - skip_tables).each do |table_name|
        i = "000"
        File.open("#{RAILS_ROOT}/test/fixtures/#{table_name}.yml", 'w') do |file|
          data = ActiveRecord::Base.connection.select_all(sql % table_name)
          file.write data.inject({}) { |hash, record|
            hash["#{table_name}_#{i.succ!}"] = record
            hash
          }.to_yaml
        end
      end
    end
  end
end