Improve import from AW on multi-store

Author: ihorvansach

Model/Import/AbstractImport.php

@@ -53,13 +53,19 @@ abstract class AbstractImport extends \Magento\Framework\Model\AbstractModel
      */
     protected $_skippedCategories = [];
 
+    /**
+     * @var \Magento\Store\Model\StoreManagerInterface
+     */
+    protected $_storeManager;
+
     /**
      * Initialize dependencies.
      *
      * @param \Magento\Framework\Model\Context $context
      * @param \Magento\Framework\Registry $registry
      * @param \Magefan\Blog\Model\PostFactory $postFactory,
      * @param \Magefan\Blog\Model\CategoryFactory $categoryFactory,
+     * @param \Magento\Store\Model\StoreManagerInterface $storeManager,
      * @param \Magento\Framework\Model\ResourceModel\AbstractResource $resource
      * @param \Magento\Framework\Data\Collection\AbstractDb $resourceCollection
      * @param array $data
@@ -69,12 +75,14 @@ public function __construct(
         \Magento\Framework\Registry $registry,
         \Magefan\Blog\Model\PostFactory $postFactory,
         \Magefan\Blog\Model\CategoryFactory $categoryFactory,
+        \Magento\Store\Model\StoreManagerInterface $storeManager,
         \Magento\Framework\Model\ResourceModel\AbstractResource $resource = null,
         \Magento\Framework\Data\Collection\AbstractDb $resourceCollection = null,
         array $data = []
     ) {
         $this->_postFactory = $postFactory;
         $this->_categoryFactory = $categoryFactory;
+        $this->_storeManager = $storeManager;
 
         parent::__construct($context, $registry, $resource, $resourceCollection, $data);
     }

Model/Import/Aw.php

@@ -52,6 +52,8 @@ public function execute()
             throw new \Exception(__('AheadWorks Blog Extension not detected.'), 1);
         }
 
+        $storeIds = array_keys($this->_storeManager->getStores(true));
+
         $categories = [];
         $oldCategories = [];
 
@@ -76,8 +78,15 @@ public function execute()
             while ($s_data = mysqli_fetch_assoc($s_result)) {
                 $data['store_ids'][] = $s_data['store_id'];
             }
-            if (empty($data['store_ids'])) {
-                $data['store_ids'][] = 0;
+
+            foreach ($data['store_ids'] as $key => $id) {
+                if (!in_array($id, $storeIds)) {
+                    unset($data['store_ids'][$key]);
+                }
+            }
+
+            if (empty($data['store_ids']) || in_array(0, $data['store_ids'])) {
+                $data['store_ids'] = 0;
             }
 
             $data['is_active'] = 1;
@@ -112,7 +121,8 @@ public function execute()
             while ($c_data = mysqli_fetch_assoc($c_result)) {
                 $oldId = $c_data['category_id'];
                 if (isset($oldCategories[$oldId])) {
-                    $postCategories[] = $oldCategories[$oldId]->getId();
+                    $id = $oldCategories[$oldId]->getId();
+                    $postCategories[$id] = $id;
                 }
             }
 
@@ -124,8 +134,13 @@ public function execute()
                 $data['store_ids'][] = $s_data['store_id'];
             }
 
-            
-            if (empty($data['store_ids'])) {
+            foreach ($data['store_ids'] as $key => $id) {
+                if (!in_array($id, $storeIds)) {
+                    unset($data['store_ids'][$key]);
+                }
+            }
+
+            if (empty($data['store_ids']) || in_array(0, $data['store_ids'])) {
                 $data['store_ids'] = 0;
             }