smart_seeds 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1df4a9fffe7d597b387d516eeaf31ffb4d40ce59
-  data.tar.gz: a8ca6dcf4effbe2e6a11b1eba2e83c8130b5f2c9
+  metadata.gz: ca1778f758a9eab74bd9fb47cb40f4862f2238b5
+  data.tar.gz: 0a42feefde292e962be94b6271cab72d29330f36
 SHA512:
-  metadata.gz: c688c984175cd36ddaf7f7cec366c785e732e5c059576cabdef5d066f7715809aabf5e6dd8279d589ee96bb3060444da4d72cf6788da786f20642332598278fd
-  data.tar.gz: 14d30ceec3e842c336b7c526bffb37bc114072dcabb3d1f4232d7a8cf7c8c3d8f12777c67ad6e686a9777a0fe3630fe3789642e600a5990211097e1bfbbfe600
+  metadata.gz: ce4e9b0a682050a6bf1826d7aa12de549fd6b5e0daecf0e5a07e39728e1f9b1fb961dacad9dc752c3b2be2842ba6721522aa4cd2b708ff21533dca476b35f368
+  data.tar.gz: 324d62e05060eae5440e81945071973303c3dee7ede347de149cd6131680c005dd3b08b860c52ee348842ff3bdcfd8951575f618c4fc2fc147829de5b2e04eaf

data/README.md

@@ -1,7 +1,9 @@
 # SmartSeeds
-Gem for auto-smart generating the seeds in your ruby app. Compatible only with ActiveRecord. 
+Gem for smart auto generating of new records with random values for your ruby app.
 
-It determines types of fields(attributes) in an AR model and generates random values for each of these attributes.
+It determines types of fields(attributes) of an ActiveRecord model and creates records with random values for each of these attributes.
+
+The gem has simple integration with Faker and supports foreign_keys, enums, etc.
 
 ## Usage
 
@@ -9,29 +11,21 @@ For example `Entity` model has `name`, `age`, `is_human` attributes.
 
 All you need to do is run this command: `SmartSeeds.plant(Entity)`
 ```
-> SmartSeeds.plant(Entity)
+ > SmartSeeds.plant(Entity)
    (0.1ms)  begin transaction
-  SQL (0.4ms)  INSERT INTO "entities" ("id", "name", "age", "is_human", "created_at", "updated_at") VALUES (?, ?, ?, ?, ?, ?)  [["id", 369], ["name", "Heil"], ["age", 344], ["is_human", false], ["created_at", 2017-01-15 16:10:50 UTC], ["updated_at", 2017-01-15 16:10:50 UTC]]
-   (1.8ms)  commit transaction
- => true 
-
-```
-And you've got an object in a database:
-```
-> Entity.last
-  Entity Load (0.2ms)  SELECT  "entities".* FROM "entities" ORDER BY "entities"."id" DESC LIMIT ?  [["LIMIT", 1]]
- => #<Entity id: 1, name: "Heil", age: 344, is_human: false, created_at: "2017-01-15 16:10:50", updated_at: "2017-01-15 16:10:50"> 
-
+  SQL (0.5ms)  INSERT INTO "entities" ("name", "age", "is_human", "created_at", "updated_at") VALUES (?, ?, ?, ?, ?)  [["name", "Numquam"], ["age", 60246], ["is_human", true], ["created_at", 2017-01-30 15:10:00 UTC], ["updated_at", 2017-01-30 15:10:00 UTC]]
+   (0.7ms)  commit transaction
+ => #<Entity id: 1, name: "Numquam", age: 1488, is_human: false, created_at: "2017-01-30 15:10:00", updated_at: "2017-01-30 15:10:00"> 
 ```
 ### Custom values
