ensure class names exist first in migration

refinery · Jul 23, 2015 · 327c2f0 · 327c2f0
1 parent ae20814
commit 327c2f0
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/pages/db/migrate/20150720155305_update_slug_and_title_in_refinery_page_parts.rb b/pages/db/migrate/20150720155305_update_slug_and_title_in_refinery_page_parts.rb
@@ -1,11 +1,13 @@
 class UpdateSlugAndTitleInRefineryPageParts < ActiveRecord::Migration
   def change
-    Refinery::PagePart.all.each do |pp|
-      unless pp.title
-        pp.title = pp.slug
+    begin
+      ::Refinery::PagePart.all.each do |pp|
+        pp.title ||= pp.slug
+        pp.slug = pp.slug.downcase.gsub(" ", "_")
+        pp.save!
       end
-      pp.slug = pp.slug.downcase.gsub(" ", "_")
-      pp.save!
+    rescue NameError
+      warn "Refinery::PagePart was not defined!"
     end
   end
-end
+end