-
-
Save greyli/ca74d71f13c52d089a8da8d2b758d519 to your computer and use it in GitHub Desktop.
# -*- coding: utf-8 -*- | |
import os | |
import uuid | |
from flask import Flask, render_template, redirect, url_for, request | |
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 | |
basedir = os.path.abspath(os.path.dirname(__file__)) | |
app = Flask(__name__) | |
app.config['SECRET_KEY'] = 'I have a dream' | |
app.config['UPLOADED_PHOTOS_DEST'] = os.path.join(basedir, 'uploads') # you'll need to create a folder named uploads | |
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, 'Image Only!'), FileRequired('Choose a file!')]) | |
submit = SubmitField('Upload') | |
@app.route('/', methods=['GET', 'POST']) | |
def upload_file(): | |
form = UploadForm() | |
if form.validate_on_submit(): | |
for f in request.files.getlist('photo'): | |
filename = uuid.uuid4().hex | |
photos.save(f, name=filename + '.') | |
success = True | |
else: | |
success = False | |
return render_template('index.html', form=form, success=success) | |
@app.route('/manage') | |
def manage_file(): | |
files_list = os.listdir(app.config['UPLOADED_PHOTOS_DEST']) | |
return render_template('manage.html', files_list=files_list) | |
@app.route('/open/<filename>') | |
def open_file(filename): | |
file_url = photos.url(filename) | |
return render_template('browser.html', file_url=file_url) | |
@app.route('/delete/<filename>') | |
def delete_file(filename): | |
file_path = photos.path(filename) | |
os.remove(file_path) | |
return redirect(url_for('manage_file')) | |
if __name__ == '__main__': | |
app.run(debug=True) |
<!DOCTYPE html> | |
<title>File Browser</title> | |
<h1>File Browser</h1> | |
<a href="{{ url_for('upload_file') }}">Upload</a> / | |
<a href="{{ url_for('manage_file') }}">Manage</a> | |
<hr> | |
<p>URL: {{ file_url }}</p> | |
<img src="{{ file_url }}"> |
<!DOCTYPE html> | |
<title>Upload File</title> | |
<h1>Upload File</h1> | |
<a href="{{ url_for('upload_file') }}">Upload</a> / | |
<a href="{{ url_for('manage_file') }}">Manage</a> | |
<hr> | |
<form method="POST" enctype="multipart/form-data"> | |
{{ form.hidden_tag() }} | |
{{ form.photo(multiple="multiple")}} | |
{% for error in form.photo.errors %} | |
<span style="color: red;">{{ error }}</span> | |
{% endfor %} | |
{{ form.submit }} | |
</form> | |
{% if success %} | |
<br> | |
<p>Upload Success!</p> | |
{% endif %} |
<!DOCTYPE html> | |
<title>File Manager</title> | |
<h1>File Manager</h1> | |
<a href="{{ url_for('upload_file') }}">Upload</a> / | |
<a href="{{ url_for('manage_file') }}">Manage</a> | |
<hr> | |
{% for photo in files_list %} | |
- {{ photo }} | |
<a href="{{ url_for('open_file', filename=photo) }}">open</a> | |
<a href="{{ url_for('delete_file', filename=photo) }}">del</a><br> | |
{% endfor %} |
ImportError: cannot import name 'secure_filename'
Please post the full error traceback.
name = hashlib.md5('admin' + str(time.time())).hexdigest()[:15]
TypeError: Unicode-objects must be encoded before hashing
This will run normally
@app.route('/', methods=['GET', 'POST'])
def upload_file():
form = UploadForm()
if form.validate_on_submit():
for filename in request.files.getlist('photo'):
str_name='admin' + str(int(time.time()))
name = hashlib.md5(str_name.encode("utf-8")).hexdigest()[:15]
photos.save(filename, name=name + '.')
success = True
else:
success = False
return render_template('index.html', form=form, success=success)
Reason is incompatible with newly released Werkzeug . You need to install werkzeug==0.16.1 and also connexion. Hope this helps!
Very useful, thanks!
For those who are struggling with 'ImportError: cannot import name 'secure_filename', just install Flask-Reuploaded and the error message will be gone.
https://pypi.org/project/Flask-Reuploaded/
Since the hashed filename isn't used for any purpose other than uniqueness, you could replace it with something like str(uuid.uuid4())
and achieve the same goal
@tassaron2 Updated, thanks for the tip.
Reason is incompatible with newly released Werkzeug . You need to install werkzeug==0.16.1 and also connexion. Hope this helps!
Help me a lot,thanks!
Very useful, thanks!
For those who are struggling with 'ImportError: cannot import name 'secure_filename', just install Flask-Reuploaded and the error message will be gone.
https://pypi.org/project/Flask-Reuploaded/
It will introduce other errors to this project from what I tested.
ImportError: cannot import name 'secure_filename'