Add data_migration:set_disease_types

This adds a data migration which sets the disease types on
`VaccinationRecord` to ensure it's populated for all vaccination
records.

Jira-Issue: MAV-2858

Author: thomasleese

lib/tasks/data_migration.rake

@@ -5,4 +5,16 @@ namespace :data_migration do
   task create_clinic_notifications: :environment do
     DataMigration::CreateClinicNotifications.call
   end
+
+  desc "Set disease types on vaccination records."
+  task set_disease_types: :environment do
+    VaccinationRecord
+      .includes(:vaccine)
+      .find_each do |vaccination_record|
+        disease_types =
+          vaccination_record.vaccine&.disease_types ||
+            vaccination_record.programme.disease_types
+        vaccination_record.update_columns(disease_types:)
+      end
+  end
 end