2 files changed, 6 insertions, 8 deletions

@@ -1,5 +1,7 @@
 module Pokeviewer
   class Location < ApplicationRecord
+    has_many :pokemon, dependent: :nullify
+
     validates :name, presence: true
   end
 end
@@ -24,6 +24,10 @@ module Pokeviewer
     validates :met_type, presence: true
     enumerize :met_type, in: [:normal, :hatched, :npc_trade, :fateful_encounter]
 
+    belongs_to :location, optional: true
+    validates :location, presence: true,
+      if: Proc.new { |c| c.met_type == :normal or c.met_type == :hatched}
+
     validates :gender, presence: true
     enumerize :gender, in: [:genderless, :female, :male]
 
@@ -105,14 +109,6 @@ module Pokeviewer
       (trainer.nil?) or (ot_name != trainer.name) or (ot_number != trainer.number)
     end
 
-    def location
-      if (met_type == :normal) or (met_type == :hatched)
-        Location.find_by_id(met_location)
-      else
-        nil
-      end
-    end
-
     def display_ot_number
       ot_number.to_s.rjust(5, '0')
     end