自从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 );

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