作为现在社交网站最基本的功能,头像上传必不可少。
这里就简单写一下我的实现
1:模型的建立
上传文件的字段,叫做FileField,先来看一下定义
可以看到主要的参数:
一个是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