Made "met location" into an actual association

1 files changed, 30 insertions, 0 deletions

@@ -0,0 +1,30 @@
+class RenamePokemonMetLocation < ActiveRecord::Migration[5.1]
+  def up
+    add_column :pokeviewer_pokemon, :location_id, :integer, null: true
+
+    add_foreign_key :pokeviewer_pokemon, :pokeviewer_locations,
+      column: :location_id
+
+    Pokeviewer::Pokemon.all.each do |p|
+      unless p.met_location.nil?
+        p.location_id = p.met_location.to_i
+        p.save!
+      end
+    end
+
+    remove_column :pokeviewer_pokemon, :met_location
+  end
+
+  def down
+    add_column :pokeviewer_pokemon, :met_location, :string
+
+    Pokeviewer::Pokemon.all.each do |p|
+      unless p.location_id.nil?
+        p.met_location = p.location_id.to_s
+        p.save!
+      end
+    end
+
+    remove_column :pokeviewer_pokemon, :location_id
+  end
+end