factoring out the shouldArchive logic

1 files changed, 6 insertions, 4 deletions

diff --git a/plugins/VisitorInterest/VisitorInterest.php b/plugins/VisitorInterest/VisitorInterest.php
index b3f948d93e..583e8c31da 100644
--- a/plugins/VisitorInterest/VisitorInterest.php
+++ b/plugins/VisitorInterest/VisitorInterest.php
@@ -126,19 +126,21 @@ class Piwik_VisitorInterest extends Piwik_Plugin
     function archivePeriod($notification)
     {
         $archiveProcessing = $notification->getNotificationObject();
-        if (!$archiveProcessing->shouldProcessReportsForPlugin($this->getPluginName())) return;
 
         $archiving = new Piwik_VisitorInterest_Archiver($archiveProcessing);
-        $archiving->archivePeriod();
+        if($archiving->shouldArchive()) {
+            $archiving->archivePeriod();
+        }
     }
 
     public function archiveDay($notification)
     {
         $archiveProcessing = $notification->getNotificationObject();
-        if (!$archiveProcessing->shouldProcessReportsForPlugin($this->getPluginName())) return;
 
         $archiving = new Piwik_VisitorInterest_Archiver($archiveProcessing);
-        $archiving->archiveDay();
+        if($archiving->shouldArchive()) {
+            $archiving->archiveDay();
+        }
     }