Safeguard rake task

jahseng-lee · Mar 31, 2024 · a23b166 · a23b166
1 parent 89f0b96
commit a23b166
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lib/tasks/countries.rake b/lib/tasks/countries.rake
@@ -7,6 +7,12 @@ namespace :countries do
       next if country.locations.empty?
 
       if Rails.env.production?
+        # Skip if one already exists
+        next if VisaInformation.find_by(
+          country_id: country.id,
+          citizenship_id: nil
+        ).present?
+
         VisaInformation.create!(
           country_id: country.id,
           citizenship_id: nil,
@@ -86,6 +92,11 @@ namespace :countries do
           # This is a manually created list of "citzenship" countries
           # which will be initially excluded
           next if NON_SEED_CITIZENSHIPS.include?(citizenship.name)
+          # Skip if one already exists
+          next if VisaInformation.find_by(
+            country_id: country.id,
+            citizenship_id: citizenship.id
+          ).present?
 
           if country.id == citizenship.id
             VisaInformation.create!(