Include Decoratable in Mongoid::Document

Closes #375
This commit is contained in:
Andrew Haines 2012-12-11 20:59:49 +00:00
parent 7efeaa492b
commit 051dca6f30
2 changed files with 6 additions and 4 deletions

View File

@ -37,7 +37,7 @@ module Draper
end end
end end
def self.setup_active_record(base) def self.setup_orm(base)
base.class_eval do base.class_eval do
include Draper::Decoratable include Draper::Decoratable
end end

View File

@ -30,9 +30,11 @@ module Draper
end end
end end
initializer "draper.setup_active_record" do |app| initializer "draper.setup_orm" do |app|
ActiveSupport.on_load :active_record do [:active_record, :mongoid].each do |orm|
Draper.setup_active_record self ActiveSupport.on_load orm do
Draper.setup_orm self
end
end end
end end