-If you want to override some default values, you can send it in a hash:
+If you want to override some default random values, you can send them in a hash:
 
 ```
-> SmartSeeds.plant(Entity, {name: 'Aleah'})
-   (0.1ms)  begin transaction
-  SQL (0.4ms)  INSERT INTO "entities" ("id", "name", "age", "is_human", "created_at", "updated_at") VALUES (?, ?, ?, ?, ?, ?)  [["id", 167], ["name", "Aleah"], ["age", 374], ["is_human", false], ["created_at", 2017-01-15 16:15:15 UTC], ["updated_at", 2017-01-15 16:15:15 UTC]]
-   (1.6ms)  commit transaction
- => true 
+ > SmartSeeds.plant(Entity, {name: 'Aleah'})
+   (0.7ms)  begin transaction
+  SQL (1.9ms)  INSERT INTO "entities" ("name", "age", "is_human", "created_at", "updated_at") VALUES (?, ?, ?, ?, ?)  [["name", "Aleah"], ["age", 15166], ["is_human", false], ["created_at", 2017-01-30 15:11:16 UTC], ["updated_at", 2017-01-30 15:11:16 UTC]]
+   (0.7ms)  commit transaction
+ => #<Entity id: 2, name: "Aleah", age: 39, is_human: true, created_at: "2017-01-30 15:11:16", updated_at: "2017-01-30 15:11:16">
 ```
 
 By the way, all attributes which has default values on AR side will be skipped by default.
@@ -52,11 +46,18 @@ For example: Model `Superhero` has `name` and `power` attributes, so the value w
 
 ```
 
+### Size
+You can set a size and generate few records also
+`SmartSeeds.plant(Entity, {}, 1000)`
+```
+=> "Done! 1000 entities was/were planted."
+```
+
 ## Installation
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'smart_seeds'
+gem 'smart_seeds', '~> 0.1.7'
 ```
 
 And then execute:

data/lib/smart_seeds.rb

@@ -1,4 +1,5 @@
 require 'faker'
+require 'bulk_insert'
 require 'smart_seeds/performing'
 require 'smart_seeds/generator/base'
 require 'smart_seeds/generator/binary'
@@ -16,7 +17,7 @@ require 'smart_seeds/generator/integer/foreign_key'
 require 'smart_seeds/generator/faker'
 
 module SmartSeeds
-  def self.plant(model, attrs={})
-    SmartSeeds::Performing.new(model, attrs).start
+  def self.plant(model, attrs={}, size=1)
+    SmartSeeds::Performing.new(model, attrs, size).start
   end
 end

data/lib/smart_seeds/generator/faker.rb

@@ -18,7 +18,7 @@ module SmartSeeds
       private
 
       def faker_classes_include_model_name?
-        # Check if Faker has a Class which called as a AR Model
+        # Check if Faker has a Class which called as an AR Model
         faker_classes = ::Faker.constants.select {|c| ::Faker.const_get(c).is_a? Class}
         faker_classes.include? model.name.to_sym
       end

data/lib/smart_seeds/generator/integer.rb

@@ -6,32 +6,17 @@ module SmartSeeds
       end
 
       def generate_value
-        if is_an_enum?
-          generate_enum
-        elsif is_foreign_key?
-          generate_foreign_key
+        enum = SmartSeeds::Generator::Enum.new(column, model)
+        foreign_key = SmartSeeds::Generator::ForeignKey.new(column, model)
+
+        if enum.enum?
+          enum.generate_value
+        elsif foreign_key.foreign_key?
+          foreign_key.generate_value
         else
           rand(1..66666)
         end
       end
-
-      private
-
-      def is_an_enum?
-        SmartSeeds::Generator::Enum.new(column, model).is_an_enum?
-      end
-
-      def generate_enum
-        SmartSeeds::Generator::Enum.new(column, model).generate_value
-      end
-
-      def is_foreign_key?
-        SmartSeeds::Generator::ForeignKey.new(column, model).is_a_foreign_key?
-      end
-
-      def generate_foreign_key
-        SmartSeeds::Generator::ForeignKey.new(column, model).generate_value
-      end
     end
   end
 end

data/lib/smart_seeds/generator/integer/enum.rb

@@ -13,9 +13,9 @@ module SmartSeeds
         values_array.sample
       end
 
-      def is_an_enum?
+      def enum?
         model.defined_enums.include?(column.name)
       end
     end
   end
-end
+end

data/lib/smart_seeds/generator/integer/foreign_key.rb

@@ -1,35 +1,55 @@
 module SmartSeeds
   module Generator
     class ForeignKey < SmartSeeds::Generator::Integer
