使用flask上传文件-上传

Question

我正在尝试使用flask-uploads扩展上载文件。问题是，每次我似乎找到了一个解决方案，我就资助了另一个问题，我觉得文档很神秘。

代码如下：

from flask.ext.uploads import UploadSet, IMAGES, configure_uploads 
import os
from werkzeug import secure_filename

photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)

@app.route('/edit_book', methods = ['GET', 'POST'])
def edit_book():
form = BookForm()
if form.validate_on_submit():
            b.summary = form.summary.data
    if request.method == 'POST':
        try:
            secure_filename = photos.save(request.files['cover'])
        except:
            print 'not ok'
            db.session.add(b)
    db.session.commit()
    return redirect('/admin')
return render_template('edit_book.html', title = 'Add a book', form = form )

当前的错误只给出了：

KeyError: 'images'.

我什么都没有了！在使用这个扩展时没有任何问题，因为我在任何地方都找不到合适的代码来提供完整的解决方案(从开始上传到将文件保存在文件夹中)。

提前感谢您的帮助。

模板代码

<!-- extend from admin layout -->
{% extends "admin.html" %}

{% block content %}
 <form action="" method="post" name="edit_book" enctype="multipart/form-data">
{{form.hidden_tag()}}

{% if dico %}
 <h1>Ajouter le livre <i>{{dico.title}}</i> à la base de données</h1>
{% else %}
{% set dico = [] %}
 <h1>Ajouter un livre à la base de données</h1>
 {% endif %}
<p>Titre (obligatoire) :    {{form.title(value=dico.title)}}</p>
<p>ISBN 13 (EAN) :      {{form.ean(value=dico.EAN)}}</p>
<p>ISBN :           {{form.isbn(value=dico.ISBN)}}</p>
<p>Maison d'édition :       {{form.publisher(value=dico.publisher)}}</p>
<p>epaisseur (cm) :     {{form.thickness(value=dico.thickness)}}</p>
<p>longueur (cm) :      {{form.length(value=dico.length)}}</p>
<p>largeur (cm) :       {{form.width(value=dico.width)}}</p>
<p>masse (kg) :         {{form.mass(value=dico.mass)}}</p>
<p>Nombre de pages :        {{form.numberofpages(value=dico.pages)}}</p>
<p>Couverture du livre :    {{form.cover(value=dico.img)}}</p>
<p>Quatrième de couverture :    {{form.summary(cols="35", rows="20")|safe}}</p>
<p><input type="submit" value="Envoyez"></p>
</form>
{% endblock %}

错误堆栈跟踪

* Running on http://127.0.0.1:5000/
* Restarting with reloader
127.0.0.1 - - [22/Jul/2013 15:16:18] "POST /edit_book HTTP/1.1" 500 -
Traceback (most recent call last):
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1701, in __call__
return self.wsgi_app(environ, start_response)
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1689, in wsgi_app
response = self.make_response(self.handle_exception(e))
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1687, in wsgi_app
response = self.full_dispatch_request()
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1360, in    full_dispatch_request
rv = self.handle_user_exception(e)
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1358, in  full_dispatch_request
rv = self.dispatch_request()
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1344, in  dispatch_request
return self.view_functions[rule.endpoint](**req.view_args)
File "/home/stephane/git/biblib-flask/app/views.py", line 117, in edit_book
if form.validate_on_submit():
File "/usr/local/lib/python2.7/dist-packages/flask_wtf/form.py", line 125, in  validate_on_submit
return self.is_submitted() and self.validate()
 File "/usr/local/lib/python2.7/dist-packages/wtforms/form.py", line 265, in validate
return super(Form, self).validate(extra)
 File "/usr/local/lib/python2.7/dist-packages/wtforms/form.py", line 130, in validate
if not field.validate(self, extra):
 File "/usr/local/lib/python2.7/dist-packages/wtforms/fields/core.py", line 176, in  validate
 stop_validation = self._run_validation_chain(form, chain)
 File "/usr/local/lib/python2.7/dist-packages/wtforms/fields/core.py", line 196, in _run_validation_chain
 validator(form, self)
 File "/usr/local/lib/python2.7/dist-packages/flask_wtf/file.py", line 69, in __call__
