WordPress 媒體庫(kù)只顯示用戶自己上傳的文件

《WordPress 限制不同用戶角色可上傳的文件類型及大小》已經(jīng)詳細(xì)介紹了用戶上傳的問(wèn)題，今天分享下在 WordPress 媒體庫(kù)只顯示用戶自己上傳的文件 的方法。在當(dāng)前主題的 functions.php 文件添加下面的代碼：

/**
 * WordPress 媒體庫(kù)只顯示用戶自己上傳的文件
 * http://www.ydqwiac.cn/view-user-own-media-only.html
 */
//在文章編輯頁(yè)面的[添加媒體]只顯示用戶自己上傳的文件
function my_upload_media( $wp_query_obj ) {
	global $current_user, $pagenow;
	if( !is_a( $current_user, 'WP_User') )
		return;
	if( 'admin-ajax.php' != $pagenow || $_REQUEST['action'] != 'query-attachments' )
		return;
	if( !current_user_can( 'manage_options' ) && !current_user_can('manage_media_library') )
		$wp_query_obj->set('author', $current_user->ID );
	return;
}
add_action('pre_get_posts','my_upload_media');

//在[媒體庫(kù)]只顯示用戶上傳的文件
function my_media_library( $wp_query ) {
    if ( strpos( $_SERVER[ 'REQUEST_URI' ], '/wp-admin/upload.php' ) !== false ) {
        if ( !current_user_can( 'manage_options' ) && !current_user_can( 'manage_media_library' ) ) {
            global $current_user;
            $wp_query->set( 'author', $current_user->id );
        }
    }
}
add_filter('parse_query', 'my_media_library' );

最終效果：非管理員用戶 在文章編輯頁(yè)面的[添加媒體]窗口（下圖1） 和 多媒體 管理界面（下圖2）只顯示自己上傳的文件（還不能修正媒體數(shù)量顯示）

如果誰(shuí)知道如何修正多媒體管理界面的媒體數(shù)量顯示，希望告知。

如果大家不想折騰代碼的，也可以試試 View Own Posts Media Only 插件。