python

超轻量级php框架startmvc

Python Django框架单元测试之文件上传测试示例

更新时间:2020-07-02 22:48:01 作者:startmvc
本文实例讲述了PythonDjango框架单元测试之文件上传测试。分享给大家供大家参考,具体如下

本文实例讲述了Python Django框架单元测试之文件上传测试。分享给大家供大家参考,具体如下:

Submitting files is a special case. To POST a file, you need only provide the file field name as a key, and a file handle to the file you wish to upload as a value. For example:


>>> c = Client()
>>> with open('test.jpg') as fp:
... c.post('/account/avatar_upload/',{'avatar':fp})

测试文件上传其实没有什么特殊的,只需要指定后端接受请求数据的对应键值即可

(The name avatar here is not relevant; use whatever name your file-processing code expects.)在这里avatar是关联的,对应着具体的后端处理程序代码,eg:


class Useravatar(View):
 def __init__(self):
 self.thumbnail_dir = os.path.join(STATIC_ROOT, 'avatar/thumbnails')
 self.dest_dir = os.path.join(STATIC_ROOT, 'avatar/origin_imgs')
 @method_decorator(login_required)
 def post(self, request):
 nt_id = request.session.get('user_id', 'default')
 user = User.objects.get(pk=nt_id) if User.objects.filter(pk=nt_id).exists() else None
 avatarImg = request.FILES['avatar']
 if not os.path.exists(self.dest_dir):
 os.mkdir(self.dest_dir)
 dest = os.path.join(self.dest_dir, nt_id+"_avatar.jpg")
 with open(dest, "wb+") as destination:
 for chunk in avatarImg.chunks():
 destination.write(chunk)
 if make_thumb(dest,self.thumbnail_dir):
 avartaPath = os.path.join(STATIC_URL, 'avatar/thumbnails', nt_id + "_avatar.jpg")
 else:
 avartaPath = os.path.join(STATIC_URL, 'avatar/origin_imgs', nt_id + "_avatar.jpg")
 User.objects.filter(nt_id=nt_id).update(avatar=avartaPath)
 return render(request, 'profile.html', {'user': user})

希望本文所述对大家基于Django框架的Python程序设计有所帮助。

Python Django 单元测试 文件上传