From 56dbedc2f773df937015c88eb18f129054d4d6da Mon Sep 17 00:00:00 2001 From: Jeremiah Date: Fri, 8 Jun 2018 17:59:54 -0700 Subject: [PATCH] Copy andymcintosh fix for #62 --- lib/draftsman/draft.rb | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lib/draftsman/draft.rb b/lib/draftsman/draft.rb index 61acbda..13108c2 100644 --- a/lib/draftsman/draft.rb +++ b/lib/draftsman/draft.rb @@ -76,12 +76,13 @@ def draft_publication_dependencies associations.each do |association| association_class = if association.options.key?(:polymorphic) - my_item.send(association.foreign_key.sub('_id', '_type')).constantize + association_class_name = my_item.send(association.foreign_key.sub('_id', '_type')) + association_class_name.constantize if association_class_name else association.klass end - if association_class.draftable? && association.name != association_class.draft_association_name.to_sym + if association_class && association_class.draftable? && association.name != association_class.draft_association_name.to_sym dependency = my_item.send(association.name) dependencies << dependency.draft if dependency.present? && dependency.draft? && dependency.draft.create? end