图片文件上传

from flask import Flask,render_template
from flask_bootstrap import Bootstrap
from flask_script import Manager
from flask_wtf import FlaskForm
from flask_wtf.file import FileField,FileRequired,FileAllowed #导入自定义文件上传的表单 所需要的字段和验证器
from wtforms import SubmitField
from flask_uploads import UploadSet,IMAGES,patch_request_class,configure_uploads #文件上传扩展库 所需要的
import os,random,string
from PIL import Image

app = Flask(name)
app.config[‘SECRET_KEY’] = ‘flask-wtf’ #密钥 值随便
app.config[‘BOOTSTRAP_SERVE_LOCAL’] = True #加载本地bootstrap样式
app.config[‘MAX_CONTENT_LENGTH’] = 1024*1024*64 #设置上传文件大小
app.config[‘UPLOADED_PHOTOS_DEST’] = os.path.join(os.getcwd(),’static/upload’)

设置上传文件的保存目录

file = UploadSet(‘photos’,IMAGES) #上传文件类型允许的设置
configure_uploads(app,file) #将配置和flask绑定
patch_request_class(app,size=None) #设置上传大小 获取MAX——CONTENT——LENGTH的大小
bootstrap = Bootstrap(app)
manager = Manager(app)

配置上传表单

class FileUpload(FlaskForm):
photos = FileField(‘上传头像’,validators=[FileRequired(message=’文件不能为空’),FileAllowed(file,message=’只能上传图片’)])
submit = SubmitField(‘上传’)

随机名字

def random_name(suffix,length=32):
Str = string.ascii_letters+string.digits
return ”.join(random.choice(Str) for i in range(length))+’.’+suffix

图片缩放函数

def img_zoom(path,width=200,height=200,prefix=’s_’):
img = Image.open(path)
print(img.size) # 获取图片的宽 高
img.thumbnail((width, height)) # 重新设计尺寸
pathSplit = os.path.split(path) # 将路径拆分成 路径和文件名
newPath = os.path.join(pathSplit[0],prefix+pathSplit[1]) #将前缀进行拼接
img.save(newPath) # 保存图片 覆盖掉原来的图片

@app.route(‘/form/’,methods=[‘GET’,’POST’])
def upload():
img_url = None
form = FileUpload()
if form.validate_on_submit():
photo = form.photos.data #request.files.get(‘photos’)
suffix = photo.filename.split(‘.’)[-1] #取出后缀
while True: #确保生成的名字唯一性
filename = random_name(suffix)
path = os.path.join(app.config[‘UPLOADED_PHOTOS_DEST’],filename)
if not os.path.exists(path):
break
#保存图片 并使用生成的随机名字
file.save(photo,name=filename)
#进行图片的缩放
img_zoom(path)
#进行图片的缩放
img_zoom(path,300,300,’m_’)
#获取上传图片的url地址
img_url = file.url(filename)
return render_template(‘form.html’,form=form,img_url=img_url)

if name == ‘main‘:
manager.run()