From aecc014d330b57d67e5a02f3b109ffde50da7b62 Mon Sep 17 00:00:00 2001 From: Jinzhu Date: Sat, 17 Jul 2010 10:20:07 +0800 Subject: [PATCH] Attributes for devise generator --- lib/generators/active_record/devise_generator.rb | 4 +++- lib/generators/active_record/templates/migration.rb | 4 ++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/lib/generators/active_record/devise_generator.rb b/lib/generators/active_record/devise_generator.rb index d6fd76c8..0ea6806b 100644 --- a/lib/generators/active_record/devise_generator.rb +++ b/lib/generators/active_record/devise_generator.rb @@ -4,6 +4,8 @@ require 'generators/devise/orm_helpers' module ActiveRecord module Generators class DeviseGenerator < ActiveRecord::Generators::Base + argument :attributes, :type => :array, :default => [], :banner => "field:type field:type" + include Devise::Generators::OrmHelpers source_root File.expand_path("../templates", __FILE__) @@ -23,4 +25,4 @@ CONTENT end end end -end \ No newline at end of file +end diff --git a/lib/generators/active_record/templates/migration.rb b/lib/generators/active_record/templates/migration.rb index dcc2aeac..53a60e48 100644 --- a/lib/generators/active_record/templates/migration.rb +++ b/lib/generators/active_record/templates/migration.rb @@ -10,6 +10,10 @@ class DeviseCreate<%= table_name.camelize %> < ActiveRecord::Migration # t.lockable :lock_strategy => :<%= Devise.lock_strategy %>, :unlock_strategy => :<%= Devise.unlock_strategy %> # t.token_authenticatable +<% for attribute in attributes -%> + t.<%= attribute.type %> :<%= attribute.name %> +<% end -%> + t.timestamps end