如何在Rails的注册表单中创建Devise用户时创建另一个对象？

Question

我的系统中有不同类型的用户。其中一种是设计师：

class Designer < ActiveRecord::Base
  attr_accessible :user_id, :portfolio_id, :some_designer_specific_field
  belongs_to :user
  belongs_to :portfolio
end

当用户注册时，会立即创建一个账户。因此，当用户填写注册表单时，会创建一个Devise User对象，同时创建一个Designer对象，并将其user_id设置为新创建的User对象的ID。如果我可以访问控制器的代码，这很容易实现。但是使用Devise时，我无法访问此注册控制器。

如何在注册时正确地创建User和Designer对象？

- at.

Answer 1

在最近的项目中，我使用了表单对象模式来一步创建Devise用户和公司。这需要绕过Devise的RegistrationsController并创建自己的SignupsController。

# config/routes.rb
# Signups
get 'signup' => 'signups#new',     as: :new_signup
post 'signup' => 'signups#create', as: :signups  


# app/controllers/signups_controller.rb
class SignupsController < ApplicationController
  def new
    @signup = Signup.new
  end

  def create
    @signup = Signup.new(params[:signup])

    if @signup.save
      sign_in @signup.user
      redirect_to projects_path, notice: 'You signed up successfully.'
    else
      render action: :new
    end
  end
end

引用的注册模型被定义为一个表单对象。

# app/models/signup.rb

# The signup class is a form object class that helps with
# creating a user, account and project all in one step and form
class Signup
  # Available in Rails 4
  include ActiveModel::Model

  attr_reader :user
  attr_reader :account
  attr_reader :membership

  attr_accessor :name
  attr_accessor :company_name
  attr_accessor :email
  attr_accessor :password

  validates :name, :company_name, :email, :password, presence: true

  def save
    # Validate signup object
    return false unless valid?

    delegate_attributes_for_user
    delegate_attributes_for_account

    delegate_errors_for_user unless @user.valid?
    delegate_errors_for_account unless @account.valid?

    # Have any errors been added by validating user and account?
    if !errors.any?
      persist!
      true
    else
      false
    end
  end

  private

  def delegate_attributes_for_user
    @user = User.new do |user|
      user.name = name
      user.email = email
      user.password = password
      user.password_confirmation = password
    end
  end

  def delegate_attributes_for_account
    @account = Account.new do |account|
      account.name = company_name
    end
  end

  def delegate_errors_for_user
    errors.add(:name, @user.errors[:name].first) if @user.errors[:name].present?
    errors.add(:email, @user.errors[:email].first) if @user.errors[:email].present?
    errors.add(:password, @user.errors[:password].first) if @user.errors[:password].present?
  end

  def delegate_errors_for_account
    errors.add(:company_name, @account.errors[:name].first) if @account.errors[:name].present?
  end

  def persist!
    @user.save!
    @account.save!
    create_admin_membership
  end

  def create_admin_membership
    @membership = Membership.create! do |membership|
      membership.user = @user
      membership.account = @account
      membership.admin = true
    end
  end
end

关于表单对象的一篇优秀文章（也是我的工作来源）是这篇 CodeClimate 博客文章 Refactoring。

总的来说，我更喜欢这种方法而不是使用 accepts_nested_attributes_for，虽然可能还有更好的方法。如果你找到了，请告诉我！

===

编辑：添加了相关模型及其关联，以更好地理解。

class User < ActiveRecord::Base
  # Memberships and accounts
  has_many :memberships
  has_many :accounts, through: :memberships
end

class Membership < ActiveRecord::Base
  belongs_to :user
  belongs_to :account
end

class Account < ActiveRecord::Base
  # Memberships and members
  has_many :memberships, dependent: :destroy
  has_many :users, through: :memberships
  has_many :admins, through: :memberships,
                    source: :user,
                    conditions: { 'memberships.admin' => true }
  has_many :non_admins, through: :memberships,
                        source: :user,
                        conditions: { 'memberships.admin' => false }
end

这个模型中的结构是与 thoughtbot 公司的 gem saucy 一起建模的。据我所知，源代码不在 Github 上，但可以从 gem 中提取它。我通过重新建模学到了很多知识。

- Thomas Klemm

---

我们如何将其保存到数据库中，假设我们有Account、User和Membership的ActiveRecord模型。您介意发布表单本身的外观以及这些控制器的ActiveRecord模型吗？谢谢。 - brg

已添加模型结构。持久化由控制器中调用的@signup.save处理，该方法创建所有对象及其关联。 - Thomas Klemm

Answer 2

如果您不想更改注册控制器，一种方法是使用ActiveRecord回调函数。

class User < ActiveRecord::Base
  after_create :create_designer

  private

  def create_designer
      Designer.create(user_id: self.id) 
  end
end

- wcpaez

---

回调函数无法工作，因为我无法访问params哈希表。 - at.