自从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 );当然了，上面的函数还可以精简，只不过为了可读性和灵活性，就不再压缩了，有能力的朋友可以自己修改。
