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

2019-04-12 本文已影响0人 AllenBigBear

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

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的时候才有的功能。