[BUG]: LoadByAttribute to include variants (#183)

* [Feature]: LoadByAttribute to include variants * Apply php-cs-fixer changes Co-authored-by: mcop1 <[email protected]>
pimcore · Mar 16, 2022 · 46317ca · 46317ca
1 parent 31906cb
commit 46317ca
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/Tool/DataObjectLoader.php b/src/Tool/DataObjectLoader.php
@@ -16,6 +16,7 @@
 namespace Pimcore\Bundle\DataImporterBundle\Tool;
 
 use Pimcore\Db;
+use Pimcore\Model\DataObject;
 use Pimcore\Model\Element\ElementInterface;
 
 class DataObjectLoader
@@ -93,6 +94,7 @@ public function loadByAttribute(string $className,
                                     string $operator = '='): ?ElementInterface
     {
         $element = null;
+        $objectTypes = [DataObject::OBJECT_TYPE_VARIANT, DataObject::OBJECT_TYPE_OBJECT];
 
         if ($includeUnpublished) {
             $className::setHideUnpublished(false);
@@ -101,9 +103,9 @@ public function loadByAttribute(string $className,
         if ($this->isObjectBrickAttribute($attributeName) === false && $operator === '=') {
             $getter = 'getBy' . $attributeName;
             if (empty($attributeLanguage) === false) {
-                $element = $className::$getter($identifier, $attributeLanguage, $limit);
+                $element = $className::$getter($identifier, $attributeLanguage, $limit, 0, $objectTypes);
             } else {
-                $element = $className::$getter($identifier, $limit);
+                $element = $className::$getter($identifier, $limit, 0, $objectTypes);
             }
         } else {
             $queryFieldName = $attributeName;
@@ -116,6 +118,7 @@ public function loadByAttribute(string $className,
             if ($limit > 0) {
                 $conditions['limit'] = $limit;
             }
+            $conditions['objectTypes'] = $objectTypes;
             $list = $className::getList($conditions);
             $dataObjects = $list->load();
             if (empty($dataObjects) === false) {