seed_gimmick 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6be9be948b6f884683db4b575af6fc91fb4488c2
-  data.tar.gz: 0edcd2d9b7c3ce9bedde7c32f063a496654bd4f0
+  metadata.gz: aceba72060bab98a41a0d8427ed09c9a1bf83338
+  data.tar.gz: 85ff18ec13553c2c3fd5fc6c36c3b3cafe0a6f66
 SHA512:
-  metadata.gz: 8349d720b3c83b1500c4807db2224084d601e70b50f5dc13249ecdfc6a3b6c5b3b7653eb74bb8aa44053e0d829fec43bc9255014ea635895415850c075ea9e7b
-  data.tar.gz: bbaf8659eae13a13b8158bf51e22fab3d698b6a5e0eac8e358c80bafc23ed8dcd098ee7a59cb81eacf3dab077c3b33a330501417629f0a86673dbafe7c9c4352
+  metadata.gz: 48768e2ace6be4b85f2558ef2726e7ef867b9f956f4bfd78cacc665be3376ef1ac517f08cf01ee19768daf71def30a73fc9c6205a79f7c5658893e33662f5a2f
+  data.tar.gz: 84d28b1cb0489fc9dd8aa2fea8ce52ba7141bca70be3cca103f38983263d1a9d454b6c46e5ad7a8c2f40b86d2163ac8d18b04781c0ac09cddab3b4d95b77c730

data/README.md

@@ -16,8 +16,35 @@ And then execute:
 
 ## Usage
 
-    $ bundle exec rails generate seed_gimmick:install
-    $ bundle exec rake db:seed
+### Generate configuration file and default seed directory
+
+Run generator of Rails.  
+Configuration file created to `config/seed_gimmick.yml`.
+
+```
+$ bundle exec rails generate seed_gimmick:install
+      create  db/seed_gimmick/.keep
+      create  config/seed_gimmick.yml
+```
+
+### Rake tasks
+
+#### Apply fixtures
+
+Read fixtures from seed directory and apply to database.  
+All fixtures applied if not specified tables.
+
+```
+$ [TABLES=table_name,table_name] bundle exec rake db:seed_gimmick
+```
+
+#### Dump fixtures
+
+Dump to fixture file from database in seed directory.
+
+```
+$ MODELS=ModelName,ModelName bundle exec rake db:seed_gimmick:dump
+```
 
 ## Contributing
 
@@ -26,3 +53,4 @@ And then execute:
 3. Commit your changes (`git commit -am 'Add some feature'`)
 4. Push to the branch (`git push origin my-new-feature`)
 5. Create a new Pull Request
+

data/lib/generators/seed_gimmick/install_generator.rb

@@ -3,9 +3,14 @@ module SeedGimmick
     class InstallGenerator < ::Rails::Generators::Base
       source_root File.expand_path(File.join(File.dirname(__FILE__), "templates"))
 
-      desc "Install SeedGimmick files"
-      def install_seed_gimmick_files
-        template "seeds.rb", "db/seeds.rb"
+      desc "Create seed_dir of SeedGimmick."
+      def create_seed_dir
+        create_file File.join("db", "seed_gimmick", ".keep")
+      end
+
+      desc "Create config file of SeedGimmick."
+      def create_config_file
+        template "seed_gimmick.yml", "config/seed_gimmick.yml"
       end
     end
   end

data/lib/generators/seed_gimmick/templates/seed_gimmick.yml

@@ -0,0 +1,16 @@
+common: &common
+  seed_dir: db/seed_gimmick
+  default_ext: yml
+  exclude_columns:
+    - created_at
+    - updated_at
+
+development:
+  <<: *common
+
+test:
+  <<: *common
+
+production:
+  <<: *common
+

data/lib/seed_gimmick/inflector.rb

@@ -2,33 +2,48 @@ module SeedGimmick
   class Inflector
     attr_reader :seed_dir
 
