Added output for shell

Author: avstudnitz

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

@@ -9,6 +9,8 @@
 class IntegerNet_Anonymizer_Model_Anonymizer
 {
     protected $_unusedCustomerData = array();
+    protected $_anonymizedCustomerIds = array();
+    protected $_anonymizedCustomerAddressIds = array();
     protected $_anonymizedOrderIds = array();
     protected $_anonymizedOrderAddressIds = array();
     protected $_anonymizedQuoteIds = array();
@@ -30,6 +32,22 @@ public function anonymizeAll()
 
         $this->_anonymizeRemainingOrders();
         $this->_anonymizeRemainingQuotes();
+
+        $this->_anonymizeRemainingOrderAddresses();
+        $this->_anonymizeRemainingQuoteAddresses();
+    }
+
+    public function getResults()
+    {
+        return array(
+            'Customers' => sizeof($this->_anonymizedCustomerIds),
+            'Customers Addresses' => sizeof($this->_anonymizedCustomerAddressIds),
+            'Orders' => sizeof($this->_anonymizedOrderIds),
+            'Order Addresses' => sizeof($this->_anonymizedOrderAddressIds),
+            'Quotes' => sizeof($this->_anonymizedQuoteIds),
+            'Quote Addresses' => sizeof($this->_anonymizedQuoteAddressIds),
+            'Newsletter Subscribers' => sizeof($this->_anonymizedNewsletterSubscriberIds),
+        );
     }
 
     /**
@@ -63,6 +81,7 @@ protected function _anonymizeCustomer($customer)
         }
 
         $customer->getResource()->save($customer);
+        $this->_anonymizedCustomerIds[] = $customer->getId();
 
         $this->_anonymizeQuotes($customer, $randomData);
         $this->_anonymizeOrders($customer, $randomData);
@@ -128,37 +147,6 @@ protected function _anonymizeRemainingOrders()
         }
     }
 
-    /**
-     *
-     */
-    protected function _anonymizeRemainingQuotes()
-    {
-        $quotes = Mage::getModel('sales/quote')
-            ->getCollection()
-            ->addFieldToFilter('entity_id', array('nin' => $this->_anonymizedQuoteIds));
-
-        foreach($quotes as $quote) {
-
-            /** @var $quote Mage_Sales_Model_Quote */
-            $randomData = $this->_getRandomData();
-            $this->_anonymizeQuote($quote, $randomData);
-
-            foreach($quote->getAddressesCollection() as $quoteAddress) {
-
-                /** @var $quoteAddress Mage_Sales_Model_Quote_Address */
-                if ($quoteAddress->getCustomerFirstname() == $quote->getOrigData('customer_firstname')
-                    && $quoteAddress->getCustomerLastname() == $quote->getOrigData('customer_lastname')) {
-
-                    $newRandomData = $randomData;
-                } else {
-                    $newRandomData = $this->_getRandomData();
-                }
-
-                $this->_anonymizeQuoteAddress($quoteAddress, $newRandomData);
-            }
-        }
-    }
-
     /**
      * @param Mage_Sales_Model_Order $order
      * @param array $randomData
@@ -179,8 +167,13 @@ protected function _anonymizeOrder($order, $randomData)
         }
 
         $order->getResource()->save($order);
-
         $this->_anonymizedOrderIds[] = $order->getId();
+
+        /* @var $quote Mage_Sales_Model_Quote */
+        $quote = Mage::getModel('sales/quote')->load($order->getQuoteId());
+        if ($quote->getId()) {
+            $this->_anonymizeQuote($quote, $randomData);
+        }
     }
 
     /**
@@ -198,6 +191,37 @@ protected function _anonymizeQuotes($customer, $randomData)
         }
     }
 
+    /**
+     *
+     */
+    protected function _anonymizeRemainingQuotes()
+    {
+        $quotes = Mage::getModel('sales/quote')
+            ->getCollection()
+            ->addFieldToFilter('entity_id', array('nin' => $this->_anonymizedQuoteIds));
+
+        foreach($quotes as $quote) {
+
+            /** @var $quote Mage_Sales_Model_Quote */
+            $randomData = $this->_getRandomData();
+            $this->_anonymizeQuote($quote, $randomData);
+
+            foreach($quote->getAddressesCollection() as $quoteAddress) {
+
+                /** @var $quoteAddress Mage_Sales_Model_Quote_Address */
+                if ($quoteAddress->getCustomerFirstname() == $quote->getOrigData('customer_firstname')
+                    && $quoteAddress->getCustomerLastname() == $quote->getOrigData('customer_lastname')) {
+
+                    $newRandomData = $randomData;
+                } else {
+                    $newRandomData = $this->_getRandomData();
+                }
+
+                $this->_anonymizeQuoteAddress($quoteAddress, $newRandomData);
+            }
+        }
+    }
+
     /**
      * @param Mage_Sales_Model_Quote $quote
      * @param array $randomData
@@ -218,7 +242,6 @@ protected function _anonymizeQuote($quote, $randomData)
         }
 
         $quote->getResource()->save($quote);
-
         $this->_anonymizedQuoteIds[] = $quote->getId();
     }
 
@@ -265,6 +288,7 @@ protected function _anonymizeCustomerAddress($customerAddress, $randomData)
         }
 
         $customerAddress->getResource()->save($customerAddress);
+        $this->_anonymizedCustomerAddressIds[] = $customerAddress->getId();
 
         $this->_anonymizeQuoteAddresses($customerAddress, $randomData);
         $this->_anonymizeOrderAddresses($customerAddress, $randomData);
@@ -285,6 +309,23 @@ protected function _anonymizeQuoteAddresses($customerAddress, $randomData)
         }
     }
 
+    /**
+     *
+     */
+    protected function _anonymizeRemainingQuoteAddresses()
+    {
+        $quoteAddresses = Mage::getModel('sales/quote_address')
+            ->getCollection()
+            ->addFieldToFilter('address_id', array('nin' => $this->_anonymizedOrderAddressIds));
+
+        foreach($quoteAddresses as $quoteAddress) {
+
+            /** @var $quoteAddress Mage_Sales_Model_Quote_Address */
+            $randomData = $this->_getRandomData();
+            $this->_anonymizeQuoteAddress($quoteAddress, $randomData);
+        }
+    }
+
     /**
      * @param Mage_Sales_Model_Quote_Address $quoteAddress
      * @param array $randomData
@@ -323,6 +364,23 @@ protected function _anonymizeOrderAddresses($customerAddress, $randomData)
         }
     }
 
+    /**
+     *
+     */
+    protected function _anonymizeRemainingOrderAddresses()
+    {
+        $orderAddresses = Mage::getModel('sales/order_address')
+            ->getCollection()
+            ->addFieldToFilter('entity_id', array('nin' => $this->_anonymizedOrderAddressIds));
+
+        foreach($orderAddresses as $orderAddress) {
+
+            /** @var $orderAddress Mage_Sales_Model_Order_Address */
+            $randomData = $this->_getRandomData();
+            $this->_anonymizeOrderAddress($orderAddress, $randomData);
+        }
+    }
+
     /**
      * @param Mage_Sales_Model_Order_Address $orderAddress
      * @param array $randomData
@@ -343,6 +401,12 @@ protected function _anonymizeOrderAddress($orderAddress, $randomData)
 
         $orderAddress->getResource()->save($orderAddress);
         $this->_anonymizedOrderAddressIds[] = $orderAddress->getId();
+
+        /* @var $quoteAddress Mage_Sales_Model_Quote_Address */
+        $quoteAddress = Mage::getModel('sales/quote_address')->load($orderAddress->getQuoteAddressId());
+        if ($quoteAddress->getId()) {
+            $this->_anonymizeQuoteAddress($quoteAddress, $randomData);
+        }
     }
 
     /**

shell/anonymizer.php

@@ -41,7 +41,12 @@ class Mage_Shell_Anonymizer extends Mage_Shell_Abstract
      */
     public function run()
     {
-        Mage::getModel('anonymizer/anonymizer')->anonymizeAll();
+        /** @var $anonymizer IntegerNet_Anonymizer_Model_Anonymizer */
+        $anonymizer = Mage::getModel('anonymizer/anonymizer');
+        $anonymizer->anonymizeAll();
+        foreach($anonymizer->getResults() as $resultLabel => $resultCount) {
+            echo 'Anonymized ' . $resultCount . ' ' . $resultLabel . ".\n";
+        }
     }
 
     /**