diff --git a/app/models/legacy/object.rb b/app/models/legacy/object.rb index dc1c53d7b..04218cd09 100644 --- a/app/models/legacy/object.rb +++ b/app/models/legacy/object.rb @@ -1,6 +1,7 @@ module Legacy class Object < Db self.table_name = :object + belongs_to :registrar, foreign_key: :upid, primary_key: :legacy_id, class_name: '::Registrar' def self.instance_method_already_implemented?(method_name) return true if method_name == 'update' diff --git a/lib/tasks/import.rake b/lib/tasks/import.rake index 882ab55d0..b92b2ba5f 100644 --- a/lib/tasks/import.rake +++ b/lib/tasks/import.rake @@ -191,7 +191,7 @@ namespace :import do x.organization.try(:strip), x.object_registry.try(:registrar).try(:id), user, - x.object.try(:upid) ? x.object.try(:upid) : x.object_registry.try(:crid), + x.object.try(:registrar).try(:name) ? x.object.try(:registrar).try(:name) : x.object_registry.try(:registrar).try(:name), x.country.try(:strip), x.id, [x.street1.try(:strip), x.street2.try(:strip), x.street3.try(:strip)].join("\n"), @@ -320,7 +320,7 @@ namespace :import do 1, 'y', user, - x.object.try(:upid) ? x.object.try(:upid) : x.object_registry.try(:crid), + x.object.try(:registrar).try(:name) ? x.object.try(:registrar).try(:name) : x.object_registry.try(:registrar).try(:name), x.id, x.object_registry.try(:crid), x.registrant, @@ -332,7 +332,7 @@ namespace :import do domain_contacts << [ 'AdminDomainContact', user, - x.object.try(:upid) ? x.object.try(:upid) : x.object_registry.try(:crid), + x.object.try(:registrar).try(:name) ? x.object.try(:registrar).try(:name) : x.object_registry.try(:registrar).try(:name), x.id, dc.contactid ] @@ -364,7 +364,7 @@ namespace :import do ips[:ipv4].try(:strip), ips[:ipv6].try(:strip), user, - x.object.try(:upid) ? x.object.try(:upid) : x.object_registry.try(:crid), + x.object.try(:registrar).try(:name) ? x.object.try(:registrar).try(:name) : x.object_registry.try(:registrar).try(:name), x.id ] end if x.nsset && x.nsset.hosts @@ -378,7 +378,7 @@ namespace :import do 3, # ds_alg 1, # ds_digest_type /SHA1) user, - x.object.try(:upid) ? x.object.try(:upid) : x.object_registry.try(:crid), + x.object.try(:registrar).try(:name) ? x.object.try(:registrar).try(:name) : x.object_registry.try(:registrar).try(:name), x.id ] end