-    def initialize(seed_dir)
-      @seed_dir = seed_dir
+    class << self
+      def build(options = nil)
+        new(options || Options.new)
+      end
+
+      def model_class(name)
+        name.try(:safe_constantize)
+      end
+
+      def pathname(path)
+        case path
+        when Pathname then path
+        when String then Pathname.new(path)
+        else nil
+        end
+      end
+    end
+
+    def initialize(options)
+      @seed_dir = options.seed_dir
     end
 
     def model_for(seed_file)
-      relative_path(without_ext(seed_file)).to_s.classify.constantize
+      class_name = relative_path(without_ext(seed_file)).to_s.classify
+      self.class.model_class(class_name)
     end
 
     def seed_for(model, format = :yml)
-      seed_dir + _pathname(model.model_name.collection).sub_ext(".#{format}")
+      seed_dir + pathname(model.model_name.collection).sub_ext(".#{format}")
     end
 
     private
-      def without_ext(pathname)
-        pathname.dirname + pathname.basename(".*")
+      def without_ext(path)
+        path.dirname + path.basename(".*")
       end
 
-      def relative_path(pathname)
-        pathname.relative? ? pathname : pathname.relative_path_from(seed_dir)
+      def relative_path(path)
+        path.relative? ? path : path.relative_path_from(seed_dir)
       end
 
-      def _pathname(path)
-        case path
-        when Pathname then path
-        when String then Pathname.new(path)
-        else nil
-        end
+      def pathname(path)
+        self.class.pathname(path)
       end
   end
 end

data/lib/seed_gimmick/options.rb

@@ -1,22 +1,78 @@
 module SeedGimmick
   class Options
-    %i(seed_dir).each do |key|
+    VALID_OPTIONS_KEYS = %i(
+    seed_dir
+    tables
+    models
+    default_ext
+    exclude_columns
+    ).freeze
+
+    VALID_OPTIONS_KEYS.each do |key|
       define_method "#{key}=" do |value|
         @options[key] = value
       end
     end
 
     def initialize(options = {})
-      @options = options.symbolize_keys
+      @options = load_config.merge(options).symbolize_keys
     end
 
     def seed_dir
-      @options[:seed_dir] || default_seed_dir
+      root_dir.join(@options[:seed_dir] || default_seed_dir)
+    end
+
+    def tables
+      @options[:tables] || ENV["TABLES"].try(:split, ",") || []
+    end
+
+    def tables!
+      tables.presence || (raise SeedGimmickError)
+    end
+
+    def models
+      @options[:models] || ENV["MODELS"].try(:split, ",") || []
+    end
+
+    def models!
+      models.presence || (raise SeedGimmickError)
+    end
+
+    def default_ext
+      @options[:default_ext] || ENV["FORMAT"] || "yml"
+    end
+
+    def exclude_columns
+      @options[:exclude_columns] || default_exclude_columns
+    end
+
+    def load_config
+      config.exist? ? YAML.load_file(config)[environment] : {}
+    end
+
+    def environment
+      ENV["RAILS_ENV"] || ENV["APP_ENV"] || default_env
     end
 
     private
       def default_seed_dir
-        (defined?(Rails) ? Rails.root : Pathname.pwd).join("db", "seeds")
+        Pathname.new("db/seed_gimmick")
+      end
+
+      def default_exclude_columns
+        ["created_at", "updated_at"]
+      end
+
+      def default_env
+        defined?(Rails) ? Rails.env : "development".inquiry
+      end
+
+      def config
+        root_dir.join("config", "seed_gimmick.yml")
+      end
+
+      def root_dir
+        defined?(Rails) ? Rails.root : Pathname.pwd
       end
   end
 end

data/lib/seed_gimmick/seed.rb

