db_to_fixtures_tasks.rake

namespace :db do
	namespace :fixtures do

		task :load_fixtures_ext do
			require 'active_record/fixtures'
			require 'fixtures_ext'
		end

		Rake::Task["db:fixtures:load"].enhance([:load_fixtures_ext])

		def model_to_fixtures(klass, order = true, table = nil)
			table ||= klass.table_name
			fn = "test/fixtures/%s.yml" % [table]
			id_field = klass.primary_key
			puts "%11s	%s" % ["update", fn]
			File.open(fn, "w") { |fh|
				fh.puts "# automatically generated fixtures from database data"
#				fh.puts "# generated: #{Time.now.to_s}"
				fh.puts
				ord = order ? "#{id_field} asc" : nil
				n = 0 # fallback counter incase o.id is nil
				klass.find(:all, :order => ord).each { |o|
					n += 1
					if klass.respond_to? :fixture_name_column
					  col = klass.fixture_name_column
					  if col.kind_of? Proc
					    key = col[o]
					  else
					    key = o[col]
					  end
					else
					  key = "%s_%d" % [klass.to_s, o.id.nil? ? n : o.id]
				  end
					key = key.downcase
					fh.puts "#{key}:"
					# remove the --- line from the yaml and indent the yaml 2 characters
					o.attributes.each { |k,v|
						o.attributes[k] = v.to_s if v.kind_of? Bignum
					}
					fh.puts o.attributes.to_yaml.split(/[\r\n]+/).reject { |l| l =~ /---/ }.collect { |l| "  #{l}" }.join("\n")
					fh.puts
				}
			}
		end

		# a hack that on the fly creates an active record class for this table
		# assuming this table is really a table
		def table_to_fixtures(table)
			klass = table.classify
			eval("class #{klass} < ActiveRecord::Base; end")
			klasso = Class.const_get(table.classify)
			model_to_fixtures(klasso, false)
		end

		desc "convert the current database into fixtures"
		task :dump do
			require 'config/environment'
			tables = Dir.glob("test/fixtures/*.yml").collect { |f| File.basename(f).gsub(/\.yml$/, '') }
			tables.each { |tbl|
				begin
					# try getting the active record class for the table
					klass = Class.const_get(tbl.classify)
					model_to_fixtures(klass, true, tbl)
				rescue NameError => e
					# class not found, dump it the hard way
					table_to_fixtures(tbl)
				end
			}
		end
	end
end