woocommerce/includes/export/class-wc-product-csv-export...

189 lines
6.6 KiB
PHP

<?php
/**
* Handles product CSV export.
*
* @author Automattic
* @category Admin
* @package WooCommerce/Export
* @version 3.1.0
*/
if ( ! defined( 'ABSPATH' ) ) {
exit;
}
/**
* Include dependencies.
*/
if ( ! class_exists( 'WC_CSV_Exporter', false ) ) {
include_once( WC_ABSPATH . 'includes/export/abstract-wc-csv-batch-exporter.php' );
}
/**
* WC_Product_CSV_Exporter Class.
*/
class WC_Product_CSV_Exporter extends WC_CSV_Batch_Exporter {
/**
* The filename to export to.
*/
protected $filename = 'wc-product-export.csv';
/**
* Type of export used in filter names.
*/
protected $export_type = 'product';
/**
* Return an array of columns to export.
*
* @return array
*/
public function get_default_column_names() {
return array(
'id' => __( 'ID', 'woocommerce' ),
'type' => __( 'Type', 'woocommerce' ),
'sku' => __( 'SKU', 'woocommerce' ),
'name' => __( 'Name', 'woocommerce' ),
'published' => __( 'Published', 'woocommerce' ),
'featured' => __( 'Is featured?', 'woocommerce' ),
'catalog_visibility' => __( 'Visibility in catalog', 'woocommerce' ),
'short_description' => __( 'Short Description', 'woocommerce' ),
'description' => __( 'Description', 'woocommerce' ),
'date_on_sale_from' => __( 'Date sale price starts', 'woocommerce' ),
'date_on_sale_to' => __( 'Date sale price ends', 'woocommerce' ),
'tax_status' => __( 'Tax Class', 'woocommerce' ),
'stock_status' => __( 'In stock?', 'woocommerce' ),
'backorders' => __( 'Backorders allowed?', 'woocommerce' ),
'sold_individually' => __( 'Sold individually?', 'woocommerce' ),
'weight' => sprintf( __( 'Weight (%s)', 'woocommerce' ), get_option( 'woocommerce_weight_unit' ) ),
'length' => sprintf( __( 'Length (%s)', 'woocommerce' ), get_option( 'woocommerce_dimension_unit' ) ),
'width' => sprintf( __( 'Width (%s)', 'woocommerce' ), get_option( 'woocommerce_dimension_unit' ) ),
'height' => sprintf( __( 'Height (%s)', 'woocommerce' ), get_option( 'woocommerce_dimension_unit' ) ),
'reviews_allowed' => __( 'Allow customer reviews?', 'woocommerce' ),
'purchase_note' => __( 'Purchase Note', 'woocommerce' ),
'price' => __( 'Price', 'woocommerce' ),
'regular_price' => __( 'Regular Price', 'woocommerce' ),
'stock_quantity' => __( 'Stock', 'woocommerce' ),
'category_ids' => __( 'Categories', 'woocommerce' ),
'tag_ids' => __( 'Tags', 'woocommerce' ),
'shipping_class_id' => __( 'Shipping Class', 'woocommerce' ),
'image_id' => __( 'Images', 'woocommerce' ),
'download_limit' => __( 'Download Limit', 'woocommerce' ),
'download_expiry' => __( 'Download Expiry Days', 'woocommerce' ),
'parent_id' => __( 'Parent', 'woocommerce' ),
'upsell_ids' => __( 'Upsells', 'woocommerce' ),
'cross_sell_ids' => __( 'Cross-sells', 'woocommerce' ),
);
}
/**
* Prepare data for export.
*/
public function prepare_data_to_export() {
$columns = $this->get_column_names();
$products = wc_get_products( array(
'status' => array( 'private', 'publish' ),
'limit' => $this->get_limit(),
'page' => $this->get_page(),
'orderby' => array(
'ID' => 'DESC',
),
'return' => 'objects',
'paginate' => true,
) );
$this->total_rows = $products->total;
$this->row_data = array();
foreach ( $products->products as $product ) {
$row = array();
foreach ( $columns as $column_id => $column_name ) {
// Format column id.
$column_id = strstr( $column_id, ':' ) ? current( explode( ':', $column_id ) ) : $column_id;
// Skip some columns if dynamically handled later.
if ( in_array( $column_id, array( 'download_name', 'download_url' ) ) ) {
continue;
}
// Skip columns if we're being selective.
if ( ! $this->is_column_exporting( $column_id ) ) {
continue;
}
// Handle special columns which don't map 1:1 to product data.
if ( 'published' === $column_id ) {
$value = 'publish' === $product->get_status( 'edit' ) ? 1 : 0;
} elseif ( 'category_ids' === $column_id ) {
$term_ids = $product->get_category_ids( 'edit' );
$value = $this->format_term_ids( $term_ids, 'product_cat' );
} elseif ( 'tag_ids' === $column_id ) {
$term_ids = $product->get_tag_ids( 'edit' );
$value = $this->format_term_ids( $term_ids, 'product_tag' );
} elseif ( 'shipping_class_id' === $column_id ) {
$term_ids = $product->get_shipping_class_id( 'edit' );
$value = $this->format_term_ids( $term_ids, 'product_shipping_class' );
} elseif ( in_array( $column_id, array( 'cross_sell_ids', 'upsell_ids', 'parent_id' ) ) ) {
$products = array_filter( array_map( 'wc_get_product', (array) $product->{"get_{$column_id}"}( 'edit' ) ) );
$product_list = array();
foreach ( $products as $product ) {
if ( $product->get_sku() ) {
$product_list[] = $product->get_sku();
} else {
$product_list[] = 'id:' . $product->get_id();
}
}
$value = implode( ',', $product_list );
} elseif ( in_array( $column_id, array( 'download_limit', 'download_expiry' ) ) ) {
if ( $product->is_type( 'downloadable' ) && is_callable( array( $product, "get_{$column_id}" ) ) ) {
$value = $product->{"get_{$column_id}"}( 'edit' );
} else {
$value = __( 'N/A', 'woocommerce' );
}
// Default and custom handling.
} else {
if ( is_callable( array( $product, "get_{$column_id}" ) ) ) {
$value = $product->{"get_{$column_id}"}( 'edit' );
} elseif ( has_filter( "woocommerce_export_{$this->export_type}_column_{$column_id}" ) ) {
$value = apply_filters( "woocommerce_export_{$this->export_type}_column_{$column_id}", '', $product );
}
}
$row[ $column_id ] = $value;
}
// Downloads are dynamic.
if ( $product->is_downloadable() && $this->is_column_exporting( 'downloads' ) ) {
$downloads = $product->get_downloads( 'edit' );
if ( $downloads ) {
$i = 1;
foreach ( $downloads as $download ) {
$this->column_names[ 'download_name:' . $i ] = sprintf( __( 'Download %d Name', 'woocommerce' ), $i );
$this->column_names[ 'download_url:' . $i ] = sprintf( __( 'Download %d URL', 'woocommerce' ), $i );
$row[ 'download_name:' . $i ] = $download->get_name();
$row[ 'download_url:' . $i ] = $download->get_file();
}
++$i;
}
}
// @todo price
// Attributes are dynamic. @todo
if ( $product->is_downloadable() ) {
}
$this->row_data[] = $row;
}
}
}