mixonic-seed-fu 1.1.0 → 1.2.0

data/README

@@ -10,8 +10,8 @@ If you have suggestions or come across problems, please report them on
 the Lighthouse project for Seed Fu:
 http://mbleigh.lighthouseapp.com/projects/10223-seed-fu/overview
 
-Usage
-=======
+Simple Usage
+============
 
 Seed data is taken from the db/fixtures directory. Simply make descriptive .rb
 files in that directory (they can be named anything, but users.rb for the User
@@ -63,4 +63,72 @@ duplication. By default, seed data will constrain to the id like so:
 Note that any constraints that are passed in must be present in the subsequent
 seed data setting.
 
-Copyright (c) 2008 Michael Bleigh, released under the MIT license
+Seed-many Usage
+===============
+
+Seed files can get pretty large.  One way to keep down the file size is to use
+the seed_many syntax.  An example re-using the first seed usage example above:
+
+    User.seed_many(:login, :email, [
+      { :login => "bob", :email => "bob@bobson.com", :first_name => "Bob", :last_name = "Bobson" },
+      { :login => "bob", :email => "bob@stevenson.com", :first_name => "Bob", :last_name = "Stevenson" }
+    ])
+
+Not as pretty, but much more concise.
+
+Generating SeedFu Files
+=======================
+
+Say you have a CSV you need to massage and store as seed files.  You can create
+an import script using SeedFu::Writer.
+
+    # script/generate_cities_seed_from_csv
+    require 'rubygems'
+    require 'fastercsv'
+    require File.join( File.dirname(__FILE__), '..', 'vendor/plugins/seed-fu/lib/seed-fu/writer' )
+
+    CITY_CSV  = File.join( File.dirname(__FILE__), '..', 'city.csv' )
+    SEED_FILE = File.join( File.dirname(__FILE__), '..', 'db', 'fixtures', 'cities.rb' )
+
+    # Create a seed_writer, walk the CSV, add to the file.
+    #
+
+    seed_writer = SeedFu::Writer::SeedMany.new(
+      :seed_file  => SEED_FILE,
+      :seed_model => 'City',
+      :seed_by    => [ :city, :state ]
+    )
+
+    FasterCSV.foreach( CITY_CSV,
+      :return_headers => false,
+      :headers => :first_row
+    ) do |row|
+
+      # Skip all but the US
+      #
+      next unless row['country_code'] == 'US'
+
+      unless us_state
+        puts "No State Match for #{row['region_name']}"
+        next
+      end
+
+      # Write the seed
+      #
+      seed_writer.add_seed({
+        :zip => row['zipcode'],
+        :state => row['state'],
+        :city => row['city'],
+        :latitude => row['latitude'],
+        :longitude => row['longitude']
+      })
+
+    end
+
+    seed_writer.finish
+
+There is also a SeedFu::Writer::Seed in case you don't want to seed_many syntax.
+Easy-peasy.
+
+
+Copyright (c) 2008 Michael Bleigh, Matthew Beale released under the MIT license

data/lib/seed-fu.rb

@@ -31,7 +31,7 @@ module SeedFu
       @data.each do |k, v|
         record.send("#{k}=", v)
       end
-      record.save!
+      raise "Error Saving: #{record.inspect}" unless record.save(false)
       puts " - #{@model_class} #{condition_hash.inspect}"      
       record
     end

data/lib/seed-fu/writer.rb

@@ -0,0 +1,3 @@
+require File.join( File.dirname(__FILE__), 'writer', 'abstract' )
+require File.join( File.dirname(__FILE__), 'writer', 'seed' )
+require File.join( File.dirname(__FILE__), 'writer', 'seed_many' )

data/lib/seed-fu/writer/abstract.rb

@@ -0,0 +1,57 @@
+module SeedFu
+  
+  module Writer
+
+    class Abstract
+      attr_accessor :seed_handle, :config, :after_first_line
+
+      def initialize(options={})
+        self.config = options
+
+        self.seed_handle = File.new(self.config[:seed_file], 'w')
+
+        write_header
+      end
+      
+      def header
+        <<-END
+# DO NOT MODIFY THIS FILE, it was auto-generated.
+# 
+# Date: #{DateTime.now}
+# Using #{self.class} to seed #{config[:seed_model]}
+# Written with the command:
+#
+#   #{$0} #{$*.join}
+#
+        END
+      end
+
+      def footer
+        <<-END
+# End auto-generated file.
+        END
+      end
+
+      def add_seed(hash)
+        $stdout.puts "Added #{hash.inspect}" unless config[:quiet]
+        self.after_first_line = true
+      end
+
+      def write_header
+        seed_handle.syswrite header
+      end
+
+      def write_footer
+        seed_handle.syswrite footer
+      end
+
+      def finish
+        write_footer
+        seed_handle.close
+      end
+
+    end
+  
+  end
+
+end

data/lib/seed-fu/writer/seed.rb

@@ -0,0 +1,23 @@
+module SeedFu
+  
+  module Writer
+
+    class Seed < Abstract
+
+      # This method uses the :seed_by set earlier.
+      #
+      def add_seed(hash, seed_by=nil)
+        seed_by ||= config[:seed_by]
+        seed_handle.syswrite( <<-END
+#{config[:seed_model]}.seed(#{seed_by.collect{|s| ":#{s}"}.join(',')}) { |s|
+#{hash.collect{|k,v| "  s.#{k} = '#{v.to_s.gsub("'", "\'")}'\n"}.join}}
+        END
+        )
+        super(hash)
+      end
+
+    end
+
+  end
+
+end

data/lib/seed-fu/writer/seed_many.rb

@@ -0,0 +1,33 @@
+module SeedFu
+  
+  module Writer
+
+    class SeedMany < Abstract
+
+      def add_seed(hash)
+        seed_handle.syswrite( (<<-END
+#{',' if self.after_first_line}
+  { #{hash.collect{|k,v| ":#{k} => '#{v.to_s.gsub("'", "\'")}'"}.join(', ')} }
+        END
+        ).chomp )
+        super(hash)
+      end
+
+      def write_header
+        super
+        seed_handle.syswrite( (<<-END
+#{config[:seed_model]}.seed_many(#{config[:seed_by].collect{|s| ":#{s}"}.join(',')},[
+        END
+        ).chomp )
+      end
+
+      def write_footer
+        seed_handle.syswrite "\n])\n"
+        super
+      end
+
+    end
+
+  end
+
+end

data/seed-fu.gemspec

@@ -1,7 +1,7 @@
 Gem::Specification.new do |s|
   s.name = 'seed-fu'
-  s.version = '1.1.0'
-  s.date = '2009-07-29'
+  s.version = '1.2.0'
+  s.date = '2009-08-14'
   
   s.summary = "Allows easier database seeding of tables."
   s.description = "Seed Fu is an attempt to once and for all solve the problem of inserting and maintaining seed data in a database. It uses a variety of techniques gathered from various places around the web and combines them to create what is hopefully the most robust seed data system around."
@@ -20,6 +20,10 @@ Gem::Specification.new do |s|
              "Rakefile",
              "init.rb",
              "lib/seed-fu.rb",
+             "lib/seed-fu/writer.rb",
+             "lib/seed-fu/writer/abstract.rb",
+             "lib/seed-fu/writer/seed.rb",
+             "lib/seed-fu/writer/seed_many.rb",
              "rails/init.rb",
              "seed-fu.gemspec",
              "spec/schema.rb",

data/tasks/seed_fu_tasks.rake

@@ -41,13 +41,11 @@ namespace :db do
 
       old_level = ActiveRecord::Base.logger.level
       begin
-        ActiveRecord::Base.validation_disabled = true
         ActiveRecord::Base.logger.level = 7
         ActiveRecord::Base.transaction do
           load file
         end
       ensure
-        ActiveRecord::Base.validation_disabled = false
         ActiveRecord::Base.logger.level = old_level
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: mixonic-seed-fu
 version: !ruby/object:Gem::Version 
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors: 
 - Michael Bleigh
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-07-29 00:00:00 -07:00
+date: 2009-08-14 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -36,6 +36,10 @@ files:
 - Rakefile
 - init.rb
 - lib/seed-fu.rb
+- lib/seed-fu/writer.rb
+- lib/seed-fu/writer/abstract.rb
+- lib/seed-fu/writer/seed.rb
+- lib/seed-fu/writer/seed_many.rb
 - rails/init.rb
 - seed-fu.gemspec
 - spec/schema.rb