fixture_me 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b938988f5a24da9822880fff81e66acf182e2e9d
-  data.tar.gz: 3a8ca5c16344831560dcdea3a0a3f5cc290fadbd
+  metadata.gz: d9810644f0c121be6dabdbe00e68646a685bc211
+  data.tar.gz: edaf8ae132ee5c0184a3058e3aebea5d1538a931
 SHA512:
-  metadata.gz: 6fd4b29acf6448bddd1820af3c491857ce6176b459aaaca4be7c5679fc6a873afd5475a925bd8ff0818be7b8e849d01c172d075fac7799362463928f0044910e
-  data.tar.gz: f8f216a50b83c4e33f95c104e5a9c559f6000be9a83a9bf6139ed8c3a8021a8ffbef1861c05f9ff07c02332351326d7b92ee74b2c53bd391cbc0d51adb387eb5
+  metadata.gz: 710dd27e7733e1aa35076e9870f1e8ae968a834ba48030ebe203df62592726ecda5eeb8a85d1c86f5bad2c261f801e34fc6a65ab737f2c5720c367cf22815a75
+  data.tar.gz: 5503d39655aabfd0a3e282f15f8d5bc69c5b3e511e3f51db7bf52daab1b3fddd4639fded7be55cd3770b0867a8ccd91386a898c1bbf4d52152feddb3228116b5

data/README.md

@@ -28,18 +28,26 @@ Or install it yourself as:
  rails console
 ```
 
-Oce your are inside Rails console
+Once your are inside Rails console
 
 To generate fixtures for all models.
 
 A new directory called fixtures would be created inside tmp directory (this is to make sure that this fixture generation would not override existing fixtures)
 
+
 ```ruby
- FixtureMe::AddFixtures.create_all_fixtures
+ fixme = FixtureMe::AddFixtures.new
+ fixme.create_all_fixtures
 ```
 
+to exclude created_at and updated_at columns
 
+```ruby
+ fixme = FixtureMe::AddFixtures.new
+ fixme.create_all_fixtures_no_timestamps
+```
 
+ 
 To generate fixtures one by one
 
 a mymodel.yml file would be generated  inside that file does not exist in the test/fixtures directory

data/lib/fixture_me/version.rb

@@ -1,3 +1,3 @@
 module FixtureMe
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/fixture_me.rb

@@ -1,5 +1,5 @@
 require "fixture_me/version"
-
+require 'fileutils'
 module FixtureMe
 
    def add_fixture
@@ -19,18 +19,88 @@ module FixtureMe
    end
 
 
+   def add_fixture_no_id_timestamps
+     hash = self.attributes.delete_if { |k, v| ["created_at", "updated_at", "id"].include? k }
+     md5 = Digest::MD5.hexdigest(hash.to_s)
+     string = "#{md5}:\n"
+     hash.each do |key,value|
+       value.gsub!("\n","\n"+" "*4) if value.class == String
+       string += " "*2 + "#{key.to_s}: #{value}\n" unless value.nil?
+     end
+     string += "\n"
+     table_name = self.class.table_name
+     file_name = table_name + ".yml"
+     File.open("#{Rails.root}/test/fixtures/#{file_name}", 'a') do |out|
+       out.write(string)
+     end
+   end
+
+
 
   class AddFixtures
 
 
-     def self.create_all_fixtures
+     def  initialize
+        @fixtures_dir =  FileUtils.mkdir_p( "#{Rails.root}/tmp/fixtures/").first
+     end
+
+     def fixtures_dir
+       @fixtures_dir
+     end
+
+
+     def all_models
+       # must eager load all the classes...
+       Dir.glob("#{RAILS_ROOT}/app/models/**/*.rb") do |model_path|
+         begin
+          require model_path
+         rescue
+          # ignore
+        end
+      end
+       # simply return them
+       ActiveRecord::Base.send(:subclasses)
+        #Dir.glob("#{Rails.root}/app/models/*.rb").map{|x| x.split("/").last.split(".").first.camelize}
+      end
+
+
+
+     def  get_list_of_unique_models_with_db_table
+
+
+              table_names = ActiveRecord::Base.connection.tables #.map{|a| a.capitalize.singularize}
+              model_names = Dir["#{Rails.root}/app/models/**/*.rb"].map {|f| File.basename(f, '.*').pluralize}
+
+              #Rails.application.eager_load! unless Rails.configuration.cache_classes
+              #ActiveRecord::Base.descendants
+              modelswithtables =  table_names & model_names
+
+     end
+
+
+     def  create_fixture(table_name, sql, model)
+
+        File.open("#{fixtures_dir}#{table_name}.yml", "w") do |file|
+          objects = ActiveRecord::Base.connection.select_all(sql)
+          objects.each_with_index do |obj, i|
+                model.columns.each do |col|
+                  if !col.null && obj[col.name].nil?
+                    obj[col.name] = ''
+                  end
+                end
+            file.write({"#{table_name}#{i}" => obj}.to_yaml.sub('---', ''))
+            file.write "\n"
+          end
+        end
+
+
+     end
 
-      fixtures_dir = "#{Rails.root}/tmp/fixtures/"
-      FileUtils.mkdir_p(fixtures_dir)
 
-      table_names = ActiveRecord::Base.connection.tables #.map{|a| a.capitalize.singularize}
-      model_names = Dir["#{Rails.root}/app/models/**/*.rb"].map {|f| File.basename(f, '.*').pluralize}
-      modelswithtables =    table_names & model_names
+     def create_all_fixtures
+
+      modelswithtables = get_list_of_unique_models_with_db_table
+
       modelswithtables.each do |table_name|
             model = table_name.classify.constantize
 
@@ -40,23 +110,45 @@ module FixtureMe
             else
               sql = "SELECT * FROM #{table_name}"
             end
-            File.open("#{fixtures_dir}#{table_name}.yml", "w") do |file|
-              objects = ActiveRecord::Base.connection.select_all(sql)
-              objects.each_with_index do |obj, i|
-                model.columns.each do |col|
-                  if !col.null && obj[col.name].nil?
-                    obj[col.name] = ''
-                  end
-                end
-                file.write({"#{table_name}#{i}" => obj}.to_yaml.sub('---', ''))
-                file.write "\n"
-              end
+
+            self.create_fixture(table_name, sql, model)
+
+            puts "extracted #{table_name}"
+          end
+
+     end
+
+
+
+
+     def create_all_fixtures_no_timestamps
+
+      modelswithtables = get_list_of_unique_models_with_db_table
+      exclude_columns = ['created_at', 'updated_at']
+
+
+      modelswithtables.each do |table_name|
+            model = table_name.classify.constantize
+            columns = model.attribute_names - exclude_columns
+
+            if model.columns.any?{|c| c.name == 'created_at'}
+              #sql = "SELECT * FROM #{table_name} ORDER BY created_at DESC"
+                sql =   model.select(columns).order("created_at DESC").to_sql
+            else
+              #sql = "SELECT * FROM #{table_name}"
+               sql =   model.select(columns).to_sql
             end
+
+            self.create_fixture(table_name, sql, model)
+
             puts "extracted #{table_name}"
           end
 
      end
 
+
+
+
   #FixtureMe::AddFixtures.create_all_fixtures
   end
 
@@ -64,3 +156,6 @@ module FixtureMe
 
 end
 ActiveRecord::Base.send(:include, FixtureMe)
+
+#c = FixtureMe::AddFixtures.new
+#c.create_all_fixtures

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fixture_me
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - anthony de silva
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-20 00:00:00.000000000 Z
+date: 2014-11-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -52,6 +52,7 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- fixture_me-0.0.1.gem
 - fixture_me.gemspec
 - lib/.DS_Store
 - lib/fixture_me.rb