使用python的Flask框架進行上傳和下載文件詳解

更新時間：2023年07月25日 09:36:57 作者：IT之一小佬

這篇文章主要介紹了使用python的Flask框架進行上傳和下載文件詳解,Flask是一個使用Pyhton編寫的輕量級Web應用框架，工具包采用Werkzeug，模板引擎則使用Jinja2，是目前十分流行的web框架,需要的朋友可以參考下

send_file()函數(shù)：

def send_file(
    path_or_file: t.Union[os.PathLike, str, t.BinaryIO],
    mimetype: t.Optional[str] = None,
    as_attachment: bool = False,
    download_name: t.Optional[str] = None,
    attachment_filename: t.Optional[str] = None,
    conditional: bool = True,
    etag: t.Union[bool, str] = True,
    add_etags: t.Optional[bool] = None,
    last_modified: t.Optional[t.Union[datetime, int, float]] = None,
    max_age: t.Optional[
        t.Union[int, t.Callable[[t.Optional[str]], t.Optional[int]]]
    ] = None,
    cache_timeout: t.Optional[int] = None,
):

參數(shù)解析：

path_or_file:要發(fā)送的文件的路徑，如果給出了相對路徑，則相對于當前工作目錄?；蛘?，以二進制模式打開的類文件對象。確保將文件指針查找到數(shù)據(jù)的開頭。
mimetype：為文件發(fā)送的MIME類型。如果沒有提供，它將嘗試從文件名檢測它。
as_attachment：指示瀏覽器應該提供給保存文件而不是顯示它。
download_name:瀏覽器保存文件時使用的默認名稱。默認為傳遞的文件名。
conditional:基于請求頭啟用條件響應和范圍響應。需要傳遞一個文件路徑和``environ``。
etag:計算文件的etag，這需要傳遞一個文件路徑。也可以是字符串來代替。
last_modified:發(fā)送文件的最后修改時間，單位為秒。如果沒有提供，它將嘗試從文件路徑檢測它。
max_age:客戶端應該緩存文件多長時間，以秒為單位。如果設置，' ' Cache-Control ' '將為' ' public ' '，否則將為' ' no-cache ' '以選擇條件緩存。

注意：

示例代碼：

import os
from flask import Flask, send_file, request, render_template, redirect
from werkzeug.utils import secure_filename
app = Flask(__name__)
UPLOAD_FOLDER = 'media'  # 注意：要提前在根目錄下新建media文件，否則會報錯
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
# 判斷上傳的文件是否是允許的后綴
def allowed_file(filename):
    return "." in filename and filename.rsplit('.', 1)[1].lower() in set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
@app.route('/')
def index():
    return "hello world!"
@app.route('/generate_file')
def generate_file():
    with open('./new_file.txt', 'w', encoding='utf-8') as f:
        f.write('我是new_file.txt文件！')
    return "<./new_file.txt>文件已經(jīng)生成！"
@app.route('/download')
def download():
    return send_file('./new_file.txt', as_attachment=True)
@app.route('/upload', methods=['GET', 'POST'])
def upload():
    if request.method == 'GET':
        return render_template('upload.html')
    else:
        if "file" not in request.files:
            return redirect(request.url)
        file = request.files.get('file')  # 獲取文件
        # 上傳空文件（無文件）
        if file.filename == '':
            return redirect(request.url)
        if file and allowed_file(file.filename):
            if not os.path.exists(f'./templates/{UPLOAD_FOLDER}'):
                os.mkdir(f'./templates/{UPLOAD_FOLDER}')
            filename = secure_filename(file.filename)  # 用這個函數(shù)確定文件名稱是否是安全 （注意：中文不能識別）
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))  # 保存文件
            return "上傳文件成功！"
if __name__ == '__main__':
    app.run()

下載和上傳路徑中可以添加用戶身份驗證：

import os
from flask import Flask, send_file, request, render_template, redirect
from werkzeug.utils import secure_filename
app = Flask(__name__)
UPLOAD_FOLDER = 'media'  # 注意：要提前在根目錄下新建media文件，否則會報錯
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
allow_users = ['dgw', 'super_user']
# 判斷上傳的文件是否是允許的后綴
def allowed_file(filename):
    return "." in filename and filename.rsplit('.', 1)[1].lower() in set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
@app.route('/')
def index():
    return "hello world!"
@app.route('/generate_file')
def generate_file():
    with open('./new_file.txt', 'w', encoding='utf-8') as f:
        f.write('我是new_file.txt文件！')
    return "<./new_file.txt>文件已經(jīng)生成！"
@app.route('/download')
def download():
    user_name = request.args.get('user_name')
    if not user_name:
        return "你沒有權限下載文件！"
    if user_name and user_name not in allow_users:
        return "你沒有權限下載文件！"
    return send_file('./new_file.txt', as_attachment=True)
@app.route('/upload/<string:user_name>', methods=['GET', 'POST'])
def upload(user_name):
    if request.method == 'GET':
        if user_name not in allow_users:
            return "您沒有權限訪問此頁面！"
        return render_template('upload.html')
    else:
        if user_name not in allow_users:
            return "您沒有權限訪問此頁面！"
        if "file" not in request.files:
            return redirect(request.url)
        file = request.files.get('file')  # 獲取文件
        # 上傳空文件（無文件）
        if file.filename == '':
            return redirect(request.url)
        if file and allowed_file(file.filename):
            if not os.path.exists(f'./templates/{UPLOAD_FOLDER}'):
                os.mkdir(f'./templates/{UPLOAD_FOLDER}')
            filename = secure_filename(file.filename)  # 用這個函數(shù)確定文件名稱是否是安全 （注意：中文不能識別）
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))  # 保存文件
            return "上傳文件成功！"
if __name__ == '__main__':
    app.run()

附錄：

upload.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Upload</title>
</head>
<body>
<form action="" method="post" enctype="multipart/form-data">
    <input type="file" name="file">
    <input type="submit" value="上傳">
</form>
</body>
</html>

到此這篇關于使用python的Flask框架進行上傳和下載文件詳解的文章就介紹到這了,更多相關python的Flask框架上傳下載內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: