/includes/wc-stock-functions.php

  1. <?php 
  2. /** 
  3. * WooCommerce Stock Functions 
  4. * 
  5. * Functions used to manage product stock levels. 
  6. * 
  7. * @author WooThemes 
  8. * @category Core 
  9. * @package WooCommerce/Functions 
  10. */ 
  11.  
  12. if ( ! defined( 'ABSPATH' ) ) { 
  13. exit; 
  14.  
  15. /** 
  16. * Update a product's stock amount. 
  17. * 
  18. * Uses queries rather than update_post_meta so we can do this in one query (to avoid stock issues). 
  19. * 
  20. * @since 3.0.0 this supports set, increase and decrease. 
  21. * @param int|WC_Product $product 
  22. * @param int|null $stock_quantity 
  23. * @param string $operation set, increase and decrease. 
  24. */ 
  25. function wc_update_product_stock( $product, $stock_quantity = null, $operation = 'set' ) { 
  26. if ( ! $product = wc_get_product( $product ) ) { 
  27. return false; 
  28. if ( ! is_null( $stock_quantity ) && $product->managing_stock() ) { 
  29. // Some products (variations) can have their stock managed by their parent. Get the correct ID to reduce here. 
  30. $product_id_with_stock = $product->get_stock_managed_by_id(); 
  31. $data_store = WC_Data_Store::load( 'product' ); 
  32. $data_store->update_product_stock( $product_id_with_stock, $stock_quantity, $operation ); 
  33. delete_transient( 'wc_low_stock_count' ); 
  34. delete_transient( 'wc_outofstock_count' ); 
  35. delete_transient( 'wc_product_children_' . ( $product->is_type( 'variation' ) ? $product->get_parent_id() : $product->get_id() ) ); 
  36. wp_cache_delete( 'product-' . $product_id_with_stock, 'products' ); 
  37.  
  38. // Re-read product data after updating stock, then have stock status calculated and saved. 
  39. $product_with_stock = wc_get_product( $product_id_with_stock ); 
  40. $product_with_stock->set_stock_status(); 
  41. $product_with_stock->save(); 
  42.  
  43. do_action( $product_with_stock->is_type( 'variation' ) ? 'woocommerce_variation_set_stock' : 'woocommerce_product_set_stock', $product_with_stock ); 
  44.  
  45. return $product_with_stock->get_stock_quantity(); 
  46. return $product->get_stock_quantity(); 
  47.  
  48. /** 
  49. * Update a product's stock status. 
  50. * 
  51. * @param int $product_id 
  52. * @param int $status 
  53. */ 
  54. function wc_update_product_stock_status( $product_id, $status ) { 
  55. $product = wc_get_product( $product_id ); 
  56. if ( $product ) { 
  57. $product->set_stock_status( $status ); 
  58. $product->save(); 
  59.  
  60. /** 
  61. * When a payment is complete, we can reduce stock levels for items within an order. 
  62. * @since 3.0.0 
  63. * @param int $order_id 
  64. */ 
  65. function wc_maybe_reduce_stock_levels( $order_id ) { 
  66. $order = wc_get_order( $order_id ); 
  67. if ( apply_filters( 'woocommerce_payment_complete_reduce_order_stock', $order && ! $order->get_data_store()->get_stock_reduced( $order_id ), $order_id ) ) { 
  68. wc_reduce_stock_levels( $order ); 
  69. add_action( 'woocommerce_payment_complete', 'wc_maybe_reduce_stock_levels' ); 
  70.  
  71. /** 
  72. * Reduce stock levels for items within an order. 
  73. * @since 3.0.0 
  74. * @param int|WC_Order $order_id 
  75. */ 
  76. function wc_reduce_stock_levels( $order_id ) { 
  77. if ( is_a( $order_id, 'WC_Order' ) ) { 
  78. $order = $order_id; 
  79. $order_id = $order->get_id(); 
  80. } else { 
  81. $order = wc_get_order( $order_id ); 
  82. if ( 'yes' === get_option( 'woocommerce_manage_stock' ) && $order && apply_filters( 'woocommerce_can_reduce_order_stock', true, $order ) && sizeof( $order->get_items() ) > 0 ) { 
  83. foreach ( $order->get_items() as $item ) { 
  84. if ( $item->is_type( 'line_item' ) && ( $product = $item->get_product() ) && $product->managing_stock() ) { 
  85. $qty = apply_filters( 'woocommerce_order_item_quantity', $item->get_quantity(), $order, $item ); 
  86. $item_name = $product->get_formatted_name(); 
  87. $new_stock = wc_update_product_stock( $product, $qty, 'decrease' ); 
  88.  
  89. if ( ! is_wp_error( $new_stock ) ) { 
  90. /** translators: 1: item name 2: old stock quantity 3: new stock quantity */ 
  91. $order->add_order_note( sprintf( __( '%1$s stock reduced from %2$s to %3$s.', 'woocommerce' ), $item_name, $new_stock + $qty, $new_stock ) ); 
  92.  
  93. // Get the latest product data. 
  94. $product = wc_get_product( $product->get_id() ); 
  95.  
  96. if ( '' !== get_option( 'woocommerce_notify_no_stock_amount' ) && $new_stock <= get_option( 'woocommerce_notify_no_stock_amount' ) ) { 
  97. do_action( 'woocommerce_no_stock', $product ); 
  98. } elseif ( '' !== get_option( 'woocommerce_notify_low_stock_amount' ) && $new_stock <= get_option( 'woocommerce_notify_low_stock_amount' ) ) { 
  99. do_action( 'woocommerce_low_stock', $product ); 
  100.  
  101. if ( $new_stock < 0 ) { 
  102. do_action( 'woocommerce_product_on_backorder', array( 'product' => $product, 'order_id' => $order_id, 'quantity' => $qty ) ); 
  103.  
  104. // ensure stock is marked as "reduced" in case payment complete or other stock actions are called 
  105. $order->get_data_store()->set_stock_reduced( $order_id, true ); 
  106.  
  107. do_action( 'woocommerce_reduce_order_stock', $order ); 
.