customer loop

2013-07-26 10:04:35 +02:00
parent 54d79dd9ae
commit 9e663145b0
3 changed files with 181 additions and 0 deletions
--- a/core/lib/Thelia/Core/Template/Loop/Customer.php
+++ b/core/lib/Thelia/Core/Template/Loop/Customer.php
@@ -0,0 +1,140 @@
+<?php
+/*************************************************************************************/
+/*                                                                                   */
+/*      Thelia	                                                                     */
+/*                                                                                   */
+/*      Copyright (c) OpenStudio                                                     */
+/*      email : info@thelia.net                                                      */
+/*      web : http://www.thelia.net                                                  */
+/*                                                                                   */
+/*      This program is free software; you can redistribute it and/or modify         */
+/*      it under the terms of the GNU General Public License as published by         */
+/*      the Free Software Foundation; either version 3 of the License                */
+/*                                                                                   */
+/*      This program is distributed in the hope that it will be useful,              */
+/*      but WITHOUT ANY WARRANTY; without even the implied warranty of               */
+/*      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                */
+/*      GNU General Public License for more details.                                 */
+/*                                                                                   */
+/*      You should have received a copy of the GNU General Public License            */
+/*	    along with this program. If not, see <http://www.gnu.org/licenses/>.         */
+/*                                                                                   */
+/*************************************************************************************/
+
+namespace Thelia\Core\Template\Loop;
+
+use Propel\Runtime\ActiveQuery\Criteria;
+use Thelia\Core\Template\Element\BaseLoop;
+use Thelia\Core\Template\Element\LoopResult;
+use Thelia\Core\Template\Element\LoopResultRow;
+
+use Thelia\Core\Template\Loop\Argument\ArgumentCollection;
+use Thelia\Core\Template\Loop\Argument\Argument;
+use Thelia\Log\Tlog;
+
+use Thelia\Model\CustomerQuery;
+use Thelia\Model\ConfigQuery;
+use Thelia\Type\TypeCollection;
+use Thelia\Type;
+
+/**
+ *
+ * Customer loop
+ *
+ *
+ * Class Customer
+ * @package Thelia\Core\Template\Loop
+ * @author Etienne Roudeix <eroudeix@openstudio.fr>
+ */
+class Customer extends BaseLoop
+{
+    /**
+     * @return ArgumentCollection
+     */
+    protected function getArgDefinitions()
+    {
+        return new ArgumentCollection(
+            Argument::createBooleanTypeArgument('current', 1),
+            Argument::createIntListTypeArgument('id'),
+            new Argument(
+                'ref',
+                new TypeCollection(
+                    new Type\AlphaNumStringListType()
+                )
+            ),
+            Argument::createBooleanTypeArgument('reseller'),
+            Argument::createIntTypeArgument('sponsor')
+        );
+    }
+
+    /**
+     * @param $pagination
+     *
+     * @return \Thelia\Core\Template\Element\LoopResult
+     */
+    public function exec(&$pagination)
+    {
+        $search = CustomerQuery::create();
+
+		$current = $this->getCurrent();
+
+        if ($current === true) {
+            $currentCustomer = $this->request->getSession()->getCustomerUser();
+            if($currentCustomer === null) {
+                return new LoopResult();
+            } else {
+                $search->filterById($currentCustomer->getId(), Criteria::EQUAL);
+            }
+        }
+
+		$id = $this->getId();
+
+        if (null !== $id) {
+            $search->filterById($id, Criteria::IN);
+        }
+
+        $ref = $this->getRef();
+
+        if (null !== $ref) {
+            $search->filterByRef($ref, Criteria::IN);
+        }
+
+        $reseller = $this->getReseller();
+
+        if ($reseller === true) {
+            $search->filterByReseller(1, Criteria::EQUAL);
+        } else if($reseller === false) {
+            $search->filterByReseller(0, Criteria::EQUAL);
+        }
+
+        $sponsor = $this->getSponsor();
+
+        if ($sponsor !== null) {
+            $search->filterBySponsor($sponsor, Criteria::EQUAL);
+        }
+
+        $customers = $this->search($search, $pagination);
+
+        $loopResult = new LoopResult();
+
+        foreach ($customers as $customer) {
+
+            if ($this->not_empty && $customer->countAllProducts() == 0) continue;
+
+            $loopResultRow = new LoopResultRow();
+            $loopResultRow->set("ID", $customer->getId());
+            $loopResultRow->set("REF", $customer->getRef());
+            $loopResultRow->set("TITLE", $customer->getTitleid());
+            $loopResultRow->set("FIRSTNAME", $customer->getFirstname());
+            $loopResultRow->set("LASTNAME", $customer->getLastname());
+            $loopResultRow->set("EMAIL", $customer->getEmail());
+            $loopResultRow->set("RESELLER", $customer->getReseller());
+            $loopResultRow->set("SPONSOR", $customer->getSponsor());
+            $loopResultRow->set("DISCOUNT", $customer->getDiscount());
+
+            $loopResult->addRow($loopResultRow);
+        }
+
+        return $loopResult;
+    }
+}