From 21d37e86c2e83c5bb7f09bea741bb11690b3334e Mon Sep 17 00:00:00 2001 From: Mike Jolley Date: Tue, 24 Jun 2014 13:03:25 +0100 Subject: [PATCH] Handle stock for self and children, as well as syncing status with the children --- includes/class-wc-product-variable.php | 213 ++++++++++++++++--------- 1 file changed, 138 insertions(+), 75 deletions(-) diff --git a/includes/class-wc-product-variable.php b/includes/class-wc-product-variable.php index c56fc319233..d21a867801b 100644 --- a/includes/class-wc-product-variable.php +++ b/includes/class-wc-product-variable.php @@ -22,9 +22,8 @@ class WC_Product_Variable extends WC_Product { public $total_stock; /** - * __construct function. + * Constructor * - * @access public * @param mixed $product */ public function __construct( $product ) { @@ -51,24 +50,20 @@ class WC_Product_Variable extends WC_Product { * @return int */ public function get_total_stock() { - if ( empty( $this->total_stock ) ) { - $transient_name = 'wc_product_total_stock_' . $this->id; if ( false === ( $this->total_stock = get_transient( $transient_name ) ) ) { - $this->total_stock = $this->stock; + $this->total_stock = max( 0, intval( $this->stock ) ); if ( sizeof( $this->get_children() ) > 0 ) { - foreach ($this->get_children() as $child_id) { - $stock = get_post_meta( $child_id, '_stock', true ); - - if ( $stock != '' ) { - $this->total_stock += intval( $stock ); + foreach ( $this->get_children() as $child_id ) { + if ( 'yes' === get_post_meta( $child_id, '_manage_stock', true ) ) { + $stock = get_post_meta( $child_id, '_stock', true ); + $this->total_stock += max( 0, intval( $stock ) ); } } } - set_transient( $transient_name, $this->total_stock, YEAR_IN_SECONDS ); } } @@ -82,14 +77,50 @@ class WC_Product_Variable extends WC_Product { * @param string $mode can be set, add, or subtract * @return int Stock */ - function set_stock( $amount = null, $mode = 'set' ) { - // Empty total stock so its refreshed + public function set_stock( $amount = null, $mode = 'set' ) { $this->total_stock = ''; - - // Call parent set_stock + delete_transient( 'wc_product_total_stock_' . $this->id ); return parent::set_stock( $amount, $mode ); } + /** + * Performed after a stock level change at product level + */ + protected function check_stock_status() { + $set_child_stock_status = ''; + + if ( ! $this->backorders_allowed() && $this->get_stock_quantity() <= get_option( 'woocommerce_notify_no_stock_amount' ) ) { + $set_child_stock_status = 'outofstock'; + } elseif ( $this->backorders_allowed() || $this->get_stock_quantity() > get_option( 'woocommerce_notify_no_stock_amount' ) ) { + $set_child_stock_status = 'instock'; + } + + if ( $set_child_stock_status ) { + foreach ( $this->get_children() as $child_id ) { + if ( 'yes' !== get_post_meta( $child_id, '_manage_stock', true ) ) { + wc_update_product_stock_status( $child_id, $set_child_stock_status ); + } + } + + // Children statuses changed, so sync self + self::sync_stock_status( $this->id ); + } + } + + /** + * set_stock_status function. + * + * @access public + * @return void + */ + public function set_stock_status( $status ) { + $status = ( 'outofstock' === $status ) ? 'outofstock' : 'instock'; + + if ( update_post_meta( $this->id, '_stock_status', $status ) ) { + do_action( 'woocommerce_product_set_stock_status', $this->id, $status ); + } + } + /** * Return the products children posts. * @@ -97,11 +128,9 @@ class WC_Product_Variable extends WC_Product { * @return array of children ids */ public function get_children( $visible_only = false ) { - $variable = $visible_only ? 'visible_children' : 'children'; - - if ( ! is_array( $this->$variable ) ) { - $this->$variable = array(); - $transient_name = 'wc_product_' . $variable . '_ids_' . $this->id; + if ( ! is_array( $this->children ) ) { + $this->children = array(); + $transient_name = 'wc_product_children_ids_' . $this->id; if ( false === ( $this->children = get_transient( $transient_name ) ) ) { $args = array( @@ -114,23 +143,28 @@ class WC_Product_Variable extends WC_Product { 'numberposts' => -1 ); - if ( $visible_only && 'yes' === get_option( 'woocommerce_hide_out_of_stock_items' ) ) { - $args['meta_query'] = array( - array( - 'key' => '_stock_status', - 'value' => 'instock', - 'compare' => '=' - ) - ); - } + $this->children = get_posts( $args ); - $this->$variable = get_posts( $args ); - - set_transient( $transient_name, $this->$variable, YEAR_IN_SECONDS ); + set_transient( $transient_name, $this->children, YEAR_IN_SECONDS ); } } - return $this->$variable; + if ( $visible_only ) { + $children = array(); + foreach( $this->children as $child_id ) { + if ( 'yes' === get_post_meta( $child_id, '_manage_stock', true ) ) { + if ( 'instock' === get_post_meta( $child_id, '_stock_status', true ) ) { + $children[] = $child_id; + } + } elseif ( $this->is_in_stock() ) { + $children[] = $child_id; + } + } + } else { + $children = $this->children; + } + + return $children; } @@ -396,49 +430,51 @@ class WC_Product_Variable extends WC_Product { $variation = $this->get_child( $child_id ); - if ( ! empty( $variation->variation_id ) ) { - $variation_attributes = $variation->get_variation_attributes(); - $availability = $variation->get_availability(); - $availability_html = empty( $availability['availability'] ) ? '' : apply_filters( 'woocommerce_stock_html', '

'. wp_kses_post( $availability['availability'] ).'

', wp_kses_post( $availability['availability'] ) ); - - if ( has_post_thumbnail( $variation->get_variation_id() ) ) { - $attachment_id = get_post_thumbnail_id( $variation->get_variation_id() ); - - $attachment = wp_get_attachment_image_src( $attachment_id, apply_filters( 'single_product_large_thumbnail_size', 'shop_single' ) ); - $image = $attachment ? current( $attachment ) : ''; - - $attachment = wp_get_attachment_image_src( $attachment_id, 'full' ); - $image_link = $attachment ? current( $attachment ) : ''; - - $image_title = get_the_title( $attachment_id ); - $image_alt = get_post_meta( $attachment_id, '_wp_attachment_image_alt', true ); - } else { - $image = $image_link = $image_title = $image_alt = ''; - } - - $available_variations[] = apply_filters( 'woocommerce_available_variation', array( - 'variation_id' => $child_id, - 'variation_is_visible' => $variation->variation_is_visible(), - 'is_purchasable' => $variation->is_purchasable(), - 'attributes' => $variation_attributes, - 'image_src' => $image, - 'image_link' => $image_link, - 'image_title' => $image_title, - 'image_alt' => $image_alt, - 'price_html' => $variation->get_price() === "" || $this->get_variation_price( 'min' ) !== $this->get_variation_price( 'max' ) ? '' . $variation->get_price_html() . '' : '', - 'availability_html' => $availability_html, - 'sku' => $variation->get_sku(), - 'weight' => $variation->get_weight() . ' ' . esc_attr( get_option('woocommerce_weight_unit' ) ), - 'dimensions' => $variation->get_dimensions(), - 'min_qty' => 1, - 'max_qty' => $variation->backorders_allowed() ? '' : $variation->stock, - 'backorders_allowed' => $variation->backorders_allowed(), - 'is_in_stock' => $variation->is_in_stock(), - 'is_downloadable' => $variation->is_downloadable() , - 'is_virtual' => $variation->is_virtual(), - 'is_sold_individually' => $variation->is_sold_individually() ? 'yes' : 'no', - ), $this, $variation ); + if ( empty( $variation->variation_id ) || ( 'yes' === get_option( 'woocommerce_hide_out_of_stock_items' ) && ! $variation->is_in_stock() ) ) { + continue; } + + $variation_attributes = $variation->get_variation_attributes(); + $availability = $variation->get_availability(); + $availability_html = empty( $availability['availability'] ) ? '' : apply_filters( 'woocommerce_stock_html', '

'. wp_kses_post( $availability['availability'] ).'

', wp_kses_post( $availability['availability'] ) ); + + if ( has_post_thumbnail( $variation->get_variation_id() ) ) { + $attachment_id = get_post_thumbnail_id( $variation->get_variation_id() ); + + $attachment = wp_get_attachment_image_src( $attachment_id, apply_filters( 'single_product_large_thumbnail_size', 'shop_single' ) ); + $image = $attachment ? current( $attachment ) : ''; + + $attachment = wp_get_attachment_image_src( $attachment_id, 'full' ); + $image_link = $attachment ? current( $attachment ) : ''; + + $image_title = get_the_title( $attachment_id ); + $image_alt = get_post_meta( $attachment_id, '_wp_attachment_image_alt', true ); + } else { + $image = $image_link = $image_title = $image_alt = ''; + } + + $available_variations[] = apply_filters( 'woocommerce_available_variation', array( + 'variation_id' => $child_id, + 'variation_is_visible' => $variation->variation_is_visible(), + 'is_purchasable' => $variation->is_purchasable(), + 'attributes' => $variation_attributes, + 'image_src' => $image, + 'image_link' => $image_link, + 'image_title' => $image_title, + 'image_alt' => $image_alt, + 'price_html' => $variation->get_price() === "" || $this->get_variation_price( 'min' ) !== $this->get_variation_price( 'max' ) ? '' . $variation->get_price_html() . '' : '', + 'availability_html' => $availability_html, + 'sku' => $variation->get_sku(), + 'weight' => $variation->get_weight() . ' ' . esc_attr( get_option('woocommerce_weight_unit' ) ), + 'dimensions' => $variation->get_dimensions(), + 'min_qty' => 1, + 'max_qty' => $variation->backorders_allowed() ? '' : $variation->stock, + 'backorders_allowed' => $variation->backorders_allowed(), + 'is_in_stock' => $variation->is_in_stock(), + 'is_downloadable' => $variation->is_downloadable() , + 'is_virtual' => $variation->is_virtual(), + 'is_sold_individually' => $variation->is_sold_individually() ? 'yes' : 'no', + ), $this, $variation ); } return $available_variations; @@ -469,6 +505,33 @@ class WC_Product_Variable extends WC_Product { } } + /** + * Sync variable product stock status with children + * @param int $product_id + */ + public static function sync_stock_status( $product_id ) { + $children = get_posts( array( + 'post_parent' => $product_id, + 'posts_per_page'=> -1, + 'post_type' => 'product_variation', + 'fields' => 'ids', + 'post_status' => 'publish' + ) ); + + $stock_status = 'outofstock'; + + foreach ( $children as $child_id ) { + $child_stock_status = get_post_meta( $child_id, '_stock_status', true ); + $child_stock_status = $child_stock_status ? $child_stock_status : 'instock'; + if ( 'instock' === $child_stock_status ) { + $stock_status = 'instock'; + break; + } + } + + wc_update_product_stock_status( $product_id, $stock_status ); + } + /** * Sync the variable product with it's children */