console_util 0.3.1 → 0.4.0

data/lib/console_util/version.rb

@@ -1,3 +1,3 @@
 module ConsoleUtil
-  VERSION = "0.3.1"
+  VERSION = "0.4.0"
 end

data/lib/console_util.rb

@@ -26,18 +26,24 @@ module ConsoleUtil
     # Import a MySQL dump file.
     # [This may work with other databases, but has only been tested with MySQL.]
     def import_mysql_dump(filename)
-      sql_dump_file = File.open(Rails.root.join('sql', "#{filename}.sql"))
+      sql_dump_file = File.open(filename)
       while sql_statement = sql_dump_file.gets(";\n") do
         ActiveRecord::Base.connection.execute(sql_statement) unless sql_statement.blank?
       end
     end
 
-    # Load all models into the environment
-    def load_all_models
-      Rails.root.join('app/models').each_entry do |model_file|
-        model_match = model_file.to_s.match(/^(.*)\.rb$/)
-        model_match[1].camelize.constantize if model_match
-      end
+    # Load all models into the environment and return them.
+    #
+    # Options:
+    #   :active_record  If true, returns only ActiveRecord models.
+    def load_all_models(options = {})
+      model_path      = Rails.root.join("app/models")
+      model_filenames = Dir[model_path.join("**/*.rb")]
+      model_filepaths = model_filenames.map { |filename| Pathname(filename).relative_path_from(model_path) }
+      model_names     = model_filepaths.map { |filepath| filepath.to_s.chomp(".rb").camelize }
+      models          = model_names.map(&:constantize)
+      models          = models.select { |model| model.is_a?(Class) && (model < ActiveRecord::Base) } if options[:active_record]
+      models
     end
 
     # Allows you to filter output to the console using grep

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: console_util
 version: !ruby/object:Gem::Version 
-  hash: 17
+  hash: 15
   prerelease: 
   segments: 
   - 0
-  - 3
-  - 1
-  version: 0.3.1
+  - 4
+  - 0
+  version: 0.4.0
 platform: ruby
 authors: 
 - Clyde Law
@@ -15,8 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-07-24 00:00:00 -07:00
-default_executable: 
+date: 2013-07-16 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rails
@@ -52,7 +51,6 @@ files:
 - lib/console_util.rb
 - lib/console_util/color.rb
 - lib/console_util/version.rb
-has_rdoc: true
 homepage: http://github.com/FutureAdvisor/console_util
 licenses: 
 - MIT
@@ -82,7 +80,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: console_util
-rubygems_version: 1.6.2
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: Contains various utilities for working in the Rails console.