flask-文件上傳/下載

廢話不說,看代碼

上傳文件

import os
from flask import Flask, render_template, send_from_directory, request, jsonify

app = Flask(__name__)

UPLOAD_FOLDER = 'upload'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER  # 設置文件上傳的目標文件夾
basedir = os.path.abspath(os.path.dirname(__file__))  # 獲取當前項目的絕對路徑
ALLOWED_EXTENSIONS = set(['txt', 'png', 'jpg', 'xls', 'JPG', 'PNG', 'xlsx', 'gif', 'GIF'])  # 允許上傳的文件後綴

# 判斷文件是否合法
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS

# 具有上傳功能的頁面
@app.route('/test/upload')
def upload_test():
    return render_template('upload.html')

@app.route('/api/upload', methods=['POST'], strict_slashes=False)
def api_upload():
    file_dir = os.path.join(basedir, app.config['UPLOAD_FOLDER'])  # 拼接成合法文件夾地址
    if not os.path.exists(file_dir):
        os.makedirs(file_dir)  # 文件夾不存在就創建
    f=request.files['myfile']  # 從表單的file字段獲取文件,myfile爲該表單的name值
    if f and allowed_file(f.filename):  # 判斷是否是允許上傳的文件類型
        fname=f.filename
        ext = fname.rsplit('.', 1)[1]  # 獲取文件後綴
        unix_time = int(time.time())
        new_filename = str(unix_time)+'.'+ext   # 修改文件名
        f.save(os.path.join(file_dir, new_filename))  #保存文件到upload目錄

        return jsonify({"errno": 0, "errmsg": "上傳成功"})
    else:
        return jsonify({"errno": 1001, "errmsg": "上傳失敗"})
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
<form id="form1" method="post" action="/api/upload" enctype="multipart/form-data">
    <div>
        <input id="File1" type="file" name="myfile"/>  <!--後臺代碼中獲取文件是通過form的name來標識的-->
        <input type="submit">提交</input>
    </div>
</form>
</body>
</html>

來看一下效果

  1. 打開上傳頁
    打開上傳頁

2.選擇文件
選擇文件

3.上傳成功
上傳成功
上傳成功

文件下載

@app.route("/download/<path:filename>")
def downloader(filename):
    dirpath = os.path.join(app.root_path, 'upload')  # 這裏是下在目錄,從工程的根目錄寫起,比如你要下載static/js裏面的js文件,這裏就要寫“static/js”
    return send_from_directory(dirpath, filename, as_attachment=True)  # as_attachment=True 一定要寫,不然會變成打開,而不是下載

看一下效果

1.地址是這樣子的
地址是這樣的
2.下載完成
下載完成

感謝閱讀

發佈了48 篇原創文章 · 獲贊 47 · 訪問量 14萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章