前端的Wordpress媒体上传被管理员重定向阻止

Question

我使用wp-editor让用户写文字＆amp;上传媒体。

我添加了一个脚本，如果他们没有以管理员身份登录，就会阻止用户进入后端。

这是剧本：

function yproject_admin_init() {
   if (! current_user_can( 'activate_plugins' )) {
     wp_redirect(site_url());
     exit();
   }
}
add_action( 'admin_init', 'yproject_admin_init' );

问题是：对于非管理员用户，当他们尝试上传文件时，文件似乎上传但最后得到消息：＆＃34;上传时出错。请稍后再试。＆＃34;

如果我删除＆＃34;重定向＆＃34;和＆＃34;退出＆＃34;行，它工作正常（但很明显，用户访问管理员方面）。

我可以添加哪个测试来防止这种情况？