data_seeder 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9201eccfce12d45bbe01559b668fb2fa32c704f1
-  data.tar.gz: a0d02f94fad7dd14f62c05d4ef33223669dc1665
+  metadata.gz: fe35b0a06aa428694ae185ff9f85906d7130b57a
+  data.tar.gz: 4dcb35cdf6202ea08fd5513f9bc09988f00fe40c
 SHA512:
-  metadata.gz: 0933f7b3628701f8ffbef5f5d6247ffed17c997ec2099beefbf2e32d83ee0a84f39256aff8eefaeea1d1e7d29ec959cb6f13b70fc3b3df5c46d2d6d8becb4501
-  data.tar.gz: 5327311bdba8e1e880d87db269eb2ec69fe834139276dc6398b9fba35f2e07c185fa66fae61c05f0ba67a2451377bd48125cd8b23ccb22c7b33b459cd346cb0f
+  metadata.gz: 4f61bf04fde69ddad46a8f18e2b45ee688d875886d282e17cc297736c15cfa202bfef7bea08ff48596d138db79fd7cf5a30f61f3afb7d1e41f02a34ab81e1d7e
+  data.tar.gz: e6ea30bcdcf7ffd25d4be978695fca17b5d770a23f2b08ccc2fbc4e2677d231b06d20f0113d031471769cde7c127bd3569a189ac8e64c67c6ed603a19b03b6e6

data/README.md

@@ -82,16 +82,18 @@ data_seeder has default loaders for txt, csv, json and yml extensions but you ca
 your own custom loaders.
 For instance, suppose you had the following tables:
 
-    create_table "app_errors", force: :cascade do |t|
-      t.integer "app_id"
-      t.string  "code"
-      t.string  "message"
-    end
-    add_index "app_errors", ["app_id"], name: "index_app_errors_on_app_id"
+```ruby
+create_table "apps", force: :cascade do |t|
+  t.string "name"
+end
 
-    create_table "apps", force: :cascade do |t|
-      t.string "name"
-    end
+create_table "app_errors", force: :cascade do |t|
+  t.integer "app_id"
+  t.string  "code"
+  t.string  "message"
+end
+add_index "app_errors", ["app_id"], name: "index_app_errors_on_app_id"
+```
 
 And you wanted to load up separate error messages for each app such as the following 2 files:
 
@@ -223,6 +225,8 @@ seeding above.
 
 Allow config-driven initialization so that we could require: false in the Gemfile and only load as needed.
 
+Add depends_on option.
+
 Meta
 ----
 

data/lib/data_seeder/loader/csv.rb

@@ -5,6 +5,11 @@ module DataSeeder
     class CSV
       include Loader
 
+      def line_number
+        # Don't count the header
+        $INPUT_LINE_NUMBER-1
+      end
+
       def load(io)
         ::CSV.foreach(io, headers: true) do |row|
           save(row.to_hash)

data/lib/data_seeder/loader.rb

@@ -1,3 +1,5 @@
+require 'english'
+
 module DataSeeder
   module Loader
     attr_accessor :file_config, :key_attribute
@@ -32,8 +34,11 @@ module DataSeeder
       dot_index       = @path.rindex('.')
       @path_minus_ext = @path[0, dot_index]
       @file_config    = {}
+      cfg_file = "#{@path_minus_ext}.cfg"
+      @file_config = eval(File.read(cfg_file)) if File.exist?(cfg_file)
       File.open(@path, 'r') do |fin|
-        load_file_config(fin)
+        load_file_config(fin) if @file_config.empty?
+        @file_config = ActiveSupport::HashWithIndifferentAccess.new(@file_config)
         setup
         load(fin)
         teardown
@@ -84,9 +89,17 @@ module DataSeeder
       throw 'Must override load'
     end
 
+    def line_number
+      $INPUT_LINE_NUMBER
+    end
+
     def save(attr)
-      key = attr[@key_attribute.to_s] || attr[@key_attribute.to_sym]
-      raise "No #{@key_attribute} in #{attr.inspect}" unless key
+      if @file_config[:use_line_number_as_id]
+        key = self.line_number
+      else
+        key = attr[@key_attribute.to_s] || attr[@key_attribute.to_sym]
+        raise "No #{@key_attribute} in #{attr.inspect}" unless key
+      end
       @old_keys.delete(key.to_s)
       model = self.klass.find_or_initialize_by(@key_attribute => key)
       model.attributes = attr

data/lib/data_seeder/version.rb

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

data/lib/data_seeder.rb

@@ -8,6 +8,8 @@ module DataSeeder
     attr_writer :config
   end
 
+  @@mutex = Mutex.new
+
   def self.config
     @config ||= Config.new
   end
@@ -25,17 +27,20 @@ module DataSeeder
   end
 
   def self.run(new_config={})
-    msec = Benchmark.ms do
-      new_config.each do |key, value|
-        self.config.send("#{key}=", value)
-      end
-      Dir.chdir(config.seed_dir) do
-        Dir['**/*'].each do |path|
-          SeedFile.load(path) if File.file?(path)
+    @@mutex.synchronize do
+      msec = Benchmark.ms do
+        new_config.each do |key, value|
+          self.config.send("#{key}=", value)
+        end
+        Dir.chdir(config.seed_dir) do
+          Dir['**/*'].each do |path|
+            next if path.end_with?('.cfg')
+            SeedFile.load(path) if File.file?(path)
+          end
         end
       end
+      logger.info { "DataSeeder.run took #{msec.to_i} msec" }
     end
-    logger.info { "DataSeeder.run took #{msec.to_i} msec" }
   end
 
   def self.test_run(new_config={})