Last active
September 25, 2022 15:28
-
-
Save greyli/a643aaac06ea8c23769c0c3d9ccaae79 to your computer and use it in GitHub Desktop.
Photo upload demo with Flask.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # -*- coding: utf-8 -*- | |
| import os | |
| from flask import Flask, request, url_for, send_from_directory | |
| from werkzeug import secure_filename | |
| ALLOWED_EXTENSIONS = set(['png', 'jpg', 'jpeg', 'gif']) | |
| app = Flask(__name__) | |
| app.config['UPLOAD_FOLDER'] = os.getcwd() | |
| app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 | |
| html = ''' | |
| <!DOCTYPE html> | |
| <title>Upload File</title> | |
| <h1>Photo Upload</h1> | |
| <form method=post enctype=multipart/form-data> | |
| <input type=file name=file> | |
| <input type=submit value=upload> | |
| </form> | |
| ''' | |
| def allowed_file(filename): | |
| return '.' in filename and \ | |
| filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS | |
| @app.route('/uploads/<filename>') | |
| def uploaded_file(filename): | |
| return send_from_directory(app.config['UPLOAD_FOLDER'], | |
| filename) | |
| @app.route('/', methods=['GET', 'POST']) | |
| def upload_file(): | |
| if request.method == 'POST': | |
| file = request.files['file'] | |
| if file and allowed_file(file.filename): | |
| filename = secure_filename(file.filename) | |
| file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename)) | |
| file_url = url_for('uploaded_file', filename=filename) | |
| return html + '<br><img src=' + file_url + '>' | |
| return html | |
| if __name__ == '__main__': | |
| app.run() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment