Added remaining quote andorder anonymization

Author: avstudnitz

…IntegerNet/Anonymizer/Model/Customer.php …tegerNet/Anonymizer/Model/Anonymizer.phpapp/code/community/IntegerNet/Anonymizer/Model/Customer.php renamed to app/code/community/IntegerNet/Anonymizer/Model/Anonymizer.php app/code/community/IntegerNet/Anonymizer/Model/Customer.php renamed to app/code/community/IntegerNet/Anonymizer/Model/Anonymizer.php

@@ -6,7 +6,7 @@
  * @package     IntegerNet_Anonymizer
  * @author      Andreas von Studnitz <avs@integer-net.de>
  */
-class IntegerNet_Anonymizer_Model_Customer
+class IntegerNet_Anonymizer_Model_Anonymizer
 {
     protected $_unusedCustomerData = array();
     protected $_anonymizedOrderIds = array();
@@ -15,7 +15,6 @@ class IntegerNet_Anonymizer_Model_Customer
     protected $_anonymizedQuoteAddressIds = array();
     protected $_anonymizedNewsletterSubscriberIds = array();
 
-
     public function anonymizeAll()
     {
         /** @var $customers Mage_Customer_Model_Resource_Customer_Collection */
@@ -28,6 +27,9 @@ public function anonymizeAll()
         $this->_fetchRandomCustomerData($customerCount * 2);
 
         $this->_anonymizeCustomers($customers);
+
+        $this->_anonymizeRemainingOrders();
+        $this->_anonymizeRemainingQuotes();
     }
 
     /**
@@ -90,25 +92,95 @@ protected function _anonymizeOrders($customer, $randomData)
             ->getCollection()
             ->addFieldToFilter('customer_email', $customer->getOrigData('email'));
 
+        foreach($orders as $order) {
+            $this->_anonymizeOrder($order, $randomData);
+        }
+    }
+
+    /**
+     *
+     */
+    protected function _anonymizeRemainingOrders()
+    {
+        $orders = Mage::getModel('sales/order')
+            ->getCollection()
+            ->addFieldToFilter('entity_id', array('nin' => $this->_anonymizedOrderIds));
+
         foreach($orders as $order) {
 
             /** @var $order Mage_Sales_Model_Order */
-            foreach ($this->_getOrderMapping() as $orderKey => $randomDataKey) {
-                if (!$order->getData($orderKey)) {
-                    continue;
+            $randomData = $this->_getRandomData();
+            $this->_anonymizeOrder($order, $randomData);
+
+            foreach($order->getAddressesCollection() as $orderAddress) {
+
+                /** @var $orderAddress Mage_Sales_Model_Order_Address */
+                if ($orderAddress->getCustomerFirstname() == $order->getOrigData('customer_firstname')
+                    && $orderAddress->getCustomerLastname() == $order->getOrigData('customer_lastname')) {
+
+                    $newRandomData = $randomData;
+                } else {
+                    $newRandomData = $this->_getRandomData();
                 }
 
-                if (strlen($randomDataKey)) {
-                    $order->setData($orderKey, $randomData[$randomDataKey]);
+                $this->_anonymizeOrderAddress($orderAddress, $newRandomData);
+            }
+        }
+    }
+
+    /**
+     *
+     */
+    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 {
-                    $order->setData($orderKey, '');
+                    $newRandomData = $this->_getRandomData();
                 }
+
+                $this->_anonymizeQuoteAddress($quoteAddress, $newRandomData);
             }
+        }
+    }
 
-            $order->getResource()->save($order);
+    /**
+     * @param Mage_Sales_Model_Order $order
+     * @param array $randomData
+     */
+    protected function _anonymizeOrder($order, $randomData)
+    {
+        /** @var $order Mage_Sales_Model_Order */
+        foreach ($this->_getOrderMapping() as $orderKey => $randomDataKey) {
+            if (!$order->getData($orderKey)) {
+                continue;
+            }
 
-            $this->_anonymizedOrderIds[] = $order->getId();
+            if (strlen($randomDataKey)) {
+                $order->setData($orderKey, $randomData[$randomDataKey]);
+            } else {
+                $order->setData($orderKey, '');
+            }
         }
+
+        $order->getResource()->save($order);
+
+        $this->_anonymizedOrderIds[] = $order->getId();
     }
 
     /**
@@ -122,24 +194,32 @@ protected function _anonymizeQuotes($customer, $randomData)
             ->addFieldToFilter('customer_id', $customer->getId());
 
         foreach($quotes as $quote) {
+            $this->_anonymizeQuote($quote, $randomData);
+        }
+    }
 
-            /** @var $quote Mage_Sales_Model_Quote */
-            foreach ($this->_getQuoteMapping() as $quoteKey => $randomDataKey) {
-                if (!$quote->getData($quoteKey)) {
-                    continue;
-                }
+    /**
+     * @param Mage_Sales_Model_Quote $quote
+     * @param array $randomData
+     */
+    protected function _anonymizeQuote($quote, $randomData)
+    {
+        /** @var $quote Mage_Sales_Model_Quote */
+        foreach ($this->_getQuoteMapping() as $quoteKey => $randomDataKey) {
+            if (!$quote->getData($quoteKey)) {
+                continue;
+            }
 
-                if (strlen($randomDataKey)) {
-                    $quote->setData($quoteKey, $randomData[$randomDataKey]);
-                } else {
-                    $quote->setData($quoteKey, '');
-                }
+            if (strlen($randomDataKey)) {
+                $quote->setData($quoteKey, $randomData[$randomDataKey]);
+            } else {
+                $quote->setData($quoteKey, '');
             }
+        }
 
-            $quote->getResource()->save($quote);
+        $quote->getResource()->save($quote);
 
-            $this->_anonymizedQuoteIds[] = $quote->getId();
-        }
+        $this->_anonymizedQuoteIds[] = $quote->getId();
     }
 
     /**

shell/anonymizer.php

@@ -41,7 +41,7 @@ class Mage_Shell_Anonymizer extends Mage_Shell_Abstract
      */
     public function run()
     {
-        Mage::getModel('anonymizer/customer')->anonymizeAll();
+        Mage::getModel('anonymizer/anonymizer')->anonymizeAll();
     }
 
     /**