Steve-Dee-Designs
/
woocommerce
mirror of https://github.com/woocommerce/woocommerce.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added full support for variations in importer

This commit is contained in:
Claudio Sanches 2017-05-24 07:26:57 -03:00
parent 86036d7040
commit 66d7aabe6c
1 changed files with 63 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
includes/import/abstract-wc-product-importer.php
View File

@ -195,7 +195,10 @@ abstract class WC_Product_Importer implements WC_Importer_Interface {
// Type is the most important part here because we need to be using the correct class and methods. // Type is the most important part here because we need to be using the correct class and methods.
if ( isset( $data['type'] ) ) { if ( isset( $data['type'] ) ) {
if ( ! in_array( $data['type'], array_keys( wc_get_product_types() ), true ) ) { $types = array_keys( wc_get_product_types() );
$types[] = 'variation';
if ( ! in_array( $data['type'], $types, true ) ) {
return new WP_Error( 'woocommerce_product_importer_invalid_type', __( 'Invalid product type.', 'woocommerce' ), array( 'status' => 401 ) ); return new WP_Error( 'woocommerce_product_importer_invalid_type', __( 'Invalid product type.', 'woocommerce' ), array( 'status' => 401 ) );
} }
@ -547,10 +550,19 @@ abstract class WC_Product_Importer implements WC_Importer_Interface {
* @return WC_Product|WP_Error * @return WC_Product|WP_Error
*/ */
protected function save_variation_data( $variation, $data ) { protected function save_variation_data( $variation, $data ) {
$parent = false;
// Check if parent exist. // Check if parent exist.
if ( isset( $data['parent_id'] ) && ! wc_get_product( $data['parent_id'] ) ) { if ( isset( $data['parent_id'] ) ) {
$variation->set_parent_id( $data['parent_id'] ); $parent = wc_get_product( $data['parent_id'] );
} else {
if ( $parent ) {
$variation->set_parent_id( $parent->get_id() );
}
}
// Stop if parent does not exists.
if ( ! $parent ) {
return new WP_Error( 'woocommerce_product_importer_missing_variation_parent_id', __( 'Missing parent ID or parent does not exist.', 'woocommerce' ), array( 'status' => 401 ) ); return new WP_Error( 'woocommerce_product_importer_missing_variation_parent_id', __( 'Missing parent ID or parent does not exist.', 'woocommerce' ), array( 'status' => 401 ) );
} }
@ -662,34 +674,26 @@ abstract class WC_Product_Importer implements WC_Importer_Interface {
} }
// Update taxonomies. // Update taxonomies.
// @todo
if ( isset( $data['attributes'] ) ) { if ( isset( $data['attributes'] ) ) {
$attributes = array(); $attributes = array();
$parent = wc_get_product( $variation->get_parent_id() ); $parent_attributes = $this->get_variation_parent_attributes( $data['attributes'], $parent );
$parent_attributes = $parent->get_attributes();
foreach ( $data['attributes'] as $attribute ) { foreach ( $data['attributes'] as $attribute ) {
$attribute_id = 0; // Get ID if is a global attribute.
$attribute_name = ''; $attribute_id = wc_attribute_taxonomy_id_by_name( $attribute['name'] );
// Check ID for global attributes or name for product attributes. if ( $attribute_id ) {
if ( ! empty( $attribute['id'] ) ) {
$attribute_id = absint( $attribute['id'] );
$attribute_name = wc_attribute_taxonomy_name_by_id( $attribute_id ); $attribute_name = wc_attribute_taxonomy_name_by_id( $attribute_id );
} elseif ( ! empty( $attribute['name'] ) ) { } else {
$attribute_name = sanitize_title( $attribute['name']); $attribute_name = sanitize_title( $attribute['name']);
} }
if ( ! $attribute_id && ! $attribute_name ) {
continue;
}
if ( ! isset( $parent_attributes[ $attribute_name ] ) || ! $parent_attributes[ $attribute_name ]->get_variation() ) { if ( ! isset( $parent_attributes[ $attribute_name ] ) || ! $parent_attributes[ $attribute_name ]->get_variation() ) {
continue; continue;
} }
$attribute_key = sanitize_title( $parent_attributes[ $attribute_name ]->get_name() ); $attribute_key = sanitize_title( $parent_attributes[ $attribute_name ]->get_name() );
$attribute_value = isset( $attribute['option'] ) ? wc_clean( stripslashes( $attribute['option'] ) ) : ''; $attribute_value = isset( $attribute['value'] ) ? current( $attribute['value'] ) : '';
if ( $parent_attributes[ $attribute_name ]->is_taxonomy() ) { if ( $parent_attributes[ $attribute_name ]->is_taxonomy() ) {
// If dealing with a taxonomy, we need to get the slug from the name posted to the API. // If dealing with a taxonomy, we need to get the slug from the name posted to the API.
@ -718,6 +722,46 @@ abstract class WC_Product_Importer implements WC_Importer_Interface {
return $variation; return $variation;
} }
/**
* Get variation parent attributes and set "is_variation".
*
* @param array $attributes Attributes list.
* @param WC_Product $parent Parent product data.
* @return array
*/
protected function get_variation_parent_attributes( $attributes, $parent ) {
$parent_attributes = $parent->get_attributes();
$require_save = false;
foreach ( $attributes as $attribute ) {
// Get ID if is a global attribute.
$attribute_id = wc_attribute_taxonomy_id_by_name( $attribute['name'] );
if ( $attribute_id ) {
$attribute_name = wc_attribute_taxonomy_name_by_id( $attribute_id );
} else {
$attribute_name = sanitize_title( $attribute['name']);
}
// Check if attribute handle variations.
if ( isset( $parent_attributes[ $attribute_name ] ) && ! $parent_attributes[ $attribute_name ]->get_variation() ) {
// Re-create the attribute to CRUD save and genarate again.
$parent_attributes[ $attribute_name ] = clone $parent_attributes[ $attribute_name ];
$parent_attributes[ $attribute_name ]->set_variation( 1 );
$require_save = true;
}
}
// Save variation attributes.
if ( $require_save ) {
$parent->set_attributes( array_values( $parent_attributes ) );
$parent->save();
}
return $parent_attributes;
}
/** /**
* Get attachment ID. * Get attachment ID.
* *