WordPress - show current user attachments

functions.php

<?php 

//Besides administrator show only the current user media files
function show_current_user_attachments( $query ) {
    if( ! current_user_can( 'manage_options' ) ) {
        $user_id = get_current_user_id();
        if ( $user_id ) {
            $query['author'] = $user_id;
        }
        return $query;
    }

    return $query;
}
add_filter( 'ajax_query_attachments_args', 'show_current_user_attachments' );