WooCommerce Parent Order Prefix - wordpress

Question

I am trying to achieve an order prefix between "Parent Order" and "Sub Orders". Currently I have the order prefix's working for products in specific categories, so 'Merchandise' Categories and products within have the order number prefix 'M' and the Categories and products within 'Coffee' have the prefix 'C', this all works fine, brilliantly see the image attached.
I do have subscription products and single products within the store but I mainly just want the one 'P' to show up for the main order regardless of the category the product is within.
I need and would like help adding for the parent order a 'P' so that I can distinguish between Parent and sub orders, (please see attached image) I cannot for the life of me figure out this function currently my code is this:
add_filter( 'woocommerce_order_number', 'change_woocommerce_order_number' );
function change_woocommerce_order_number( $order_id ) {
$order = wc_get_order( $order_id );
$category = array();
if ( ! empty($order) ) {
foreach ( $order->get_items() as $item_id => $item ) {
$product_id = $item['product_id'];
$terms = get_the_terms ( $product_id, 'product_cat' );
$parent_id = $order->get_parent_id( $parent, 'product_id' );
if ( ! empty($terms) ) {
foreach ( $terms as $term ) {
$category[] = $term->term_id;
}
}
}
}
if ( $parent ) {
$order_id = 'P - '. $order_id;
} else {
if ( in_array( 23, $category ) ) {
$order_id = 'M - '. $order_id;
} else {
if ( in_array( 16, $category ) ) {
$order_id = 'C - '. $order_id;
}
}
return $order_id;
}
}
The 'M' & 'C' work perfectly... just the parent order 'P' does not so instead of a 'P' I have what seems to be a random M or C in the parent order section. All my orders split in categories and this still works as expected (please see image). I mainly just need the 'P' Main Parent order on the left from the image attached
Parent Order Prefix 'P' Location
Thanks in advance.

Answer 1

You are currently returning the order number only within the else statement.
A value should always be returned at the end of the function.
Also the $parent variable is not set and you are getting the parent id the wrong way.
Try this, hope it works.
The get_parent_id() method returns 0 if the function has no parent order.
add_filter( 'woocommerce_order_number', 'change_woocommerce_order_number' );
function change_woocommerce_order_number( $order_id ) {
$order = wc_get_order( $order_id );
$category = array();
if ( ! empty($order) ) {
foreach ( $order->get_items() as $item_id => $item ) {
$product_id = $item['product_id'];
$terms = get_the_terms ( $product_id, 'product_cat' );
if ( ! empty($terms) ) {
foreach ( $terms as $term ) {
$category[] = $term->term_id;
}
}
}
}
$parent_id = $order->get_parent_id();
if ( $parent_id > 0 ) {
if ( in_array( 23, $category ) ) {
$order_id = 'M - '. $order_id;
} else {
if ( in_array( 16, $category ) ) {
$order_id = 'C - '. $order_id;
}
}
} else {
$order_id = 'P - '. $order_id;
}
return $order_id;
}
Also you could optimize the function like this:
// change the WooCommerce order number
add_filter( 'woocommerce_order_number', 'change_woocommerce_order_number' );
function change_woocommerce_order_number( $order_id ) {
$order = wc_get_order( $order_id );
$prefix = '';
$has_cat_id_23 = false;
$has_cat_id_16 = false;
// get the parent order id
$parent_id = $order->get_parent_id();
// if it does not have a parent order it returns the order number with the prefix "P - "
if ( $parent_id == 0 ) {
$prefix = 'P - ';
return $prefix . $order_id;
// otherwise set the prefix to the child orders
} else {
foreach ( $order->get_items() as $item_id => $item ) {
// if the product belongs to the product category with ID 23
if ( has_term( 23, 'product_cat', $item['product_id'] ) ) {
$has_cat_id_23 = true;
break;
}
// if the product belongs to the product category with ID 16
if ( has_term( 16, 'product_cat', $item['product_id'] ) ) {
$has_cat_id_16 = true;
break;
}
}
// if the product belongs to the product category with ID 23 set the prefix "M - "
if ( $has_cat_id_23 ) {
$prefix = 'M - ';
}
// if the product belongs to the product category with ID 16 set the prefix "C - "
if ( $has_cat_id_16 ) {
$prefix = 'C - ';
}
}
return $prefix . $order_id;
}
The code has been tested and works (I didn't install the plugin but actually only used Wordpress and WooCommerce functions and methods). The result will look like:
Try it out and let me know if it works for you too.