colonization 0.3.2 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 714467c64b4b0faff399afd1299551c801cdee66
-  data.tar.gz: 08292d778ac84e466d93be3f9a4a5d8598c91d47
+  metadata.gz: f7b238a1f52eb0e3154bc44755b8a9b9412824b4
+  data.tar.gz: 14c8ec0adc365c5747e15d35e66f9cb60c24ced9
 SHA512:
-  metadata.gz: 8c49b6dfaeab5a06282b4275562afd7ec0c7d8a6e9a7b9d04e131b24713e7dad3c02a4b4371a2d84eb9c8ed38f11e0f97ac876c5b6fb52050142d7f6d7b41da6
-  data.tar.gz: 6036cc75b861f309fef8c312ee4f60ac92241c09847be606c73fa43dfce45771b4bef031cedc544305e1b24147fcde89fcc75a2dbec6b69a41344cb4ed3e2429
+  metadata.gz: 0eaaa8beb2cbefd30416cea77350558864f7f1cde56c8eee70eb05d2f4cdc041385244e9eae76f840b3d0ee92396c2f5d397372b18c35df49d986031ef97fddd
+  data.tar.gz: 43ef128aae6cfddfb2f702cf5ae29c0bb6bdb14bb975f112083b52261444abbff7dda7046ed51bc699913f6a8953d08c32d2d17cc5d1bf6a62c584a39b8af737

data/README.md

@@ -1,6 +1,6 @@
 # Colonization
 
-Gem for development database populating. Data seeding regarding to your models fields, their types and validations. It is in development now - only presence and acceptance validations currently supported.
+Gem for development database populating. Data seeding regarding to your models fields, their types and validations. It is in development now - only presence, exclusion and acceptance validations currently supported.
 
 ## Installation
 

data/lib/colonization/ship.rb

@@ -1,13 +1,14 @@
 require 'colonization/investigator'
-require 'colonization/colonizer'
+require 'colonization/strategies/default/colonizer'
+require 'colonization/strategies/fixed/colonizer'
 
 class Colonization::Ship
 
-  attr_reader :colonizers
+  attr_reader :model_colonizers
 
   def initialize
-    models = investigate_database
-    @colonizers = get_colonizers(models)
+    models = Colonization::Investigator.new.investigate
+    @model_colonizers = get_model_colonizers(models)
   end
 
   def colonize
@@ -18,29 +19,33 @@ class Colonization::Ship
   private
 
   def clear_db
-    colonizers.each do |item|
-      item.clear_table
+    model_colonizers.each do |model_colnizer|
+      model_colnizer.clear_table
     end
   end
 
   def populate
-    colonizers.each do |item|
-      item.print_action
-      item.colonize
+    model_colonizers.each do |model_colonizer|
+      model_colonizer.print_action
+      model_colonizer.colonize
     end
   end
 
-  def investigate_database
-    investigator = Colonization::Investigator.new
-    investigator.investigate
+  def get_model_colonizers(models)
+    models.map do |model|
+      colonizer_class.new(model)
+    end
   end
 
-  def get_colonizers(colonize_list)
-    result = []
-    colonize_list.each do |item|
-      result << Colonization::Colonizer.new(item)
+  def colonizer_class
+    case Colonization.configuration.seed_type
+    when :default
+      Colonization::Strategies::Default::Colonizer
+    when :fixed
+      Colonization::Strategies::Fixed::Colonizer
+    else
+      raise UnknownStrategyError
     end
-    result
   end
 
 end

data/lib/colonization/strategies/default/colonizer.rb

