在flask unittest自動化測試中如何上傳圖片或者文件
import StringIO picture = StringIO.StringIO(open('/home/admin/Pictures/test.jpg', 'rb').read()) data = dict( {'picture': (picture, 'test.jpg')}, timestamp=self.timestamp, check = check, ) base_url = '/uploadPhoto' rv = self.app.post(base_url, data=data, headers=HEADERS, ) print rv.data
上傳文件也是一樣額,把picture改為文件內容即可
flask表單上傳圖片
上傳單張
html代碼
<input type="file" id="pic" name="pic"/>
flask代碼
from flask import request pic = request.files.get('pic') #最好用flask的secure_name檢查文件擴展名 pic.save() #二進制數據 pic.stream
同時上傳多張
html代碼 只是加了個multiple
<input type="file" id="pic" name="pic" multiple/>
flask代碼
from flask import request pics = request.files.getlist('pic') for pic in pics: #最好用flask的secure_name檢查文件擴展名 pic.save() #二進制數據 pic.stream
