diff --git a/lib/tasks/yaml_db_tasks.rake b/lib/tasks/yaml_db_tasks.rake index 24472ca..f6fe4a9 100644 --- a/lib/tasks/yaml_db_tasks.rake +++ b/lib/tasks/yaml_db_tasks.rake @@ -1,45 +1,45 @@ namespace :db do - desc "Dump schema and data to db/schema.rb and db/data.yml" - task(:dump => [ "db:schema:dump", "db:data:dump" ]) + desc "Dump schema and data to db/schema.rb and db/data.yml" + task(:dump => [ "db:schema:dump", "db:data:dump" ]) - desc "Load schema and data from db/schema.rb and db/data.yml" - task(:load => [ "db:schema:load", "db:data:load" ]) + desc "Load schema and data from db/schema.rb and db/data.yml" + task(:load => [ "db:schema:load", "db:data:load" ]) - namespace :data do - def db_dump_data_file (extension = "yml") - "#{dump_dir}/data.#{extension}" - end - - def dump_dir(dir = "") - "#{RAILS_ROOT}/db#{dir}" - end + namespace :data do + def db_dump_data_file (extension = "yml") + "#{dump_dir}/data.#{extension}" + end - desc "Dump contents of database to db/data.extension (defaults to yaml)" - task :dump => :environment do - format_class = ENV['class'] || "YamlDb::Helper" - helper = format_class.constantize - SerializationHelper::Base.new(helper).dump db_dump_data_file helper.extension - end + def dump_dir(dir = "") + "#{RAILS_ROOT}/db#{dir}" + end - desc "Dump contents of database to curr_dir_name/tablename.extension (defaults to yaml)" - task :dump_dir => :environment do - format_class = ENV['class'] || "YamlDb::Helper" - dir = ENV['dir'] || "#{Time.now.to_s.gsub(/ /, '_')}" - SerializationHelper::Base.new(format_class.constantize).dump_to_dir dump_dir("/#{dir}") - end + desc "Dump contents of database to db/data.extension (defaults to yaml)" + task :dump => :environment do + format_class = ENV['class'] || "YamlDb::Helper" + helper = format_class.constantize + SerializationHelper::Base.new(helper).dump db_dump_data_file helper.extension + end - desc "Load contents of db/data.extension (defaults to yaml) into database" - task :load => :environment do - format_class = ENV['class'] || "YamlDb::Helper" - helper = format_class.constantize - SerializationHelper::Base.new(helper).load (db_dump_data_file helper.extension) - end + desc "Dump contents of database to curr_dir_name/tablename.extension (defaults to yaml)" + task :dump_dir => :environment do + format_class = ENV['class'] || "YamlDb::Helper" + dir = ENV['dir'] || "#{Time.now.to_s.gsub(/ /, '_')}" + SerializationHelper::Base.new(format_class.constantize).dump_to_dir dump_dir("/#{dir}") + end - desc "Load contents of db/data_dir into database" - task :load_dir => :environment do - dir = ENV['dir'] || "base" - format_class = ENV['class'] || "YamlDb::Helper" - SerializationHelper::Base.new(format_class.constantize).load_from_dir dump_dir("/#{dir}") - end - end + desc "Load contents of db/data.extension (defaults to yaml) into database" + task :load => :environment do + format_class = ENV['class'] || "YamlDb::Helper" + helper = format_class.constantize + SerializationHelper::Base.new(helper).load (db_dump_data_file helper.extension) + end + + desc "Load contents of db/data_dir into database" + task :load_dir => :environment do + dir = ENV['dir'] || "base" + format_class = ENV['class'] || "YamlDb::Helper" + SerializationHelper::Base.new(format_class.constantize).load_from_dir dump_dir("/#{dir}") + end + end end