Skip to content

Instantly share code, notes, and snippets.

@vahiwe
Forked from greyli/app.py
Created December 5, 2019 14:48
Show Gist options
  • Save vahiwe/d2c4685494e18e47c9c7826bdc1c7a2e to your computer and use it in GitHub Desktop.
Save vahiwe/d2c4685494e18e47c9c7826bdc1c7a2e to your computer and use it in GitHub Desktop.
Photo upload and manage with Flask and Flask-Uploads (Multiple file upload support!).
# -*- coding: utf-8 -*-
import os
import time
import hashlib
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 filename in request.files.getlist('photo'):
name = hashlib.md5('admin' + str(time.time())).hexdigest()[:15]
photos.save(filename, name=name + '.')
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 %}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment