Django NBA Web开发 - Step 10 头像上传

作为现在社交网站最基本的功能，头像上传必不可少。
这里就简单写一下我的实现

1：模型的建立

上传文件的字段，叫做FileField，先来看一下定义

upload_to参数

callable function

可以看到主要的参数:
一个是upload_to，一看名字就知道是用于指定文件存放于哪个位置。默认来说文件会保存在MEDIA_ROOT设定的路径下，具体路径保存在数据库内.
第二个是filename，也就是你可以通过这个参数来任意修改文件名字.

2:实际的应用

def handle_avatar(instance,filename):
    return 'user_%s'% instance.id                #根据用户的id，来生成user_id这样形式的唯一文件名

class User(AbstractUser):
    nickname = models.CharField(max_length=50,blank=True)
    avatar = models.FileField(upload_to=handle_avatar, default="default_img.png")

    def set_default_foreignkey(self):
        return self.id

    class Meta(AbstractUser.Meta):
        pass

我也同样写了一个handle_avatar的函数来用于设置保存的形式，并设置了默认值，让那些没有设置过头像的用户有一个默认头像.

3: 视图函数views的编写

def edit_profile(request):
    check_exist = os.path.exists(MEDIA_ROOT+'/'+'user_'+str(request.user.id))     #定义布尔值，关于以用户id为名字的头像文件是否已经存在
    profile = UserForm()                                                          #空表格
    instance = User.objects.get(id=request.user.id)                               #定义是对应于哪个用户模型上的文件
    if request.method == "POST":
        profile = UserForm(request.POST,request.FILES,instance=instance)          #这里的instance就是表示和哪个User实例所绑定
        if profile.is_valid():                                                    #如果表单内容合格
            if not check_exist:                                                   #且这个文件没有存在
                profile.save()                                                    #保存新头像
            else:
                os.remove(MEDIA_ROOT+'/'+'user_'+str(request.user.id))            #如果已经存在，则先进行删除，再保存
                profile.save()
            return redirect('users:user_profile',user_name=request.user.username)
        else:
            return HttpResponse('failed')
    return render(request,'edit_profile.html',locals())

为了不让客户上传的头像占用过多的服务器资源，我先做了一个判断，看是否已经存在以用户id为名字形式的文件，如果有的话，就删除，没有的话直接创建.
完成修改提交后重定向到用户资料页面.
PS:这里最终要的一个环节就是UserForm里面的instance=instance这一句，这一句的作用就是将post过来的信息和该User实例捆绑起来。这个是只有在用ModelForm的时候才有的功能。

4:操作举例

我选择了任意一个图片并上传

选择图片

头像就成功了

效果图

我们再来重新上传一张其他的

选择另一张照片

上传后，服务里还是只有这个用户名（id为6）的一个文件。

服务器头像文件夹

5：程序优化

头像上传功能是做好了，不过在实际使用过程中，发现一个问题，就是上传头像后，无法刷新出新的头像，原因是头像的url指向的是服务器上文件的名字没有改变，导致了读取的是浏览器缓存内的文件，只有shift F5强制刷新才行，这可不是我们希望看到的

那如何来解决这个问题呢？有一个办法是在页面的url后面添加查询字符串，比如
http://127.0.0.1:8000/users/profile/allentest6?time=201904122200
但是如果每次访问用户页面都是需要强制读取的话，开销又不是很合理。
所以，我在头像上传完之后的重定向时，才让他进行一次添加查询字符串的请求
如下

def edit_profile(request):
    check_exist = os.path.exists(MEDIA_ROOT+'/'+'user_'+str(request.user.id))     #定义布尔值，关于以用户id为名字的头像文件是否已经存在
    profile = UserForm()                                                          #空表格
    instance = User.objects.get(id=request.user.id)                               #定义是对应于哪个用户模型上的文件
    if request.method == "POST":
        profile = UserForm(request.POST,request.FILES,instance=instance)          #这里的instance就是表示和哪个User实例所绑定
        if profile.is_valid():                                                    #如果表单内容合格
            if not check_exist:                                                   #且这个文件没有存在
                profile.save()#保存新头像
            else:
                os.remove(MEDIA_ROOT+'/'+'user_'+str(request.user.id))            #如果已经存在，则先进行删除，再保存
                profile.save()
            #在下面的重定向时，添加当前时间的查询字符串，以达到强制刷新的作用    
            return redirect(reverse('users:user_profile',kwargs={'user_name':request.user.username})+'?time='+str(time.time()))
        else:
            return HttpResponse('failed')
    return render(request,'edit_profile.html',locals())

重定向带上查询字符串

这样，最简单的头像上传功能就做好了.

参考资料：
https://docs.djangoproject.com/en/2.1/ref/models/fields/#django.db.models.FileField.upload_to