Woocommerce：调用不同的自定义归档产品模板进行搜索

Question

我在functions.php文件中使用以下代码段为Woo商店页面调用自定义模板。一切正常。

add_filter( 'template_include', 'custom_shop_page_template_include', 50, 1 );
function custom_shop_page_template_include( $template ) {
    if( is_shop() ) {
        $template = get_stylesheet_directory() . '/woocommerce/archive-product-custom-shop-page.php';
    } 
    return $template;
}

但是，此模板也会被调用以搜索“未显示产品”。

如何修改上面的代码段，以使搜索结果显示在不同的模板中：

If shop = product-archive-custom-shop-page.php
If search = product-archive-custom-search-page.php