28 lines
557 B
PHP
28 lines
557 B
PHP
<?php
|
|
namespace ElementorPro\Modules\Woocommerce\Traits;
|
|
|
|
if ( ! defined( 'ABSPATH' ) ) {
|
|
exit; // Exit if accessed directly
|
|
}
|
|
|
|
trait Product_Id_Trait {
|
|
|
|
public function get_product( $product_id = false ) {
|
|
if ( 'product_variation' === get_post_type() ) {
|
|
return $this->get_product_variation( $product_id );
|
|
}
|
|
|
|
$product = wc_get_product( $product_id );
|
|
|
|
if ( ! $product ) {
|
|
$product = wc_get_product();
|
|
}
|
|
|
|
return $product;
|
|
}
|
|
|
|
public function get_product_variation( $product_id = false ) {
|
|
return wc_get_product( get_the_ID() );
|
|
}
|
|
}
|