14.06.16

Administrator
1 parent 9495311a
Showing 2 changed files with 22 additions and 3 deletions Show diff stats
common/modules/product/models/Import.php
common/modules/product/models/ProductVariant.php
@@ -131,16 +131,22 @@ class Import extends Model {
                     $stock->save();
                 }
  
-                $productStock = ProductStock::find()->where(['product_variant_id' => $this->product_variant_id, 'stock_id' => $stock->stock_id])->one();
+                $productStock = ProductStock::find()->where(['product_variant_id' => $productVariant->product_variant_id, 'stock_id' => $stock->stock_id])->one();
+                if(!$productStock instanceof ProductStock) {
+                    $productStock = new ProductStock;
+                    $productStock->product_variant_id = $productVariant->product_variant_id;
+                    $productStock->stock_id = $stock->stock_id;
+                    $productStock->product_id = $productVariant->product_id;
+                }
                 $productStock->quantity = $count;
                 $productStock->save();
  
-                $productStocks = ProductStock::find()->where(['product_variant_id' => $this->product_variant_id])->andWhere(['<>', 'stock_id', $stock->stock_id])->all();
+                $productStocks = ProductStock::find()->where(['product_variant_id' => $productVariant->product_variant_id])->andWhere(['<>', 'stock_id', $stock->stock_id])->all();
  
                 $quantity = array_sum(ArrayHelper::getColumn($productStocks, 'quantity')) + $count;
             } else {
  
-                $productStocks = ProductStock::find()->where(['product_variant_id' => $this->product_variant_id])->all();
+                $productStocks = ProductStock::find()->where(['product_variant_id' => $productVariant->product_variant_id])->all();
  
                 if($productStocks instanceof  ProductStock){
                     $quantity = array_sum(ArrayHelper::getColumn($productStocks, 'quantity')) + $count;
@@ -169,6 +169,19 @@ class ProductVariant extends \yii\db\ActiveRecord
         $this->stocks = (array) $stocks;
     }
  
+    public function afterSave($insert, $changedAttributes)
+    {
+        if (!is_null($this->stocks)) {
+            //ProductStock::deleteAll(['product_variant_id' => $this->product_variant_id]);
+            $values = [];
+            foreach ($this->stocks as $id => $quantity) {
+                $productStock = ProductStock::find()->where(['product_variant_id' => $this->product_variant_id, 'stock_id' => $id])->one();
+                $productStock->quantity = $quantity;
+                $productStock->save();
+            }
+        }
+        parent::afterSave($insert, $changedAttributes);
+    }
  
     public function beforeDelete() {
         ProductImage::deleteAll(['product_variant_id' => $this->product_variant_id]);