View difference between Paste ID: 9GFawPAN and qmkHnqtv

 /**
  * Get products sizes
  */
-function bis_get_sizes($continue_import, $post_id, $data, $import_id) {
+function bis_get_sizes($continue_import, $post_id, $data, $import_id) {
-	// Get the product ID & size
+	$transient_name = "bis_size_{$import_id}";
-	$product_id = $data['manufacturer_artno'];
+
-	$product_size = $data['size'];
+    // Get the product ID & size
     $product_id = $data['manufacturer_artno']; // A field that will be used for SKU
-	$sizes = (array) get_transient('bis_sizes');
+    $product_size = $data['size'];
-	$sizes[$product_id][] = $product_size;
+
-	set_transient('bis_sizes', $sizes);
+    $sizes = (array) get_transient($transient_name);
     $sizes[$product_id][] = $product_size;
-	return $continue_import;
+    set_transient($transient_name, $sizes);
 
     return $continue_import;
 }
 add_filter('wp_all_import_is_post_to_update', 'bis_get_sizes', 10, 4);
- * Reset transient
+
 /**
-function bis_reset_sizes() {
+
-	set_transient('bis_sizes', '', 60*60);
+
 function bis_save_sizes($import_id) {
-add_action('pmxi_before_xml_import', 'bis_reset_sizes');
+	global $wpdb;
 
 	$transient_name = "bis_size_{$import_id}";
     $all_sizes = (array) get_transient($transient_name);
     $taxonomy = 'pa_size';
-function bis_save_sizes() {
+
-	$all_sizes = (array) get_transient('bis_sizes');
+	// Get products IDs from last import
-	$taxonomy = 'pa_size';
+	$product_ids = $wpdb->get_col("SELECT post_id FROM {$wpdb->prefix}pmxi_posts WHERE import_id = {$import_id}");
 
-	// Remove empty items
+    // Remove empty items from the array with sizes
-	$all_sizes = array_filter($all_sizes);
+    $all_sizes = array_filter($all_sizes);
 
-	if(!empty($all_sizes)) {
+    if(!empty($all_sizes)) {
-		foreach($all_sizes as $sku => $product_sizes) {
+        foreach($all_sizes as $sku => $product_sizes) {
-			$product_id = wc_get_product_id_by_sku($sku);
+			// Find the product with exact SKU in the imported products list
-			$product_terms = array();
+            $product_id = $wpdb->get_var(sprintf("SELECT p.ID FROM {$wpdb->posts} AS p LEFT JOIN {$wpdb->postmeta} AS pm ON (p.ID = pm.post_ID) WHERE p.ID IN (%s) AND pm.meta_key = '_sku' AND pm.meta_value = '%s'", implode(", ", $product_ids), $sku));
-
+
-			// Add product sizes again
+            // Add product sizes again to the product
-			foreach($product_sizes as $product_size) {
+			if(!empty($product_id)) {
-				$term = term_exists((string) $product_size, $taxonomy);
+				$product_terms = array();
 
-				if(empty($term)) {
+				foreach($product_sizes as $product_size) {
-					$term = wp_insert_term((string) $product_size, $taxonomy);
+					$term = term_exists((string) $product_size, $taxonomy);
 
 					if(empty($term)) {
-				$product_terms[] = (int) $term['term_id'];
+						$term = wp_insert_term((string) $product_size, $taxonomy);
 					}
-
+
-			wp_set_post_terms($product_id, $product_terms, $taxonomy);
+					$product_terms[] = (int) $term['term_id'];
-		}
+
-	}
+
 				wp_set_post_terms((int) $product_id, $product_terms, $taxonomy);
 			}
         }
     }
-	if(isset($_GET['test'])) {
+
-		bis_save_sizes();
+
-		die();
+
-	}
+
     if(isset($_GET['test']) && is_numeric($_GET['test'])) {
 		$import_id = (int) $_GET['test'];
 
         bis_save_sizes($import_id);
 
 		$transient_name = "bis_size_{$import_id}";
 		$all_sizes = get_transient($transient_name);
 		print_r($all_sizes);
         die();
     }
 });