自从WordPress可以自定义文章类型功能一来，开发人员在进行WordPress主题及WordPress插件开发中都会用到自定义文章类型，今天就分享一下自定义WordPress首页、搜索、作者存档及feed订阅输出文章类型教程。
以下的代码都可以添加到当前主题的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 );
当然了，上面的函数还可以精简，只不过为了可读性和灵活性，就不再压缩了，有能力的朋友可以自己修改。
