allow user to specify devise model name when executing install_admin …

…generator
cameronpriest · Dec 16, 2010 · ff9c3db · ff9c3db
1 parent 155bf99
commit ff9c3db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/generators/rails_admin/install_admin_generator.rb b/lib/generators/rails_admin/install_admin_generator.rb
@@ -1,6 +1,7 @@
 module RailsAdmin
   class InstallAdminGenerator < Rails::Generators::Base
     source_root File.expand_path('../templates', __FILE__)
+    argument :model_name, :type => :string, :default => 'user'
 
     desc "RailsAdmin Install"
 
@@ -61,7 +62,7 @@ def parse_route_files
     def set_devise
       puts "Setting up devise for you!
 ======================================================"
-      invoke 'devise', ['user']
+      invoke 'devise', [model_name]
     end
 
     def copy_locales_files