wzj/Mrdoc
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

优化上传URL图片时的图片格式校验

Browse Source
This commit is contained in:
zmister 2021-10-27 21:25:56 +08:00
parent 39b2f92b10
commit 0fff3cf126
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
app_doc/util_upload_img.py
View File

@ -115,6 +115,16 @@ def ice_url_img_upload(url,user):
r = requests.get(url, headers=header, stream=True) r = requests.get(url, headers=header, stream=True)
if r.status_code == 200: if r.status_code == 200:
# 判断是否为允许上传的图片类型
remote_type = r.headers['Content-Type'].split("/")[1]
if remote_type not in settings.ALLOWED_IMG:
logger.error("上传了不允许的URL图片：{}".format(url))
resp_data = {
'error': 0,
'name': {},
'file':{}
}
return resp_data
with open(path_file, 'wb') as f: with open(path_file, 'wb') as f:
f.write(r.content) # 保存文件 f.write(r.content) # 保存文件
Image.objects.create( Image.objects.create(
@ -274,6 +284,16 @@ def url_img_upload(url,dir_name,user):
try: try:
r = requests.get(url, headers=header, stream=True) r = requests.get(url, headers=header, stream=True)
if r.status_code == 200: if r.status_code == 200:
# 判断是否为允许上传的图片类型
remote_type = r.headers['Content-Type'].split("/")[1]
if remote_type not in settings.ALLOWED_IMG:
logger.error("上传了不允许的URL图片：{}".format(url))
resp_data = {
'msg': '',
'code': 1,
'data': {}
}
return resp_data
with open(path_file, 'wb') as f: with open(path_file, 'wb') as f:
f.write(r.content) # 保存文件 f.write(r.content) # 保存文件
Image.objects.create( Image.objects.create(