read the model files

Eric-Guo · Oct 4, 2014 · c220967 · c220967
1 parent 5d7e1d0
commit c220967
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/lib/attr_accessible2strong_params.rb b/lib/attr_accessible2strong_params.rb
@@ -1,10 +1,19 @@
+require 'find'
 require 'active_support/core_ext/object'
 require 'active_support/inflector'
 
 class AttrAccessible2StrongParams
-  def self.convert(filename)
-    t = Converter.new
-    t.read_attr_accessible filename
+  def self.convert(file_or_dir)
+    @files = []
+    file_or_dir ||= '.'
+    if File.directory?(file_or_dir)
+      Find.find("#{file_or_dir}/app/models/") do |path|
+        next if path.end_with? '.rb.rb'
+        @files << path if path.end_with? '.rb'
+      end
+    else
+      @files << file_or_dir
+    end
   end
 end
 

diff --git a/test/test_attr_accessible2strong_params.rb b/test/test_attr_accessible2strong_params.rb
@@ -5,6 +5,10 @@
 require 'attr_accessible2strong_params'
 
 class AttrAccessible2StrongParamsTest < MiniTest::Test
+  def test_convert
+    AttrAccessible2StrongParams.convert('test/railsapp')
+  end
+
   def test_read_model_attr_accessible_sym_list
     c = AttrAccessible2StrongParams::Converter.new
     assert_equal [:create_badge, :create_shift_code, :in_time, :out_time, :profile_check, :update_badge, :update_shift_code] \