if not self.upload_set.file_allowed(field.data, field.data.filename):
 File "/usr/local/lib/python2.7/dist-packages/Flask_Uploads-0.1.3-    py2.7.egg/flaskext/uploads.py", line 346, in file_allowed
 return self.extension_allowed(extension(basename))
 File "/usr/local/lib/python2.7/dist-packages/Flask_Uploads-0.1.3-py2.7.egg/flaskext/uploads.py", line 356, in extension_allowed
 return ((ext in self.config.allow) or
 File "/usr/local/lib/python2.7/dist-packages/Flask_Uploads-0.1.3-py2.7.egg/flaskext/uploads.py", line 308, in config
 return current_app.upload_set_config[self.name]
 KeyError: 'images'
 127.0.0.1 - - [22/Jul/2013 15:16:18] "GET /edit_book?    __debugger__=yes&cmd=resource&f=style.css HTTP/1.1" 200 -
 127.0.0.1 - - [22/Jul/2013 15:16:18] "GET /edit_book?__debugger__=yes&cmd=resource&f=jquery.js HTTP/1.1" 200 -
 127.0.0.1 - - [22/Jul/2013 15:16:18] "GET /edit_book?__debugger__=yes&cmd=resource&f=debugger.js HTTP/1.1" 200 -
 127.0.0.1 - - [22/Jul/2013 15:16:18] "GET /edit_book?__debugger__=yes&cmd=resource&f=ubuntu.ttf HTTP/1.1" 200 -
 127.0.0.1 - - [22/Jul/2013 15:16:18] "GET /edit_book?__debugger__=yes&cmd=resource&f=console.png HTTP/1.1" 200 -
 127.0.0.1 - - [22/Jul/2013 15:16:18] "GET /edit_book?__debugger__=yes&cmd=resource&f=source.png HTTP/1.1" 200 -

我只是更多地检查了代码，发现只有一件事是确切的“图像”：表单中的验证器。

 from flaskext.uploads import UploadSet, IMAGES
 images = UploadSet("images", IMAGES)

 class BookForm(Form):
    summary = TextAreaField('summary', [validators.optional()])
cover = FileField("cover", [validators.optional(),file_allowed(images, "Images only!")])

Answer 1

问题是，您有两个上传集:一个用于views.py (照片)，另一个用于forms.py(图像)。您应该从您的app.py导入上传集photos。

此外，Flask-Uploads会调用secure_filename()为您验证文件名，因此不需要此行：

from werkzeug import secure_filename

完整的演示在这里：

import os
from flask import Flask, render_template
from flask_uploads import UploadSet, configure_uploads, IMAGES, patch_request_class
from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired, FileAllowed
from wtforms import SubmitField

app = Flask(__name__)
app.config['SECRET_KEY'] = 'I have a dream'
app.config['UPLOADED_PHOTOS_DEST'] = os.getcwd()

photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)
patch_request_class(app)  # set maximum file size, default is 16MB


class UploadForm(FlaskForm):
    photo = FileField(validators=[FileAllowed(photos, u'Image only!'), FileRequired(u'File was empty!')])
    submit = SubmitField(u'Upload')


@app.route('/', methods=['GET', 'POST'])
def upload_file():
    form = UploadForm()
    if form.validate_on_submit():
        filename = photos.save(form.photo.data)
        file_url = photos.url(filename)
    else:
        file_url = None
    return render_template('index.html', form=form, file_url=file_url)


if __name__ == '__main__':
    app.run()

模板：

<h1>Photo Upload</h1>
<form method="POST" enctype="multipart/form-data">
    {{ form.hidden_tag() }}
    {{ form.photo }}
    {% for error in form.photo.errors %}
        <span style="color: red;">{{ error }}</span>
    {% endfor %}
    {{ form.submit }}
</form>

{% if file_url %}
<br>
<img src="{{ file_url }}">
{% endif %}

要点链接：https://gist.github.com/greyli/81d7e5ae6c9baf7f6cdfbf64e8a7c037

Answer 2

我也遇到了同样的问题。

问题在于，在您的配置中配置了以下UploadSet

photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)

但是表单字段cover使用以下UploadSet

images = UploadSet("images", IMAGES)

您只需导入photos UploadSet并将cover字段更改为：

cover = FileField("cover", [validators.optional(),file_allowed(photos, "Images only!")])

这对我来说很管用。

Answer 3

photos = 'png'，'jpeg'，'gif‘#要验证cover = FileField("cover", validators=[FileRequired(), FileAllowed(photos, 'Message')])的文件扩展名

这将会起到作用;)