From b9a072b61ab5a0a2b8ff1b3fc67e2813c35aff95 Mon Sep 17 00:00:00 2001 From: Rohan Mitchell Date: Wed, 31 Oct 2012 14:32:47 +1100 Subject: [PATCH] Make sure enterprises have a valid address after pickup address field has been removed --- ...0_remove_pickup_address_from_enterprises.rb | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/db/migrate/20121028070200_remove_pickup_address_from_enterprises.rb b/db/migrate/20121028070200_remove_pickup_address_from_enterprises.rb index 2aa38222b4..d1c7fd0d56 100644 --- a/db/migrate/20121028070200_remove_pickup_address_from_enterprises.rb +++ b/db/migrate/20121028070200_remove_pickup_address_from_enterprises.rb @@ -1,5 +1,21 @@ class RemovePickupAddressFromEnterprises < ActiveRecord::Migration - def change + class Enterprise < ActiveRecord::Base; end + + def up + Enterprise.all.each do |e| + e.address_id ||= e.pickup_address_id + e.save! + end + remove_column :enterprises, :pickup_address_id end + + def down + add_column :enterprises, :pickup_address_id, :integer + + Enterprise.all.each do |e| + e.pickup_address_id ||= e.address_id + e.save! + end + end end