From 7707f5ff8e703061c90953945ea43260cbabd3d2 Mon Sep 17 00:00:00 2001
From: Martin Lensment <mlensment@gmail.com>
Date: Mon, 2 Mar 2015 17:38:04 +0200
Subject: [PATCH] Fix relations sql

---
 lib/tasks/import.rake | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/lib/tasks/import.rake b/lib/tasks/import.rake
index ef794ae8b..5bb18b73f 100644
--- a/lib/tasks/import.rake
+++ b/lib/tasks/import.rake
@@ -148,7 +148,14 @@ namespace :import do
     Address.import address_columns, addresses, validate: false
 
     puts '-----> Updating relations...'
-    ActiveRecord::Base.connection.execute('UPDATE addresses SET contact_id = legacy_contact_id WHERE legacy_contact_id IS NOT NULL AND contact_id IS NULL')
+    ActiveRecord::Base.connection.execute(
+      "UPDATE addresses "\
+      "SET contact_id = contacts.id "\
+      "FROM contacts "\
+      "WHERE contacts.legacy_id = legacy_contact_id "\
+      "AND legacy_contact_id IS NOT NULL "\
+      "AND contact_id IS NULL"
+    )
     puts "-----> Imported #{count} new contacts in #{(Time.now.to_f - start).round(2)} seconds"
   end
 end