kaplan 0.1.2 → 0.2.0

data/README.md

@@ -14,7 +14,7 @@ Doesn't Rails already provide a `db:reset` Rake task? Yes, but it runs `db:schem
 
 Also, Rails doesn't give you a way to programmatically reset (or at least seed) your database. So if you want to do it inside of a script, you're forced to `require 'rake'` and then say `Rake::Task.invoke["db:reset"]`. That's ridiculous. Kaplan gives you methods for plowing and seeding the database.
 
-Additionally, Kaplan improves Rails' seeding by 1) truncating the tables that will be seeded first, 2) allowing you to keep environment-specific seed files, and 2) allowing you to use YAML or text files since that's a simpler way of representing data (and you're probably used to seeing YAML for fixture data).
+Additionally, Kaplan improves Rails' seeding by 1) truncating the tables that will be seeded first, 2) allowing you to keep environment-specific seed files, and 3) allowing you to use YAML or text files since that's a simpler way of representing data (and you're probably used to seeing YAML for fixture data).
 
 Finally, Kaplan was designed to be framework- and ORM-agnostic. Currently the following are supported:
 
@@ -76,6 +76,10 @@ Then, when I run `rake kaplan:db:seed` in different environments, this is which
     | production  | people                  |
     | test        | people, cars, car_types |
     +-------------+-------------------------+
+    
+Besides running seeds in a certain environment, you also have the option to run specific seed files, regardless of which environment they're stored under. So if for some reason we wanted to run the cars.rb seed in the development environment and the car_types.yml seed in the test environment, we could do that like this:
+
+    rake db:seed ONLY=seeds/development/cars.rb,seeds/test/car_types.yml
 
 ## Author/License
 

data/lib/kaplan.rb

@@ -97,10 +97,20 @@ module Kaplan
       end
     end
 
-    def seeds(env)
-      (Dir["#{project_root}/seeds/*.{yml,rb,txt,csv}"] + Dir["#{project_root}/seeds/#{env}/*.{yml,rb,txt,csv}"]).
-      reject {|filename| File.directory?(filename) }.
-      map do |filename|
+    def seeds(env, options={})
+      if options[:only]
+        files = options[:only].
+          map {|path| Dir["#{project_root}/#{path}"] }.
+          flatten.
+          select {|file| File.file?(file) && file =~ /\.(yml|yaml|rb|txt|csv)/ }
+          
+        if files.empty?
+          raise "Error in Kaplan.seeds: Couldn't find the seed files you specified."
+        end
+      else
+        files = Dir["#{project_root}/seeds/*.{yml,yaml,rb,txt,csv}"] + Dir["#{project_root}/seeds/#{env}/*.{yml,yaml,rb,txt,csv}"]
+      end
+      files.map do |filename|
         basename = ::File.basename(filename)
         basename =~ /^(.+?)\.([^.]+)$/
         collection_name = $1
@@ -117,7 +127,8 @@ module Kaplan
       options.reverse_merge!(:env => current_environment)
       puts "Seeding the #{options[:env]} database..." if level > 0
       establish_database(options[:env])
-      seeds(options[:env]).each do |filename, ext, collection_name, model|
+      seeds = seeds(options[:env], :only => options[:only])
+      seeds.each do |filename, ext, collection_name, model|
         if ext == "rb"
           puts " - Adding data for #{collection_name}..." if level > 1
           records = eval(File.read(filename))
@@ -141,16 +152,16 @@ module Kaplan
       options.reverse_merge!(:env => current_environment)
       puts "Plowing the #{options[:env]} database..." if level > 0
       establish_database(options[:env])
-      collections = options[:all] ? all_collections : seedable_collections(options[:env])
+      collections = options[:all] ? all_collections : seedable_collections(options[:env], :only => options[:only])
       collections.each do |coll|
         plow_collection(coll)
         puts " - Plowed #{coll}" if level > 1
       end
     end
 
-    def seedable_collections(env)
+    def seedable_collections(env, options={})
       # Remove collections that don't exist
-      seeds(env).map {|filename, ext, collection_name, model| collection_name } & all_collections
+      seeds(env, options).map {|filename, ext, collection_name, model| collection_name } & all_collections
     end
 
   private

data/lib/kaplan/tasks/kaplan.rake

@@ -4,35 +4,38 @@ namespace :kaplan do
     task :reset, [:env] => :environment do |t, args|
       env = args[:env] || ENV["RAILS_ENV"]
       raise "Can't reset the dev database" if env == "development"
-      #raise "No environment specified. Pass the environment to the task as an argument to specify the environment." unless env
       puts "Resetting #{env} database by cloning dev schema and seeding db..."
       Rake::Task['kaplan:db:clone_structure'].invoke(env)
       Rake::Task['kaplan:db:seed'].invoke(env)
     end
 
     desc "Seeds a database of your choice (default: development) with bootstrap data.\nThe relevant tables are truncated first so you don't have to."
-    task :seed, [:env] => :environment do |t, args|
+    task :seed, [:env, :only] => :environment do |t, args|
       env = args[:env] || ENV["RAILS_ENV"] || "development"
-      #raise "No environment specified. Pass the environment to the task as an argument to specify the environment." unless env
-      Rake::Task['kaplan:db:plow'].invoke(env)
-      Kaplan.seed_database(:env => env)
+      if only = args[:only] || ENV["ONLY"]
+        only = only.split(",") unless Array === only
+      end
+      Rake::Task['kaplan:db:plow'].invoke(env, only)
+      Kaplan.seed_database(:env => env, :only => only)
       Kaplan.establish_database(env)
       Rake::Task['db:seed'].invoke
     end
 
     desc "Truncates tables in a database of your choice (default: development).\nBy default this just truncates the seed tables, if you want all of them pass ALL=true."
-    task :plow, [:env] => :environment do |t, args|
+    task :plow, [:env, :only] => :environment do |t, args|
       env = args[:env] || ENV["RAILS_ENV"] || "development"
-      #raise "No environment specified. Pass the environment to the task as an argument to specify the environment." unless env
-      all = !!ENV["ALL"]
-      Kaplan.plow_database(:env => env, :all => all)
+      unless all = !!ENV["ALL"]
+        if only = args[:only] || ENV["ONLY"]
+          only = only.split(",") unless Array === only
+        end
+      end
+      Kaplan.plow_database(:env => env, :only => only, :all => all)
     end
 
     desc "Dumps the structure of the development database to file and copies it to the database of your choice.\nAdapters must be the same."
     task :clone_structure, [:env] => :environment do |t, args|
       env = args[:env] || ENV["RAILS_ENV"]
       raise "Can't clone the dev database to the dev database" if env == "development"
-      #raise "No environment specified. Pass the environment to the task as an argument to specify the environment." unless env
       puts "Cloning dev structure to #{env} database..."
 
       abcs = ActiveRecord::Base.configurations
@@ -74,7 +77,6 @@ namespace :kaplan do
     desc "Creates a database of your choice (default: development)"
     task :create, [:env] => :environment do |t, args|
       env = args[:env] || ENV["RAILS_ENV"] || "development"
-      #raise "No environment specified. Pass the environment to the task as an argument to specify the environment." unless env
       puts "Creating #{env} database..."
       Kaplan.current_environment = env
       Kaplan.establish_database
@@ -84,7 +86,6 @@ namespace :kaplan do
     desc "Drops the database of your choice (default: development)"
     task :drop, [:env] => :environment do |t, args|
       env = args[:env] || ENV["RAILS_ENV"] || "development"
-      #raise "No environment specified. Pass the environment to the task as an argument to specify the environment." unless env
       puts "Dropping #{env} database..."
       Kaplan.current_environment = env
       Kaplan.establish_database

data/lib/kaplan/version.rb

@@ -1,3 +1,3 @@
 module Kaplan
-  VERSION = "0.1.2"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: kaplan
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 23
   prerelease: false
   segments: 
   - 0
-  - 1
   - 2
-  version: 0.1.2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Elliot Winkler
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-01-17 00:00:00 -07:00
+date: 2011-01-19 00:00:00 -07:00
 default_executable: 
 dependencies: []