@@ -0,0 +1,105 @@
+require 'colonization/strategies/model_field'
+require 'colonization/strategies/default/model_relation'
+require 'colonization/configuration'
+
+module Colonization::Strategies::Default
+  class Colonizer
+
+    attr_accessor :model
+
+    def initialize(model)
+      @model = model
+    end
+
+    def print_action
+      puts "colonize #{model}"
+    end
+
+    def colonize(times = nil)
+      res = []
+      (times || model_instances_number).times do
+        res << model.create!(params_with_values)
+      end
+      res
+    end
+
+    def clear_table
+      model.destroy_all if clear_table?
+    end
+
+    private
+
+    def params_with_values
+      res = {}
+
+      model_fields_objects.each do |field_object|
+        res[field_object.name] = field_object.value
+      end
+
+      model_relation_objects.each do |field_object|
+        res[field_object.name] = field_object.value
+      end
+
+      res
+    end
+
+    def model_fields
+      model.column_names - unnecessary_fields
+    end
+
+    def model_relations_strings
+      all_relations.map do |relation|
+        "#{relation.name}_id"
+      end
+    end
+
+    def all_relations
+      model.reflect_on_all_associations
+    end
+
+    def belongs_to_relations
+      model.reflect_on_all_associations(:belongs_to) || []
+    end
+
+    def has_many_relations
+      model.reflect_on_all_associations(:has_many) || []
+    end
+
+    def unnecessary_fields
+      ["created_at", "updated_at", "id"] + model_relations_strings
+    end
+
+    def model_fields_objects
+      res = []
+      model_fields.each do |field|
+        res << Colonization::Strategies::ModelField.new(model, field, values_generator)
+      end
+      res
+    end
+
+    def model_relation_objects
+      res = []
+      has_many_relations.each do |reflection|
+        res << Colonization::Strategies::Default::ModelRelation.new(reflection)
+      end
+      res
+    end
+
+    def model_instances_number
+      if clear_table?
+        Colonization.configuration.count_of_instances_to_seed - model.count
+      else
+        Colonization.configuration.count_of_instances_to_seed
+      end
+    end
+
+    def values_generator
+      nil #TODO config[model][values_generator] || cofig[values_generator]
+    end
+
+    def clear_table?
+      Colonization.configuration.clear_table
+    end
+
+  end
+end

data/lib/colonization/strategies/default/model_relation.rb

@@ -0,0 +1,38 @@
+require 'colonization/model_instances_repository'
+require 'colonization/strategies/default/colonizer'
+
+module Colonization::Strategies::Default
+  class ModelRelation
+
+    attr_accessor :reflection
+
+    def initialize(reflection)
+      @reflection = reflection
+    end
+
+    def name
+      reflection.name
+    end
+
+    def value
+      found_instances = Colonization::ModelInstancesRepository.new.get_instance_without_relation(
+                          model: reflection_class,
+                          relation: reflection_owner,
+                          limit: Colonization.configuration.seed_index
+                        )
+      count_to_colonize = Colonization.configuration.seed_index - (found_instances ? found_instances.count : 0)
+      Colonization::Strategies::Default::Colonizer.new(reflection_class).colonize(count_to_colonize)
+    end
+
+    private
+
+    def reflection_class
+      reflection.class_name.constantize
+    end
+
+    def reflection_owner
+      reflection.active_record
+    end
+
+  end
+end

data/lib/colonization/strategies/fixed/colonizer.rb

@@ -0,0 +1,105 @@
+require 'colonization/strategies/model_field'
+require 'colonization/strategies/fixed/model_relation'
+require 'colonization/configuration'
+
+module Colonization::Strategies::Fixed
+  class Colonizer
+
+    attr_accessor :model
+
+    def initialize(model)
+      @model = model
+    end
+
+    def print_action
+      puts "colonize #{model}"
+    end
+
+    def colonize(times = nil)
+      res = []
+      (times || model_instances_number).times do
+        res << model.create!(params_with_values)
+      end
+      res
+    end
+
+    def clear_table
+      model.destroy_all if clear_table?
+    end
+
+    private
+
+    def params_with_values
+      res = {}
+
+      model_fields_objects.each do |field_object|
+        res[field_object.name] = field_object.value
+      end
+
+      model_relation_objects.each do |field_object|
+        res[field_object.name] = field_object.value
+      end
+
+      res
+    end
+
+    def model_fields
+      model.column_names - unnecessary_fields
+    end
+
+    def model_relations_strings
+      all_relations.map do |relation|
+        "#{relation.name}_id"
+      end
+    end
+
+    def all_relations
+      model.reflect_on_all_associations
+    end
+
+    def belongs_to_relations
+      model.reflect_on_all_associations(:belongs_to) || []
+    end
+
+    def has_many_relations
+      model.reflect_on_all_associations(:has_many) || []
+    end
+
+    def unnecessary_fields
+      ["created_at", "updated_at", "id"] + model_relations_strings
+    end
+
+    def model_fields_objects
+      res = []
+      model_fields.each do |field|
+        res << Colonization::Strategies::ModelField.new(model, field, values_generator)
+      end
+      res
+    end
+
+    def model_relation_objects
+      res = []
+      belongs_to_relations.each do |reflection|
+        res << Colonization::Strategies::Fixed::ModelRelation.new(reflection)
+      end
+      res
+    end
+
+    def model_instances_number
+      if clear_table?
+        Colonization.configuration.count_of_instances_to_seed - model.count
+      else
+        Colonization.configuration.count_of_instances_to_seed
+      end
+    end
+
+    def values_generator
+      nil #TODO config[model][values_generator] || cofig[values_generator]
+    end
+
+    def clear_table?
+      Colonization.configuration.clear_table
+    end
+
+  end
+end

data/lib/colonization/strategies/fixed/model_relation.rb

@@ -0,0 +1,36 @@
+require 'colonization/model_instances_repository'
+require 'colonization/strategies/fixed/colonizer'
+
+module Colonization::Strategies::Fixed
+  class ModelRelation
+
+    attr_accessor :reflection
+
+    def initialize(reflection)
+      @reflection = reflection
+    end
+
+    def name
+      reflection.name
+    end
+
+    def value
+      instance = Colonization::ModelInstancesRepository.new.get_instance(model: reflection_class)
+      unless instance
+        instance = Colonization::Strategies::Fixed::Colonizer.new(reflection_class).colonize
+      end
+      instance
+    end
+
+    private
+
+    def reflection_class
+      reflection.class_name.constantize
+    end
+
+    def reflection_owner
+      reflection.active_record
+    end
+
+  end
+end

data/lib/colonization/strategies/model_field.rb

@@ -0,0 +1,45 @@
+require 'colonization/values_generator'
+
+module Colonization::Strategies
+  class ModelField
+
+    attr_reader :model, :field
+
+    def initialize(model, field, values_generator)
+      @model = model
+      @field = field
+      @values_generator = values_generator
+    end
+
+    def name
+      field
+    end
+
+    def value
+      values_generator.call
+    end
+
+    private
+
+    def values_generator
+      if @values_generator.class == Proc
+        @values_generator
+      else 
+        (@values_generator || default_values_generator).new(field_type, validators)
+      end
+    end
+
+    def default_values_generator
+      Colonization::ValuesGenerator
+    end
+
+    def validators
+      model.validators_on(field)
+    end
+
+    def field_type
+      model.columns_hash[field].type
+    end
+
+  end
+end

data/lib/colonization/values_generator.rb

@@ -5,7 +5,7 @@ class Colonization::ValuesGenerator
 
   attr_reader :validators, :validators_objects
 
-  def initialize(type, validators, strategy)
+  def initialize(type, validators)
     @config = { validations: {}, type: type }
     @validators = validators
     set_custom_validators_objects
@@ -58,6 +58,8 @@ class Colonization::ValuesGenerator
         raise DisaparateValidatorError
       when :presence
         return Faker::Number.between(1, 2)
+      when :exclusion
+        return ((1..100).to_a - validator.options[:in]).sample
       else
         raise validator.inspect
       end
@@ -70,6 +72,11 @@ class Colonization::ValuesGenerator
       case validator.type
       when :acceptance
         raise DisaparateValidatorError
+      when :exclusion
+        begin
+          num = Faker::Number.decimal(2)
+        end while validator.options[:in].include?(num)
+        return num
       else
         raise validator.inspect
       end
@@ -83,6 +90,11 @@ class Colonization::ValuesGenerator
       when :acceptance
         return validator.options[:accept].sample if validator.options[:accept].present?
         return "1"
+      when :exclusion
+        begin
+          word = Faker::Lorem.word
+        end while validator.options[:in].include?(word)
+        return word
       else
         raise validator.inspect
       end
@@ -95,6 +107,11 @@ class Colonization::ValuesGenerator
       case validator.type
       when :acceptance
         raise DisaparateValidatorError
+      when :exclusion
+        begin
+          sentence = Faker::Lorem.sentence
+        end while validator.options[:in].include?(sentence)
+        return sentence
       else
         raise validator.inspect
       end
@@ -102,11 +119,16 @@ class Colonization::ValuesGenerator
   end
 
   def date_case
