From c8e34f3403d31fdde9585210f6333b9b67cd67a5 Mon Sep 17 00:00:00 2001 From: Claudio Sanches Date: Fri, 11 Nov 2016 14:19:12 -0200 Subject: [PATCH] Fixed method returning in write context --- includes/class-wc-product-variation.php | 161 ++++++++++++++++++++++++ 1 file changed, 161 insertions(+) diff --git a/includes/class-wc-product-variation.php b/includes/class-wc-product-variation.php index 4ca4e813b35..5124a1cb28f 100644 --- a/includes/class-wc-product-variation.php +++ b/includes/class-wc-product-variation.php @@ -324,6 +324,167 @@ class WC_Product_Variation extends WC_Product_Simple { return $this->get_prop( 'attributes', $context ); } + /** + * Reads a product from the database and sets its data to the class. + * + * @since 2.7.0 + * @param int $id Variation ID. + */ + public function read( $id ) { + $this->set_defaults(); + + if ( ! $id || ! ( $post_object = get_post( $id ) ) ) { + return; + } + + $this->set_id( $id ); + $this->set_parent_id( $post_object->post_parent ); + + // The post doesn't have a parent id, therefore its invalid and we should prevent this being created. + if ( ! $this->get_parent_id() ) { + throw new Exception( sprintf( 'No parent product set for variation #%d', $this->get_id() ), 422 ); + } + + // The post parent is not a valid variable product so we should prevent this being created. + if ( 'product' !== get_post_type( $this->get_parent_id() ) ) { + throw new Exception( sprintf( 'Invalid parent for variation #%d', $this->get_id() ), 422 ); + } + + $this->set_props( array( + 'name' => get_the_title( $post_object ), + 'slug' => $post_object->post_name, + 'date_created' => $post_object->post_date, + 'date_modified' => $post_object->post_modified, + 'status' => $post_object->post_status, + 'menu_order' => $post_object->menu_order, + 'reviews_allowed' => 'open' === $post_object->comment_status, + ) ); + $this->read_product_data(); + $this->read_meta_data(); + $this->read_attributes(); + + // Set object_read true once all data is read. + $this->set_object_read( true ); + } + + /** + * Read post data. Can be overridden by child classes to load other props. + * + * @since 2.7.0 + */ + protected function read_product_data() { + $id = $this->get_id(); + $this->set_props( array( + 'description' => get_post_meta( $id, '_variation_description', true ), + 'regular_price' => get_post_meta( $id, '_regular_price', true ), + 'sale_price' => get_post_meta( $id, '_sale_price', true ), + 'date_on_sale_from' => get_post_meta( $id, '_sale_price_dates_from', true ), + 'date_on_sale_to' => get_post_meta( $id, '_sale_price_dates_to', true ), + 'tax_status' => get_post_meta( $id, '_tax_status', true ), + 'manage_stock' => get_post_meta( $id, '_manage_stock', true ), + 'stock_status' => get_post_meta( $id, '_stock_status', true ), + 'shipping_class_id' => current( $this->get_term_ids( 'product_shipping_class' ) ), + 'virtual' => get_post_meta( $id, '_virtual', true ), + 'downloadable' => get_post_meta( $id, '_downloadable', true ), + 'downloads' => array_filter( (array) get_post_meta( $id, '_downloadable_files', true ) ), + 'gallery_image_ids' => array_filter( explode( ',', get_post_meta( $id, '_product_image_gallery', true ) ) ), + 'download_limit' => get_post_meta( $id, '_download_limit', true ), + 'download_expiry' => get_post_meta( $id, '_download_expiry', true ), + 'image_id' => get_post_thumbnail_id( $id ), + 'backorders' => get_post_meta( $id, '_backorders', true ), + 'sku' => get_post_meta( $id, '_sku', true ), + 'stock_quantity' => get_post_meta( $id, '_stock', true ), + 'weight' => get_post_meta( $id, '_weight', true ), + 'length' => get_post_meta( $id, '_length', true ), + 'width' => get_post_meta( $id, '_width', true ), + 'height' => get_post_meta( $id, '_height', true ), + 'tax_class' => get_post_meta( $id, '_tax_class', true ), + ) ); + + if ( $this->is_on_sale() ) { + $this->set_price( $this->get_sale_price() ); + } else { + $this->set_price( $this->get_regular_price() ); + } + + $this->parent_data = array( + 'sku' => get_post_meta( $this->get_parent_id(), '_sku', true ), + 'manage_stock' => get_post_meta( $this->get_parent_id(), '_manage_stock', true ), + 'backorders' => get_post_meta( $this->get_parent_id(), '_backorders', true ), + 'stock_quantity' => get_post_meta( $this->get_parent_id(), '_stock', true ), + 'weight' => get_post_meta( $this->get_parent_id(), '_weight', true ), + 'length' => get_post_meta( $this->get_parent_id(), '_length', true ), + 'width' => get_post_meta( $this->get_parent_id(), '_width', true ), + 'height' => get_post_meta( $this->get_parent_id(), '_height', true ), + 'tax_class' => get_post_meta( $this->get_parent_id(), '_tax_class', true ), + 'image_id' => get_post_thumbnail_id( $this->get_parent_id() ), + ); + } + + /** + * Create a new product. + * + * @since 2.7.0 + */ + public function create() { + $this->set_date_created( current_time( 'timestamp' ) ); + + $id = wp_insert_post( apply_filters( 'woocommerce_new_product_variation_data', array( + 'post_type' => $this->post_type, + 'post_status' => $this->get_status() ? $this->get_status() : 'publish', + 'post_author' => get_current_user_id(), + 'post_title' => get_the_title( $this->get_parent_id() ) . ' –' . wc_get_formatted_variation( $this->get_attributes(), true ), + 'post_content' => '', + 'post_parent' => $this->get_parent_id(), + 'comment_status' => 'closed', + 'ping_status' => 'closed', + 'menu_order' => $this->get_menu_order(), + 'post_date' => date( 'Y-m-d H:i:s', $this->get_date_created() ), + 'post_date_gmt' => get_gmt_from_date( date( 'Y-m-d H:i:s', $this->get_date_created() ) ), + ) ), true ); + + if ( $id && ! is_wp_error( $id ) ) { + $this->set_id( $id ); + $this->update_post_meta(); + $this->update_terms(); + $this->update_attributes(); + $this->save_meta_data(); + do_action( 'woocommerce_create_' . $this->post_type, $id ); + } + } + + /** + * Updates an existing product. + * + * @since 2.7.0 + */ + public function update() { + $post_data = array( + 'ID' => $this->get_id(), + 'post_title' => get_the_title( $this->get_parent_id() ) . ' –' . wc_get_formatted_variation( $this->get_attributes(), true ), + 'post_parent' => $this->get_parent_id(), + 'comment_status' => 'closed', + 'post_status' => $this->get_status() ? $this->get_status() : 'publish', + 'menu_order' => $this->get_menu_order(), + ); + wp_update_post( $post_data ); + $this->update_post_meta(); + $this->update_terms(); + $this->update_attributes(); + $this->save_meta_data(); + do_action( 'woocommerce_update_' . $this->post_type, $this->get_id() ); + } + + /** + * Helper method that updates all the post meta for a product based on it's settings in the WC_Product class. + * + * @since 2.7.0 + */ + public function update_post_meta() { + update_post_meta( $this->get_id(), '_variation_description', $this->get_description() ); + parent::update_post_meta(); + } + /** * Save data (either create or update depending on if we are working on an existing product). *