From 2430bfebf2f5ed3471e88f0b8ed2597b603c96e6 Mon Sep 17 00:00:00 2001
From: Stas <stanislav@perfectline.co>
Date: Tue, 26 Jan 2016 17:17:36 +0200
Subject: [PATCH] 111864739-contact_type_added

---
 lib/tasks/documents.rake | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/lib/tasks/documents.rake b/lib/tasks/documents.rake
index d16025785..5ae8fb4df 100644
--- a/lib/tasks/documents.rake
+++ b/lib/tasks/documents.rake
@@ -1,6 +1,5 @@
 namespace :documents do
 
-
   desc 'Generate all'
   task all: :environment do
     Rake::Task['documents:log'].invoke
@@ -12,11 +11,16 @@ namespace :documents do
     puts '-----> Adding documets id for PaperTrail log...'
     count = 0
 
-    LegalDocument.where(documentable_type: Domain).find_each do |x|
+    LegalDocument.find_each do |x|
 
       next if x.documentable_id.blank?
 
-      dc = DomainVersion.where(item_id: x.documentable_id)
+      document_type = case x.documentable_type
+                        when 'Domain' then DomainVersion
+                        when 'Contact'then ContactVersion
+                      end
+
+      dc = document_type.where(item_id: x.documentable_id)
 
       dc.each do |y|