-    return Faker::Time.between(300.days.ago, 300.days.since, :day) unless validators_objects.present?
+    return Faker::Date.between(30.days.ago, Date.today) unless validators_objects.present?
     validators_objects.each do |validator|
       case validator.type
       when :acceptance
         raise DisaparateValidatorError
+      when :exclusion
+        begin
+          date = Faker::Date.between(30.days.ago, Date.today)
+        end while validator.options[:in].include?(date)
+        return date
       else
         raise validator.inspect
       end
@@ -114,11 +136,16 @@ class Colonization::ValuesGenerator
   end
 
   def datetime_case
-    return Faker::Time.between(300.days.ago, 300.days.since, :day) unless validators_objects.present?
+    return Faker::Time.between(30.days.ago, Date.today) unless validators_objects.present?
     validators_objects.each do |validator|
       case validator.type
       when :acceptance
         raise DisaparateValidatorError
+      when :exclusion
+        begin
+          datetime = Faker::Time.between(30.days.ago, Date.today)
+        end while validator.options[:in].include?(datetime)
+        return datetime
       else
         raise validator.inspect
       end
@@ -131,6 +158,14 @@ class Colonization::ValuesGenerator
       case validator.type
       when :acceptance
         return true
+      when :exclusion
+        if validator.options[:in].exclude?(false)
+          return false
+        elsif validator.options[:in].exclude?(true)
+          return true
+        else
+          return nil
+        end
       else
         raise validator.inspect
       end
@@ -138,11 +173,16 @@ class Colonization::ValuesGenerator
   end
 
   def time_case
-    return Faker::Time.between(300.days.ago, 300.days.since, :day) unless validators_objects.present?
+    return Faker::Time.between(30.days.ago, Date.today) unless validators_objects.present?
     validators_objects.each do |validator|
       case validator.type
       when :acceptance
         raise DisaparateValidatorError
+      when :exclusion
+        begin
+          time = Faker::Time.between(30.days.ago, Date.today)
+        end while validator.options[:in].include?(time)
+        return time
       else
         raise validator.inspect
       end

data/lib/colonization/version.rb

@@ -1,3 +1,3 @@
 module Colonization
-  VERSION = "0.3.2"
+  VERSION = "0.3.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: colonization
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.3
 platform: ruby
 authors:
 - Denys Dvoriashyn
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-12-25 00:00:00.000000000 Z
+date: 2016-12-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -81,14 +81,15 @@ files:
 - bin/setup
 - colonization.gemspec
 - lib/colonization.rb
-- lib/colonization/colonizer.rb
 - lib/colonization/configuration.rb
 - lib/colonization/investigator.rb
-- lib/colonization/model_entity.rb
-- lib/colonization/model_field.rb
-- lib/colonization/model_instaces_repository.rb
-- lib/colonization/model_relation.rb
+- lib/colonization/model_instances_repository.rb
 - lib/colonization/ship.rb
+- lib/colonization/strategies/default/colonizer.rb
+- lib/colonization/strategies/default/model_relation.rb
+- lib/colonization/strategies/fixed/colonizer.rb
+- lib/colonization/strategies/fixed/model_relation.rb
+- lib/colonization/strategies/model_field.rb
 - lib/colonization/validators/active_model/absence_validator.rb
 - lib/colonization/validators/active_model/acceptance_validator.rb
 - lib/colonization/validators/active_model/confirmation_validator.rb

data/lib/colonization/colonizer.rb

@@ -1,111 +0,0 @@
-require 'colonization/model_field'
-require 'colonization/model_relation'
-require 'colonization/configuration'
-
-class Colonization::Colonizer
-
-  attr_accessor :model
-
-  def initialize(model)
-    @model = model
-  end
-
-  def print_action
-    puts "colonize #{model}"
-  end
-
-  def colonize(times = nil)
-    res = []
-    (times || model_instances_number).times do
-      res << model.create!(params_with_values)
-    end
-    res
-  end
-
-  def clear_table
-    model.destroy_all if clear_table?
-  end
-
-  private
-
-  def params_with_values
-    res = {}
-
-    model_fields_objects.each do |field_object|
-      res[field_object.name] = field_object.value
-    end
-
-    model_relation_objects.each do |field_object|
-      res[field_object.name] = field_object.value
-    end
-
-    res
-  end
-
-  def model_fields
-    model.column_names - unnecessary_fields
-  end
-
-  def model_relations_strings
-    all_relations.map do |relation|
-      "#{relation.name}_id"
-    end
-  end
-
-  def all_relations
-    model.reflect_on_all_associations
-  end
-
-  def belongs_to_relations
-    model.reflect_on_all_associations(:belongs_to) || []
-  end
-
-  def has_many_relations
-    model.reflect_on_all_associations(:has_many) || []
-  end
-
-  def unnecessary_fields
-    ["created_at", "updated_at", "id"] + model_relations_strings
-  end
-
-  def model_fields_objects
-    res = []
-    model_fields.each do |field|
-      res << Colonization::ModelField.new(model, field, values_generator, strategy)
-    end
-    res
-  end
-
-  def model_relation_objects
-    res = []
-    needed_relations.each do |reflection|
-      res << Colonization::ModelRelation.new(reflection, strategy)
-    end
-    res
-  end
-
-  def needed_relations
-    strategy == :fixed ? belongs_to_relations : has_many_relations
-  end
-
-  def model_instances_number
-    if clear_table?
-      Colonization.configuration.count_of_instances_to_seed - model.count
-    else
-      Colonization.configuration.count_of_instances_to_seed
-    end
-  end
-
-  def values_generator
-    nil #TODO config[model][values_generator] || cofig[values_generator]
-  end
-
-  def strategy
-    Colonization.configuration.seed_type
-  end
-
-  def clear_table?
-    Colonization.configuration.clear_table
-  end
-
-end

data/lib/colonization/model_entity.rb

@@ -1,11 +0,0 @@
-class Colonization::ModelEntity
-
-  def name
-    raise NotImplementedException
-  end
-
-  def value
-    raise NotImplementedException
-  end
-
-end

data/lib/colonization/model_field.rb

@@ -1,45 +0,0 @@
-require 'colonization/model_entity'
-require 'colonization/values_generator'
-
-class Colonization::ModelField < Colonization::ModelEntity
-
-  attr_reader :model, :field, :strategy
-
-  def initialize(model, field, values_generator, strategy)
-    @model = model
-    @field = field
-    @values_generator = values_generator
-    @strategy = strategy
-  end
-
-  def name
-    field
-  end
-
-  def value
-    values_generator.call
-  end
-
-  private
-
-  def values_generator
-    if @values_generator.class == Proc
-      @values_generator
-    else 
-      (@values_generator || default_values_generator).new(field_type, validators, strategy)
-    end
-  end
-
-  def default_values_generator
-    Colonization::ValuesGenerator
-  end
-
-  def validators
-    model.validators_on(field)
-  end
-
-  def field_type
-    model.columns_hash[field].type
-  end
-
-end

data/lib/colonization/model_relation.rb

@@ -1,54 +0,0 @@
-require 'colonization/model_entity'
-require 'colonization/model_instaces_repository'
-require 'colonization/colonizer'
-
-class Colonization::ModelRelation < Colonization::ModelEntity
-
-  attr_accessor :reflection, :strategy
-
-  def initialize(reflection, strategy)
-    @reflection = reflection
-    @strategy = strategy || {}
-  end
-
-  def name
-    reflection.name
-  end
-
-  def value
-    if strategy == :fixed #TODO: refactor this shit
-      generate_relation_value_with_fixed_strategy
-    else
-      generate_relation_value_with_default_strategy
-    end
-  end
-
-  private
-
-  def generate_relation_value_with_fixed_strategy
-    instance = Colonization::ModelInstancesRepository.new.get_instance(model: reflection_class)
-    unless instance
-      instance = Colonization::Colonizer.new(reflection_class).colonize
-    end
-    instance
-  end
-
-  def generate_relation_value_with_default_strategy
-    found_instances = Colonization::ModelInstancesRepository.new.get_instance_without_relation(
-                        model: reflection_class,
-                        relation: reflection_owner,
-                        limit: Colonization.configuration.seed_index
-                      )
-    count_to_colonize = Colonization.configuration.seed_index - (found_instances ? found_instances.count : 0)
-    Colonization::Colonizer.new(reflection_class).colonize(count_to_colonize)
-  end
-
-  def reflection_class
-    reflection.class_name.constantize
-  end
-
-  def reflection_owner
-    reflection.active_record
-  end
-
-end