+      include ActiveSupport::Inflector
+
       def initialize(column, model)
         super
       end
 
       def generate_value
-        klass = convert_column_name.constantize
-        klass.ids.sample
+        klass = determine_klass
+
+        if klass.ids.empty?
+          raise ActiveRecord::ActiveRecordError,
+                "There are no records in #{camelized_model_name}. You have to add those first. You can run SmartSeeds.plant(#{camelized_model_name})"
+        else
+          klass.ids.sample
+        end
       end
 
-      def is_a_foreign_key?
-        true if splitted_column_name.last == 'id'
+      def foreign_key?
+        true if associations_include_column?
       end
 
       private
 
-      def convert_column_name
-        # If a foreign key contains of two words splitted by dash like super_hero_id(SuperHero model)
-        if splitted_column_name.count > 2
-          # Pop the last element from an splitted array(_id)
-          arr = splitted_column_name.take(splitted_column_name.count - 1)
-          arr.map(&:capitalize).join
-        # Or it can be just a hero_id(Hero model)
+      def determine_klass
+        assocation = model_associations.select { |obj| obj.name == column_name_without_id.to_sym }
+
+        # if a class_name has been overrided in a model's class
+        if assocation.first.options.key? :class_name
+          assocation.first.options[:class_name].constantize
         else
-          splitted_column_name.first.capitalize
+          camelized_model_name
         end
       end
 
-      def splitted_column_name
-        column.name.split('_')
+      def model_associations
+        model.reflect_on_all_associations(:belongs_to)
+      end
+
+      def column_name_without_id
+        column.name.sub(/^*_id/, '')
+      end
+
+      def associations_include_column?
+        model_associations.map(&:name).include?(column_name_without_id.to_sym)
+      end
+
+      def camelized_model_name
+        # converts underscore words like big_category to BigCategory constant
+        camelize(column_name_without_id).constantize
       end
     end
   end

data/lib/smart_seeds/performing.rb

@@ -1,27 +1,32 @@
 module SmartSeeds
   class Performing
-    def initialize(model, attrs)
+    include ActiveSupport::Inflector
+
+    def initialize(model, attrs, size)
       @attrs = attrs
       @model = model
       @object = model.new
-      @skippable_attributes = %w(id created_at updated_at)
+      @size = size
+      @skippable_attributes = %w(id)
     end
 
     def start
-      add_skippable_attributes
-      set_default_values
+      size.times do
+        add_skippable_attributes
+        set_default_values
 
-      # User can send custom values in a hash: SmartSeeds.(Entity, {name: 'Aleah'})
-      # This method overrides default values to custom('name' in the example above)
-      set_custom_values if attrs.any?
+        # User can send custom values in a hash: SmartSeeds.plant(Entity, {name: 'Aleah'})
+        # This method overrides default values to custom('name' in the example above)
+        set_custom_values if attrs.any?
+        object.save
+      end
 
-      object.save!
-      object
+      return "Done! #{size} #{model.name.downcase.pluralize(size)} was/were planted."
     end
 
     private
 
-    attr_reader :attrs, :model, :object, :skippable_attributes
+    attr_reader :attrs, :model, :object, :size, :skippable_attributes, :worker
 
     def add_skippable_attributes
       # All default attributes which defined in AR object should be skipped by default
@@ -56,4 +61,4 @@ module SmartSeeds
       skippable_attributes.include?(column_name)
     end
   end
-end
+end

data/lib/smart_seeds/version.rb

@@ -1,3 +1,3 @@
 module SmartSeeds
-  VERSION = '0.1.6'
+  VERSION = '0.1.7'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: smart_seeds
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Vitaliy Fry
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-01-27 00:00:00.000000000 Z
+date: 2017-06-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -66,7 +66,8 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: Description of Autoseed.
+description: It determines types of fields(attributes) of an ActiveRecord model and
+  creates records with random values for each of these attributes.
 email:
 - vitaliy.fry@gmail.com
 executables: []
@@ -113,9 +114,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.5.1
+rubygems_version: 2.6.8
 signing_key: 
 specification_version: 4
-summary: Summary of Autoseed.
+summary: Gem for smart auto generating of new records with random values for your
+  ruby app.
 test_files: []
-has_rdoc: