colonization 0.2.2 → 0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9351d53d58975968b6a9c7c4625246502821fe12
-  data.tar.gz: b46d5083fc72a7980e15c62777cb0c41ccf1add3
+  metadata.gz: e9907296322411ab264d5463c48bd55edfc83d5a
+  data.tar.gz: a7a487a862999ce8e6b35db4a32dec3b0fcf55ce
 SHA512:
-  metadata.gz: 47660950a66d668feb8c4a04041b97a08c10196abb6f4528319edd1dd6c35e7058fb66b073ddc00f3c77f95d614ad22bedcf93c2b857c1a90e20c75f82c3ac49
-  data.tar.gz: 967cb6e497fb73b2606cf9b7dbab994f82b2180aa1ef25e978e121918250cf935233af514474df4b8003992f537f65adc9e1d1af0fa3c1ee893392be1ea4d103
+  metadata.gz: ad86fc64496fac7815bab8ff3953c3611d78e866dd6cbefdd212db619e56113d0597884ee7b1a9c99544b4a3d1dd25f423623a78aaf6421125f8c1d161e3b06b
+  data.tar.gz: 78e4eaa67eaab63afaff5da1053b0984b937581c01f6ff7684bdaab3187b0d73c34b4bf258a69ce32196b461a774b447aa5797ca624c5de546cf947ae36dc11c

data/lib/colonization.rb

@@ -1,6 +1,17 @@
 require "colonization/version"
 require "colonization/ship"
+require "colonization/configuration"
 
 module Colonization
-  # Your code goes here...
+  class << self
+
+    def configure(&block)
+      yield(configuration)
+    end
+
+    def configuration
+      @configuration ||= Colonization::Configuration.new
+    end
+
+  end
 end

data/lib/colonization/colonizer.rb

@@ -1,30 +1,24 @@
 require 'colonization/model_field'
 require 'colonization/model_relation'
-require 'pry'
+require 'colonization/configuration'
 
 class Colonization::Colonizer
 
-  DEFAULT_MODEL_INSTANCES_NUMBER = 2
+  attr_accessor :model
 
   def initialize(model)
     @model = model
   end
 
-  def 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
 
@@ -71,17 +65,13 @@ class Colonization::Colonizer
   end
 
   def unnecessary_fields
-    ["created_at", "updated_at", "id"] + model_relations_strings + custom_excepted_fields
-  end
-
-  def custom_excepted_fields
-    [] #TODO
+    ["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(field))
+      res << Colonization::ModelField.new(model, field, values_generator, strategy)
     end
     res
   end
@@ -89,35 +79,33 @@ class Colonization::Colonizer
   def model_relation_objects
     res = []
     needed_relations.each do |reflection|
-      res << Colonization::ModelRelation.new(reflection, strategy(reflection))
+      res << Colonization::ModelRelation.new(reflection, strategy)
     end
     res
   end
 
   def needed_relations
-    strategy[:type] == :fixed ? belongs_to_relations : has_many_relations
+    strategy == :fixed ? belongs_to_relations : has_many_relations
   end
 
   def model_instances_number
-    #TODO config[model][:model_instances_number] || DEFAULT_MODEL_INSTANCES_NUMBER
-    DEFAULT_MODEL_INSTANCES_NUMBER - model.count
+    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(field = nil)
-    #TODO config[model][field]
-    { type: :default }
+  def strategy
+    Colonization.configuration.seed_type
   end
 
   def clear_table?
-    true #TODO config[model][:clear_table] || config[:clear_table] || true
-  end
-
-  def config
-    raise #TODO
+    Colonization.configuration.clear_table
   end
 
 end

data/lib/colonization/configuration.rb

@@ -0,0 +1,12 @@
+class Colonization::Configuration
+
+  attr_accessor :seed_type, :count_of_instances_to_seed, :seed_index, :clear_table
+
+  def initialize
+    @seed_type = :default
+    @seed_index = 3
+    @count_of_instances_to_seed = 100
+    @clear_table = true
+  end
+
+end

data/lib/colonization/model_relation.rb

@@ -4,19 +4,11 @@ require 'colonization/colonizer'
 
 class Colonization::ModelRelation < Colonization::ModelEntity
 
-  DEAFULT_SEED_INDEX = 3
+  attr_accessor :reflection, :strategy
 
   def initialize(reflection, strategy)
     @reflection = reflection
-    @strategy = strategy
-  end
-
-  def reflection
-    @reflection
-  end
-
-  def strategy
-    @strategy || {}
+    @strategy = strategy || {}
   end
 
   def name
@@ -24,7 +16,7 @@ class Colonization::ModelRelation < Colonization::ModelEntity
   end
 
   def value
-    if strategy[:type] == :fixed
+    if strategy == :fixed #TODO: refactor this shit
       generate_relation_value_with_fixed_strategy
     else
       generate_relation_value_with_default_strategy
@@ -42,11 +34,13 @@ class Colonization::ModelRelation < Colonization::ModelEntity
   end
 
   def generate_relation_value_with_default_strategy
-    seed_index = strategy[:seed_index] || DEAFULT_SEED_INDEX
-
-    found_instances = Colonization::ModelInstancesRepository.new.get_instance_without_relation(model: reflection_class, relation: reflection_owner, limit: seed_index)
-    result_instances = Colonization::Colonizer.new(reflection_class).colonize(seed_index - (found_instances ? found_instances.count : 0))
-    result_instances
+    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

data/lib/colonization/version.rb

@@ -1,3 +1,3 @@
 module Colonization
-  VERSION = "0.2.2"
+  VERSION = "0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: colonization
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: '0.3'
 platform: ruby
 authors:
 - Denys Dvoriashyn
@@ -82,6 +82,7 @@ files:
 - 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