model_probe 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0cff622189ea7fcacf08ecfb65e1b0f459f43a04
-  data.tar.gz: 5e02a614bfd05e297054ae6e0158198f412b34e5
+  metadata.gz: 63309a1abc353f70b51746460188b18635d131ea
+  data.tar.gz: 05b90d5cc95032b79bf93ac79aa1981265b9314c
 SHA512:
-  metadata.gz: edd4072303a148f7513c2c7d2c5fd51128fae4f654a78c7d8eb756207f3246f26cc79d55c61c99c98330ba88dc514607618cd0b56550250c91efe490bc30783d
-  data.tar.gz: 917f11eb0ebc37a0b186f97cee843e8c3e9bff8cb56c0211b368b1a7130652b97a693e3747ee521cafb8c13804d7b77d523b22a8c222a1654e3107fd0f52d602
+  metadata.gz: 54277c2c7bbe5ff57cbc2cf3e9728a77fb8f699e489f84e78a43fe98d2e75fb85ce66c0bd7d0fb9c3dd97f73435887e8d849fbfe1ba62a1e0b5c6abf5fdf912a
+  data.tar.gz: 388befa8639539e3623d398d2bc4bec309603f80616de1502046188bf13b4b83de793fa876c33910d1470b31539a8d9ba048cbf169fe7161a69eb4d0c013ed49

data/lib/model_probe/version.rb

@@ -1,3 +1,3 @@
 module ModelProbe
-  VERSION = "1.0.3"
+  VERSION = "1.0.4"
 end

data/lib/model_probe.rb

@@ -12,7 +12,7 @@ module ModelProbe
 
     columns.sort{ |a, b| a.name <=> b.name }.map do |column|
       name = column.name
-      name = "* #{name}" if column.name == primary_key
+      name = "* #{name}" if primary_key_column?(column)
       print yellow(name.to_s.rjust(name_pad))
       print " "
       print blue(column.type.to_s.ljust(type_pad, "."))
@@ -26,14 +26,13 @@ module ModelProbe
 
   # Prints a stub that can be used for a test fixture
   def print_fixture
-    name = self.name.underscore
-    hash = { name => {} }
-
-    columns.sort{ |a, b| a.name <=> b.name }.map do |column|
-      next if column.name =~ /^(created_at|updated_at)$/
-      hash[name][column.name] = column.default || "value"
+    values = columns.sort_by(&:name).each_with_object({}) do |column, memo|
+      next if primary_key_column?(column)
+      next if timestamp_column?(column)
+      memo[column.name] = column.default || "value"
     end
 
+    hash = { self.name.underscore => values }
     puts hash.to_yaml
     nil
   end
@@ -46,13 +45,13 @@ module ModelProbe
     puts if relation_columns.size > 0
     puts "  # relationships ............................................................."
     relation_columns.sort_by(&:name).each do |column|
-      next if column.name == primary_key
+      next if primary_key_column?(column)
       puts "  belongs_to :#{column.name.sub(/_id\z/, "")}" if column.name =~ /_id\z/
     end
     puts if relation_columns.size > 0 || validation_columns.size > 0
     puts "  # validations ..............................................................."
     validation_columns.sort_by(&:name).each do |column|
-      next if column.name == primary_key
+      next if primary_key_column?(column)
       puts "  validates :#{column.name}, presence: :true" unless column.null
       if %i(text string).include?(column.type) && column.limit.to_i > 0
         puts "  validates :#{column.name}, length: { maximum: #{column.limit} }"
@@ -75,6 +74,7 @@ module ModelProbe
     puts "  # private instance methods .................................................."
     puts "  private"
     puts "end"
+    nil
   end
 
   private
@@ -91,6 +91,14 @@ module ModelProbe
     end
   end
 
+  def primary_key_column?(column)
+    column.name == primary_key
+  end
+
+  def timestamp_column?(column)
+    column.type == :datetime && column.name =~ /(created|updated|modified)/
+  end
+
   def relation_column?(column)
     return false if column.name == primary_key
     column.name.end_with?("_id")
@@ -101,5 +109,4 @@ module ModelProbe
     return true unless column.null
     %i(text string).include?(column.type) && column.limit.to_i > 0
   end
-
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: model_probe
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Nathan Hopkins
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-08-12 00:00:00.000000000 Z
+date: 2016-08-13 00:00:00.000000000 Z
 dependencies: []
 description: 
 email: