php - 在 Woocommerce 的前端仅显示当前用户的产品

Question

我需要在前端仅显示当前用户的产品，即对于登录的用户来说，他/她应该只能在网站上查看自己的产品。我尝试了下面的代码，但这限制了对当前用户未创建的所有帖子和页面的访问。我只需要限制 woocommerce 产品。

function shapeSpace_set_only_author($query) {
    global $current_user;
    $query->set('author', $current_user->ID);
}
add_action('pre_get_posts', 'shapeSpace_set_only_author');

Answer 1

您可以使用专用 woocommerce_product_query 操作 Hook 将显示的产品限制为:

add_action( 'woocommerce_product_query', 'shapeSpace_set_only_author', 20, 2 );
function shapeSpace_set_only_author ( $q, $query ) {
    if( is_admin() ) return;

    global $current_user;
    $q->set('author', $current_user->ID);
}

代码位于事件子主题(或主题)的 function.php 文件中。

经过测试并有效