@@ -0,0 +1,77 @@
+module SeedGimmick
+  class Seed
+    class << self
+      def find(options = nil)
+        options ||= Options.new
+        seed_files(options).map {|file|
+          new(file, options)
+        }.select {|seed|
+          options.tables.empty? || options.tables.include?(seed.table_name)
+        }
+      end
+
+      private
+        def seed_files(options)
+          Pathname.glob(options.seed_dir.join("**", "*")).select(&:file?)
+        end
+    end
+
+    def initialize(file_or_model, options = nil)
+      @options = options || Options.new
+      @inflector = Inflector.build(@options)
+      unless @model = Inflector.model_class(file_or_model)
+        @seed_file = Inflector.pathname(file_or_model)
+      end
+      self_validation!
+    end
+
+    def model
+      self_validation!
+      @model ||= @inflector.model_for(@seed_file)
+    end
+
+    def seed_file(ext = nil)
+      self_validation!
+      @seed_file ||= @inflector.seed_for(@model, (ext || @options.default_ext))
+    end
+
+    def table_name
+      model.model_name.plural
+    end
+
+    def dump_columns(exclude_columns = [], all = false)
+      return model.column_names if all
+      exclude_columns = exclude_columns.presence || @options.exclude_columns
+      model.column_names - exclude_columns
+    end
+
+    def load_file
+      SeedIO.get(seed_file).load_data
+    end
+
+    def write_file(array_of_hashes)
+      SeedIO.get(seed_file).dump_data(array_of_hashes)
+    end
+
+    def bootstrap
+      ActiveRecord::Migration.say_with_time(table_name) do
+        model.transaction do
+          model.delete_all
+          model.import(load_file.map {|hash| model.new(hash) })
+        end
+      end
+    rescue LoadFailed => e
+      $stdout.print e.message
+    end
+
+    def dump(exclude_columns = [])
+      write_file(model.select(*dump_columns(exclude_columns)).map(&:attributes))
+    end
+
+    private
+      def self_validation!
+        @model || @seed_file || (raise SeedGimmickError)
+      end
+  end
+end
+

data/lib/seed_gimmick/seed_io/base.rb

@@ -10,6 +10,16 @@ module SeedGimmick
       def load_data
         raise NotImplementedError
       end
+
+      def dump_data(array_of_hashes)
+        raise NotImplementedError
+      end
+
+      private
+        def write_raw(data)
+          seed_file.dirname.mkpath
+          seed_file.open("w") {|f| f.write data }
+        end
     end
   end
 end

data/lib/seed_gimmick/seed_io/yaml_file.rb

@@ -5,6 +5,20 @@ module SeedGimmick
         data = YAML.load_file(seed_file) || (raise LoadFailed.new(seed_file))
         data.values
       end
+
+      def dump_data(array_of_hashes)
+        data = {}
+        array_of_hashes.each.with_index(1) do |row, i|
+          data[data_key(i)] = row
+        end
+
+        write_raw(data.to_yaml)
+      end
+
+      private
+        def data_key(id)
+          "data%d" % id
+        end
     end
   end
 end

data/lib/seed_gimmick/seed_io.rb

@@ -3,11 +3,17 @@ require "seed_gimmick/seed_io/yaml_file"
 
 module SeedGimmick
   module SeedIO
-    def self.get(seed_file)
-      ext = File.extname(seed_file.to_s).presence || (raise SeedGimmickError)
-      ext.sub!(/\A\./, "")
-      ext = "yaml" if ext == "yml"
-      const_get("#{ext.capitalize}File", false).new(seed_file)
+    class << self
+      def get(seed_file)
+        const_get(io_class_name_for(seed_file), false).new(seed_file)
+      end
+
+      private
+        def io_class_name_for(seed_file)
+          ext = Inflector.pathname(seed_file).extname.sub(/\A\./, "")
+          ext = "yaml" if ext == "yml"
+          "%sFile" % ext.capitalize
+        end
     end
   end
 end

data/lib/seed_gimmick/version.rb

@@ -1,3 +1,3 @@
 module SeedGimmick
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/seed_gimmick.rb

@@ -10,22 +10,22 @@ require "seed_gimmick/version"
 require "seed_gimmick/errors"
 require "seed_gimmick/options"
 require "seed_gimmick/inflector"
-require "seed_gimmick/finder"
 require "seed_gimmick/seed_io"
-require "seed_gimmick/seed_file"
+require "seed_gimmick/seed"
 require "seed_gimmick/railtie" if defined?(Rails)
 
 module SeedGimmick
-  def self.bootstrap!(options = nil)
-    SeedFile.find(options).each do |seed|
-      seed.bootstrap!
+  class << self
+    def bootstrap(options = nil)
+      Seed.find(options).each {|seed| seed.bootstrap }
     end
-  end
 
-  def self.seed_from(path)
-    path = Pathname.new(path.to_s) unless path.is_a?(Pathname)
-    options = Options.new(seed_dir: path)
-    bootstrap!(options)
+    def dump(options = nil)
+      options ||= Options.new
+      options.models.each do |model_name|
+        Seed.new(model_name, options).dump
+      end
+    end
   end
 end
 

data/lib/tasks/seed_gimmick.rake

@@ -1,6 +1,23 @@
-namespace :seed_gimmick do
-  task :dump => :environment do
-    raise NotImplementedError
+namespace :db do
+  task :seed_gimmick => :environment do
+    SeedGimmick.bootstrap
+  end
+
+  namespace :seed_gimmick do
+    task :dump => :environment do
+      SeedGimmick.dump
+    end
   end
 end
 
+namespace :seed_gimmick do
+  require "pp"
+
+  task :config do
+    pp SeedGimmick::Options.new.load_config
+  end
+
+  task :seed_files do
+    pp SeedGimmick::Seed.find.map {|seed| seed.seed_file.to_s }
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: seed_gimmick
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - i2bskn
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-20 00:00:00.000000000 Z
+date: 2015-01-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -93,14 +93,13 @@ files:
 - README.md
 - Rakefile
 - lib/generators/seed_gimmick/install_generator.rb
-- lib/generators/seed_gimmick/templates/seeds.rb
+- lib/generators/seed_gimmick/templates/seed_gimmick.yml
 - lib/seed_gimmick.rb
 - lib/seed_gimmick/errors.rb
-- lib/seed_gimmick/finder.rb
 - lib/seed_gimmick/inflector.rb
 - lib/seed_gimmick/options.rb
 - lib/seed_gimmick/railtie.rb
-- lib/seed_gimmick/seed_file.rb
+- lib/seed_gimmick/seed.rb
 - lib/seed_gimmick/seed_io.rb
 - lib/seed_gimmick/seed_io/base.rb
 - lib/seed_gimmick/seed_io/yaml_file.rb

data/lib/generators/seed_gimmick/templates/seeds.rb

@@ -1,3 +0,0 @@
-# Bootstrap of SeedGimmick
-SeedGimmick.bootstrap!
-

data/lib/seed_gimmick/finder.rb

@@ -1,18 +0,0 @@
-module SeedGimmick
-  module Finder
-    extend ActiveSupport::Concern
-
-    module ClassMethods
-      def find(options = nil)
-        options ||= Options.new
-        seed_files(options).map {|file| SeedFile.new(options.seed_dir, file) }
-      end
-
-      private
-        def seed_files(options)
-          Pathname.glob(options.seed_dir.join("**", "*")).select(&:file?)
-        end
-    end
-  end
-end
-

data/lib/seed_gimmick/seed_file.rb

@@ -1,39 +0,0 @@
-module SeedGimmick
-  class SeedFile
-    include Finder
-
-    attr_reader :inflector, :seed_file
-
-    def initialize(seed_dir, seed_file)
-      @inflector = Inflector.new(seed_dir)
-      @seed_file = seed_file
-    end
-
-    def load_file
-      SeedIO.get(seed_file).load_data
-    end
-
-    def bootstrap!
-      custom_action do |model, data|
-        ActiveRecord::Migration.say_with_time(model.model_name.plural) do
-          model.transaction do
-            model.delete_all
-            model.import(data.map {|rec| model.new(rec) })
-          end
-        end
-      end
-    rescue LoadFailed => e
-      puts e.message
-    end
-
-    def custom_action
-      yield(_model, load_file) if block_given?
-    end
-
-    private
-      def _model
-        @_model ||= inflector.model_for(seed_file)
-      end
-  end
-end
-