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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 上传模块配置样例: # 上传大小限制(包括所有内容) client_max_body_size 100m; # 上...
    SkTj阅读 13,171评论 0 3
  • 一、概要 模型是你的数据的唯一的信息源。它包含你所储存数据的必要字段和行为。通常,每个模型对应数据库中唯一的一张表...
    唯老阅读 704评论 0 1
  • 点击查看原文 Web SDK 开发手册 SDK 概述 网易云信 SDK 为 Web 应用提供一个完善的 IM 系统...
    layjoy阅读 13,944评论 0 15
  • 数据问题 Q:使用Tunnel Java SDK上传数据,上传数据可以自动分配到各个分区吗? A:目前Tunnel...
    许此一生阅读 2,516评论 0 0
  • 模块间联系越多,其耦合性越强,同时表明其独立性越差( 降低耦合性,可以提高其独立性)。软件设计中通常用耦合度和内聚...
    riverstation阅读 2,097评论 0 8