经常有用户给我反应，要求增加自定义头像功能。WordPress的头像可以通过统一gravatar服务设置，但很多用户反应说太麻烦，想要自定义效果。没办法，为了满足你们，特意写了这篇文章。纯代码实现WordPress自定义用户头像功能，不是插件，当然wordpress管理后台也没有设置的地方，此功能适用于前端用户，非要管理员使用也是可以的，自己想想办法吧。
头像上传接口
头像是一张图像，自定义就需要保存用户的头像到服务器。这里上传图像到服务器可以使用wordpress提供的函数，也可以使用PHP提供的基础文件上传函数，这样做需要注意安全，别被传马了。
复制
<?php
if ('POST' != $_SERVER['REQUEST_METHOD']) {
header('Allow: POST');
header('HTTP/1.1 405 Method Not Allowed');
header('Content-Type: text/plain');
exit;
}
header('Content-Type:application/json; charset=utf-8');
require dirname(__FILE__) . '/../../../../wp-load.php';
if (!is_user_logged_in()) {
print_r(json_encode(array('error' => 1, 'msg' => '必须登录才能操作')));
exit;
}
global $current_user;
$path = WP_CONTENT_DIR.'/uploads/avatar/'; //在 uploads目录下创建一个 avatar目录
if(!is_dir($path)){
mkdir($path);
}
extract($_POST);
$pre = strrchr($_FILES['file']["name"],".");
if($_FILES['file']["error"]>0){
switch($_FILES['file']["error"]){
case 1:
print_r(json_encode(array('msg'=>'文件大小超过php.ini设置的大小 2M')));
exit;
break;
case 2:
print_r(json_encode(array('msg'=>'文件大小超过表单设置的大小')));
exit;
case 3:
print_r(json_encode(array('msg'=>'文件只有部分被上传')));
exit;
break;
case 4:
print_r(json_encode(array('msg'=>'没有文件被上传')));
exit;
break;
case 6:
print_r(json_encode(array('msg'=>'找不到临时文件夹')));
exit;
break;
case 7:
print_r(json_encode(array('msg'=>'文件写入失败')));
exit;
break;
}
}
//图片类型过滤
$pic_arr = array("image/jpeg","image/jpg","image/pjpeg","image/png","image/x-png");
if(!in_array($_FILES['file']["type"],$pic_arr)){
print_r(json_encode(array('msg'=>'图片类型不允许')));
exit;
}
//图片大小过滤
if($_FILES['file']["size"]>(1*1024*1024)){
print_r(json_encode(array('msg'=>'图片大小不允许')));
exit;
}
$img_name=base64_encode($current_user->user_email).$pre;
if(is_uploaded_file($_FILES['file']["tmp_name"])){
move_uploaded_file($_FILES['file']["tmp_name"],$path.$img_name);
$img_url = WP_CONTENT_URL.'/uploads/avatar/'.$img_name;
update_user_meta($current_user->ID,'tb_local_avatar',$img_url);
print_r(json_encode(array('error' => 0)));
exit;
}上面是我测试自定义头像功能使用的代码，没考虑太多安全问题，使用时需要谨慎。
前端上传
前端上传代码多种多样，可以使用HTML表单也可以使用jquery，我一般使用jquery，jquery上传文件的教程我这里就不写了。不清楚的可以问下搜索引擎，或者看下这篇使用表单上传的文章PHP文件上传实现代码。
由于我的测试服务器即将过期，就不发测试链接给大家看效果了。
评论 （1）