WordPress自定义首页、搜索、作者存档、Feed订阅输出的文章类型

自从WordPress添加了自定义文章类型功能以来，现在很多网站开发或WordPress主题、插件都会用到自定义文章类型，今天倡萌来说下如何在首页、搜索结果页、作者存档以及Feed订阅输出中限制文章类型。这里所指的都是WordPress的主循环（即这些类型页面默认的循环）。

注：以下的代码都可以添加到当前主题的functions.php中，或者也可以添加到你自己的插件中。

限定首页主循环的文章类型

如果你的首页是常规的博客列表，那这个方法是有效的。

/**
* 自定义首页主循环的文章类型
*/
function wpkj_homepage_pre_get_posts( $query ) {

    // 限定首页的主循环
    if ( $query->is_home() && $query->is_main_query() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

}
add_action( 'pre_get_posts', 'wpkj_homepage_pre_get_posts', 99, 1 );

限定搜索结果页的文章类型

/**
* 自定义搜索结果页面的文章类型
*/
function wpkj_search_archive_pre_get_posts( $query ) {

    //限定搜索主循环，并且是前台页面
    if ( $query->is_search && $query->is_main_query() && !is_admin() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

}
add_action( 'pre_get_posts', 'wpkj_search_archive_pre_get_posts', 99, 1 );

限定作者存档页的文章类型

/**
* 自定义作者存档页的文章类型
*/
function wpkj_author_archive_pre_get_posts( $query ) {

    //限定是作者存档的主循环
    if ( $query->is_author() && $query->is_main_query() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

}
add_action( 'pre_get_posts', 'wpkj_author_archive_pre_get_posts', 99, 1 );

限定Feed订阅输出的文章类型

/**
* 自定义Feed订阅的文章类型
*/
function wpkj_feed_pre_get_posts( $query ) {

    //限定是Feed输出
    if ( $query->is_feed() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

}
add_action( 'pre_get_posts', 'wpkj_feed_pre_get_posts', 99, 1 );

将上面的所有功能集成到一个函数中

大家仔细看的上面的函数代码，其实都是同一个模板，我们可以将他们集中在一个函数中：

/**
* 自定义首页、搜索、作者存档、Feed订阅输出的文章类型
*/
function wpkj_custom_pre_get_posts( $query ) {

    // 自定义首页主循环的文章类型
    if ( $query->is_home() && $query->is_main_query() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

    // 自定义搜索结果页面的文章类型
    if ( $query->is_search && $query->is_main_query() && !is_admin() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

    // 自定义作者存档页的文章类型
    if ( $query->is_author() && $query->is_main_query() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

    // 自定义Feed订阅的文章类型
    if ( $query->is_feed() && $query->is_main_query() ) {
        // 请根据你的实际需要，修改下面这个array数组的文章类型
        $post_type = array( 'post', 'product', 'book' ); 
        $query->set( 'post_type', $post_type );

    }

}
add_action( 'pre_get_posts', 'wpkj_custom_pre_get_posts', 99, 1 );

当然了，上面的函数还可以精简，只不过为了可读性和灵活性，就不再压缩了，有能力的朋友可以自己修改。

pre_get_posts 钩子讲解

在上面的代码中，我们都用到了 pre_get_posts 钩子，它可以更改 WP_Query 查询的各种参数，不仅限于文章类型 'post_type'，还可以是分类、每页数量等等。