Added newsletter subscriber anonymization and error handling

Author: avstudnitz

app/code/community/IntegerNet/Anonymizer/Model/Anonymizer.php

@@ -30,6 +30,8 @@ public function anonymizeAll()
 
         $this->_anonymizeCustomers($customers);
 
+        $this->_anonymizeRemainingNewsletterSubscribers();
+
         $this->_anonymizeRemainingOrders();
         $this->_anonymizeRemainingQuotes();
 
@@ -70,24 +72,18 @@ protected function _anonymizeCustomer($customer)
         $customer->getResource()->save($customer);
         $this->_anonymizedCustomerIds[] = $customer->getId();
 
+        /* @var $subscriber Mage_Newsletter_Model_Subscriber */
+        $subscriber = Mage::getModel('newsletter/subscriber');
+        $subscriber->loadByEmail($customer->getOrigData('email'));
+        if ($subscriber->getId()) {
+            $this->_anonymizeNewsletterSubscriber($subscriber, $randomData);
+        }
+
         $this->_anonymizeQuotes($customer, $randomData);
         $this->_anonymizeOrders($customer, $randomData);
         $this->_anonymizeCustomerAddresses($customer, $randomData);
     }
 
-    /**
-     * @return array
-     */
-    protected function _getRandomData()
-    {
-        $randomData = array_pop($this->_unusedCustomerData);
-        if (is_null($randomData)) {
-            $this->_fetchRandomCustomerData(100);
-            $randomData = array_pop($this->_unusedCustomerData);
-        }
-        return $randomData;
-    }
-
     /**
      * @param Mage_Customer_Model_Customer $customer
      * @param array $randomData
@@ -396,6 +392,35 @@ protected function _anonymizeOrderAddress($orderAddress, $randomData)
         }
     }
 
+    /**
+     *
+     */
+    protected function _anonymizeRemainingNewsletterSubscribers()
+    {
+        $newsletterSubscribers = Mage::getModel('newsletter/subscriber')
+            ->getCollection()
+            ->addFieldToFilter('subscriber_id', array('nin' => $this->_anonymizedNewsletterSubscriberIds));
+
+        foreach($newsletterSubscribers as $newsletterSubscriber) {
+
+            /** @var $newsletterSubscriber Mage_Newsletter_Model_Subscriber */
+            $randomData = $this->_getRandomData();
+            $this->_anonymizeNewsletterSubscriber($newsletterSubscriber, $randomData);
+        }
+    }
+
+    /**
+     * @param Mage_Newsletter_Model_Subscriber $subscriber
+     * @param array $randomData
+     */
+    protected function _anonymizeNewsletterSubscriber($subscriber, $randomData)
+    {
+        $subscriber->setData('subscriber_email', $randomData['email']);
+        $subscriber->getResource()->save($subscriber);
+
+        $this->_anonymizedNewsletterSubscriberIds[] = $subscriber->getId();
+    }
+
     /**
      * @return array
      */
@@ -465,6 +490,19 @@ protected function _getAddressMapping()
         );
     }
 
+    /**
+     * @return array
+     */
+    protected function _getRandomData()
+    {
+        $randomData = array_pop($this->_unusedCustomerData);
+        if (is_null($randomData)) {
+            $this->_fetchRandomCustomerData(100);
+            $randomData = array_pop($this->_unusedCustomerData);
+        }
+        return $randomData;
+    }
+
     /**
      * @param int $count
      * @return array
@@ -473,6 +511,9 @@ protected function _fetchRandomCustomerData($count)
     {
         $url = "http://fakester.biz/json?n=$count";
         $json = file_get_contents($url);
+        if ($json === false) {
+            Mage::throwException('Connection to http://fakester.biz failed.');
+        }
         $this->_unusedCustomerData = Zend_Json::decode($json);
 
         /*

shell/anonymizer.php

@@ -42,10 +42,14 @@ class Mage_Shell_Anonymizer extends Mage_Shell_Abstract
     public function run()
     {
         /** @var $anonymizer IntegerNet_Anonymizer_Model_Anonymizer */
-        $anonymizer = Mage::getModel('anonymizer/anonymizer');
-        $anonymizer->anonymizeAll();
-        foreach($anonymizer->getResults() as $resultLabel => $resultCount) {
-            echo 'Anonymized ' . $resultCount . ' ' . $resultLabel . ".\n";
+        try {
+            $anonymizer = Mage::getModel('anonymizer/anonymizer');
+            $anonymizer->anonymizeAll();
+            foreach ($anonymizer->getResults() as $resultLabel => $resultCount) {
+                echo 'Anonymized ' . $resultCount . ' ' . $resultLabel . ".\n";
+            }
+        } catch (Exception $e) {
+            echo 'Error: ' . $e->getMessage();
         }
     }