c0f6fa182f
Generalize the Budget::Investment::Milestone model to a polymorphic Milestone model so it can be used for entities other than Budget::Investment.
90 lines
2.2 KiB
Ruby
90 lines
2.2 KiB
Ruby
namespace :globalize do
|
|
def translatable_classes
|
|
[
|
|
AdminNotification,
|
|
Banner,
|
|
Milestone,
|
|
I18nContent,
|
|
Legislation::DraftVersion,
|
|
Legislation::Process,
|
|
Legislation::Question,
|
|
Legislation::QuestionOption,
|
|
Poll,
|
|
Poll::Question,
|
|
Poll::Question::Answer,
|
|
SiteCustomization::Page,
|
|
Widget::Card
|
|
]
|
|
end
|
|
|
|
def migrate_data
|
|
@errored = false
|
|
|
|
translatable_classes.each do |model_class|
|
|
logger.info "Migrating #{model_class} data"
|
|
|
|
fields = model_class.translated_attribute_names
|
|
|
|
model_class.find_each do |record|
|
|
fields.each do |field|
|
|
locale = if model_class == SiteCustomization::Page && record.locale.present?
|
|
record.locale
|
|
else
|
|
I18n.locale
|
|
end
|
|
|
|
translated_field = record.localized_attr_name_for(field, locale)
|
|
|
|
if record.send(translated_field).blank?
|
|
record.send(:"#{translated_field}=", record.untranslated_attributes[field.to_s])
|
|
end
|
|
end
|
|
|
|
begin
|
|
record.save!
|
|
rescue ActiveRecord::RecordInvalid
|
|
logger.warn "Failed to save #{model_class} with id #{record.id}"
|
|
@errored = true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def logger
|
|
@logger ||= Logger.new(STDOUT).tap do |logger|
|
|
logger.formatter = proc { |severity, _datetime, _progname, msg| "#{severity} #{msg}\n" }
|
|
end
|
|
end
|
|
|
|
def errored?
|
|
@errored
|
|
end
|
|
|
|
desc "Simulates migrating existing data to translation tables"
|
|
task simulate_migrate_data: :environment do
|
|
logger.info "Starting migrate data simulation"
|
|
|
|
ActiveRecord::Base.transaction do
|
|
migrate_data
|
|
raise ActiveRecord::Rollback
|
|
end
|
|
|
|
if errored?
|
|
logger.warn "Some database records will not be migrated"
|
|
else
|
|
logger.info "Migrate data simulation ended successfully"
|
|
end
|
|
end
|
|
|
|
desc "Migrates existing data to translation tables"
|
|
task migrate_data: :environment do
|
|
logger.info "Starting data migration"
|
|
migrate_data
|
|
logger.info "Finished data migration"
|
|
|
|
if errored?
|
|
logger.warn "Some database records couldn't be migrated; please check the log messages"
|
|
end
|
|
end
|
|
end
|