lib/ar2gostruct.rb in ar2gostruct-0.0.1 vs lib/ar2gostruct.rb in ar2gostruct-0.0.2

@@ -85,17 +85,13 @@
     puts "// #{model_file_name}"
     puts info
   end
 
   def self.get_model_names
-    models = ARGV.dup
-    models.shift
-
-    if models.empty?
-      Dir.chdir(MODEL_DIR) do
-        models = Dir["**/*.rb"]
-      end
+    models = []
+    Dir.chdir(MODEL_DIR) do
+      models = Dir["**/*.rb"]
     end
     models
   end
 
   def self.convert!
@@ -107,10 +103,10 @@
         if klass < ActiveRecord::Base && !klass.abstract_class?
           annotated << class_name
           self.convert_to_gostruct(klass)
         end
       rescue Exception => e
-        puts "// Unable to annotate #{class_name}: #{e.message}"
+        puts "// Unable to convert #{class_name}: #{e.message}"
       end
 
     end
   end
 end