165 lines
6.0 KiB
Python
165 lines
6.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
@Author：lybbn
|
||
@Program：django-vue-lyadmin
|
||
@Remark: 自定义图片(文件)上传
|
||
"""
|
||
import os
|
||
import datetime
|
||
from django.conf import settings
|
||
from utils.common import renameuploadimg,getfulldomian
|
||
from config import DOMAIN_HOST
|
||
|
||
def ContentTypeWhiteList(file):
|
||
"""
|
||
content-type检查
|
||
参数：
|
||
file：文件对象
|
||
return：bool
|
||
"""
|
||
startswith_white = ('image/','video/','audio/','text/','application/')
|
||
equal_white = ['application/vnd.android.package-archive']
|
||
content_type = file.content_type
|
||
if content_type.startswith(startswith_white):
|
||
return True
|
||
for ew in equal_white:
|
||
if content_type == ew:
|
||
return True
|
||
return False
|
||
|
||
def ImageUpload(request,dirs):
|
||
"""
|
||
文件上传（白名单文件可上传，除apk文件外其他类型均重命名）
|
||
request:请求
|
||
dirs:要上传到那个目录
|
||
"""
|
||
image = request.FILES.getlist('file')
|
||
msg = {}
|
||
if not image:
|
||
msg['code'] = 400
|
||
msg['msg'] = "上传的图片不能为空"
|
||
return msg
|
||
|
||
notimg_file = []
|
||
img_file = []
|
||
try:
|
||
# 多图片上传，也可单图片
|
||
for img in image:
|
||
img_name = img.name
|
||
# 图片类型content-type检查
|
||
if not ContentTypeWhiteList(img):
|
||
msg['code'] = 400
|
||
msg['msg'] = "请上传正确的文件格式"
|
||
return msg
|
||
|
||
if not img_name.endswith(
|
||
('.jpg', '.jpeg', '.png', 'gif', '.bmp', '.JPG', '.JPEG', '.PNG', 'GIF', '.BMP', '.mp4','.flv','.txt','.docx','.doc','.xls','.xlsx','.apk')):
|
||
notimg_file.append(img_name)
|
||
|
||
if img.size > 1024 * 500000:
|
||
msg['code'] = 400
|
||
msg['msg'] = "图片大小不能超过500M"
|
||
return msg
|
||
|
||
else:
|
||
curr_time = datetime.datetime.now()
|
||
if not img_name.endswith('.apk'): # apk文件不重命名
|
||
image_name = renameuploadimg(img_name)
|
||
else:
|
||
image_name = img_name
|
||
time_path = curr_time.strftime("%Y-%m-%d")
|
||
img_task_dir = dirs #对应models中的上传路径
|
||
sub_path = os.path.join(settings.MEDIA_ROOT, img_task_dir, time_path)
|
||
if not os.path.exists(sub_path):
|
||
os.makedirs(sub_path)
|
||
image_path = os.path.join(sub_path, image_name)
|
||
# web_img_url = settings.MEDIA_URL + img_task_dir + "/" + time_path + "/" + image_name#相对路径/media/xxx/xxxx/xxx.png
|
||
web_img_url = DOMAIN_HOST+settings.MEDIA_URL + img_task_dir + "/" + time_path + "/" + image_name # 绝对路径http://xxx.xxx.com/media/xxx/xxxx/xxx.png
|
||
f = open(image_path, 'wb')
|
||
for i in img.chunks():
|
||
f.write(i)
|
||
f.close()
|
||
img_file.append(web_img_url)
|
||
|
||
if notimg_file:
|
||
msg['code'] = 400
|
||
msg['msg'] = '请检查是否支持的文件，失败文件部分如下：{0}'.format(','.join(notimg_file[:10]))
|
||
return msg
|
||
|
||
msg['code'] = 200
|
||
msg['img'] = img_file#['/media/xxx/xxx/xxx.png']
|
||
msg['msg'] = '上传成功'
|
||
return msg
|
||
|
||
except Exception as e:
|
||
msg['code'] = 400
|
||
msg['msg'] = '上传失败'
|
||
return msg
|
||
|
||
def ImageUpload2(request,paramsname,dirs):
|
||
"""
|
||
根据指定的名称参数名获取上传的文件
|
||
request:请求
|
||
paramsname:为formData中提交数据的名称
|
||
dirs:要上传到那个目录
|
||
"""
|
||
image = request.FILES.getlist(paramsname)
|
||
msg = {}
|
||
if not image:
|
||
msg['code'] = 400
|
||
msg['msg'] = "上传的图片不能为空"
|
||
return msg
|
||
|
||
notimg_file = []
|
||
img_file = []
|
||
try:
|
||
# 多图片上传，也可单图片
|
||
for img in image:
|
||
img_name = img.name
|
||
# 图片类型content-type检查
|
||
if not img.content_type.startswith('image/'):
|
||
msg['code'] = 400
|
||
msg['msg'] = "请上传正确的图片格式"
|
||
return msg
|
||
|
||
if not img_name.endswith(
|
||
('.jpg', '.jpeg', '.png', 'gif', '.bmp', '.JPG', '.JPEG', '.PNG', 'GIF', '.BMP')):
|
||
notimg_file.append(img_name)
|
||
|
||
if img.size > 1024 * 50000:
|
||
msg['code'] = 400
|
||
msg['msg'] = "图片大小不能超过50M"
|
||
return msg
|
||
|
||
else:
|
||
curr_time = datetime.datetime.now()
|
||
image_name = renameuploadimg(img_name)
|
||
time_path = curr_time.strftime("%Y-%m-%d")
|
||
img_task_dir = dirs #对应models中的上传路径
|
||
sub_path = os.path.join(settings.MEDIA_ROOT, img_task_dir, time_path)
|
||
if not os.path.exists(sub_path):
|
||
os.makedirs(sub_path)
|
||
image_path = os.path.join(sub_path, image_name)
|
||
# web_img_url = settings.MEDIA_URL + img_task_dir + "/" + time_path + "/" + image_name#相对路径/media/xxx/xxxx/xxx.png
|
||
web_img_url = DOMAIN_HOST+settings.MEDIA_URL + img_task_dir + "/" + time_path + "/" + image_name # 绝对路径http://xxx.xxx.com/media/xxx/xxxx/xxx.png
|
||
f = open(image_path, 'wb')
|
||
for i in img.chunks():
|
||
f.write(i)
|
||
f.close()
|
||
img_file.append(web_img_url)
|
||
|
||
if notimg_file:
|
||
msg['code'] = 400
|
||
msg['msg'] = '请检查是否支持的图片，失败文件部分如下：{0}'.format(','.join(notimg_file[:10]))
|
||
return msg
|
||
|
||
msg['code'] = 200
|
||
msg['img'] = img_file#['/media/xxx/xxx/xxx.png']
|
||
msg['msg'] = '上传成功'
|
||
return msg
|
||
|
||
except Exception as e:
|
||
msg['code'] = 400
|
||
msg['msg'] = '图片上传失败'
|
||
return msg |