基于登录表单，flask 20
支持多文件上传
upload
app.py
from flask import Flask,flash,redirect,render_template, url_for,session,send_from_directory,request import os import uuid from flask_wtf.csrf import validate_csrf from wtforms import ValidationError from form import LoginForm,UploadForm,MultiUploadForm
app = Flask(name) app.secret_key = os.getenv('SECRET_KEY', 'secret string') app.jinja_env.trim_blocks = True app.jinja_env.lstrip_blocks = True
Custom config
app.config['UPLOAD_PATH'] = os.path.join(app.root_path, 'uploads') app.config['ALLOWED_EXTENSIONS'] = ['png', 'jpg', 'jpeg', 'gif'] app.config['MAX_CONTENT_LENGTH'] = 3 * 1024 * 1024 # 3Mb def random_filename(filename): ext = os.path.splitext(filename)[1] new_filename = uuid.uuid4().hex ext return new_filename @app.route('/uploads/<path:filename>') def get_file(filename): return send_from_directory(app.config['UPLOAD_PATH'], filename) @app.route('/uploaded-images') def show_images(): return render_template('uploaded.html') @app.route('/upload', methods=['GET', 'POST']) def upload(): form = UploadForm() if form.validate_on_submit(): f = form.photo.data filename = random_filename(f.filename) f.save(os.path.join(app.config['UPLOAD_PATH'], filename)) flash('Upload success.') session['filenames'] = [filename] return redirect(url_for('show_images')) return render_template('upload.html', form=form) def allowed_file(filename): return '.' in filename and filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS'] @app.route('/multi-upload', methods=['GET', 'POST']) def multi_upload(): form = MultiUploadForm()
代码语言：javascript复制if request.method == 'POST':
filenames = []
# check csrf token
try:
validate_csrf(form.csrf_token.data)
except ValidationError:
flash('CSRF token error.')
return redirect(url_for('multi_upload'))
# check if the post request has the file part
if 'photo' not in request.files:
flash('This field is required.')
return redirect(url_for('multi_upload'))
for f in request.files.getlist('photo'):
# if user does not select file, browser also
# submit a empty part without filename
# if f.filename == '':
# flash('No selected file.')
# return redirect(url_for('multi_upload'))
# check the file extension
if f and allowed_file(f.filename):
filename = random_filename(f.filename)
f.save(os.path.join(
app.config['UPLOAD_PATH'], filename
))
filenames.append(filename)
else:
flash('Invalid file type.')
return redirect(url_for('multi_upload'))
flash('Upload success.')
session['filenames'] = filenames
return redirect(url_for('show_images'))
return render_template('upload.html', form=form)form.py
from flask_wtf.file import FileField, FileRequired, FileAllowed
class UploadForm(FlaskForm): photo = FileField('Upload Image', validators=[FileRequired(), FileAllowed(['jpg', 'jpeg', 'png', 'gif'],message="必须是图片类型")]) submit = SubmitField() class MultiUploadForm(FlaskForm): photo = MultipleFileField('Upload Image', validators=[DataRequired()]) submit = SubmitField()
templates/macros.html
{% macro form_field(field) %} {{ field.label }} {{ field(**kwargs) }} {% if field.errors -%} {% for error in field.errors -%} <small class="error">{{ error }}</small> {%- endfor %} {%- endif %} {% endmacro %}
templates/upload.html
<!DOCTYPE html> {% extends 'base.html' %} {% from 'macros.html' import form_field %}
{% block content %} <h2>File Upload Form</h2>
<form method="post" enctype="multipart/form-data"> {{ form.csrf_token }} {{ form_field(form.photo) }} {{ form.submit }} </form> {% endblock %}
templates/uploaded.html
{% extends 'base.html' %} {% from 'macros.html' import form_field %}
{% block title %}Home{% endblock %}
{% block content %} {% if session.filenames %} {% for filename in session.filenames %} <a href="{{ url_for('get_file', filename=filename) }}" target="_blank"> <img src="{{ url_for('get_file', filename=filename) }}"> </a> {% endfor %} {% endif %} {% endblock %}
