Flask學習筆記:檔案上傳原生實現
阿新 • • 發佈:2019-01-30
import os
from flask import Flask, request, redirect, url_for
from werkzeug.utils import secure_filename
UPLOAD_FOLDER = '/path/to/the/uploads'
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
UPLOAD_FOLDER代表上傳的檔案所儲存的位置。ALLOWED_EXTENSIONS代表允許的檔案拓展名。
以下函式檢查上傳的檔案的拓展名是否有效並將使用者重定向到上傳檔案的URL:
def allowed_file(filename): return '.' in filename and \ filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS @app.route('/', methods=['GET', 'POST']) def upload_file(): if request.method == 'POST': # 檢查post請求是否含有檔案 if 'file' not in request.files: flash('No file part') return redirect(request.url) file = request.files['file'] # 如果使用者沒有選擇檔案 # 瀏覽器也會提交沒有檔名的空部分 if file.filename == '': flash('No selected file') return redirect(request.url) if file and allowed_file(file.filename): filename = secure_filename(file.filename) file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename)) return redirect(url_for('uploaded_file', filename=filename)) return ''' <!doctype html> <title>Upload new File</title> <h1>Upload new File</h1> <form method=post enctype=multipart/form-data> <p><input type=file name=file> <input type=submit value=Upload> </form> '''
之前匯入的secure_filename()就是用於檢查使用者上傳的檔案的拓展名是否合法的。