使用Python Flask實現簡易文件上傳功能

更新時間：2024年05月16日 09:24:03 作者：輕松運維小豬

在平時工作中,文件上傳是一項常見的需求,例如將應用異常時通過腳本生成的dump文件收集起來進行分析,但實現起來卻可能相當復雜,在本文中,我們將探討如何使用Flask實現文件上傳功能,編寫Dockerfile將應用程序通過docker部署,需要的朋友可以參考下

引言

在平時工作中，文件上傳是一項常見的需求，例如將應用異常時通過腳本生成的dump文件收集起來進行分析，但實現起來卻可能相當復雜。幸運的是，Flask框架提供了一種簡單而高效的方式來處理文件上傳，代碼不到100行。在本文中，我們將探討如何使用Flask實現文件上傳功能，編寫Dockerfile將應用程序通過docker部署。

Flask 文件上傳流程

Flask文件上傳的基本流程包括以下幾個步驟：

創(chuàng)建一個帶有 enctype=multipart/form-data 屬性的 <form> 表單，并在其中放置一個 <input type=file> 元素。
應用程序通過請求對象的 files 字典訪問文件。
使用文件的 save() 方法將文件保存到文件系統(tǒng)的某個位置。

實現文件上傳

以下是使用Flask實現文件上傳的示例代碼：

import os
from flask import Flask, flash, request, redirect, url_for, send_from_directory
from werkzeug.utils import secure_filename

UPLOAD_FOLDER = '/data/file_server/upload/'  # 上傳文件存儲的路徑
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif', 'pcap'}  # 允許上傳的文件擴展名

app = Flask(__name__)
app.secret_key = '3ccfa213427578f707a015a87c5f94959df16cfd'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

# 檢查文件擴展名是否合法
def allowed_file(filename):
    return '.' in filename and \
    filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

# 上傳文件并重定向用戶到上傳后的文件URL
@app.route('/', methods=['GET', 'POST'])
def upload_file():
    if request.method == '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>
      <input type=file name=file>
      <input type=submit value=Upload>
    </form>
    '''

# 提供已上傳文件的下載鏈接
@app.route('/uploads/<filename>')
def uploaded_file(filename):
    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)

if __name__ == '__main__':
    from waitress import serve
    serve(app, host="0.0.0.0", port=9090)

安全性考量

在處理文件上傳時，安全性是一個重要的考慮因素。例如，我們不希望用戶能夠上傳任何文件類型，因為這可能導致跨站腳本攻擊（XSS）或其他安全問題。因此代碼邏輯加上限制允許上傳的文件擴展名后綴。
另外對于上傳文件的文件名也是需要考慮的，secure_filename() 函數用于確保文件名的安全，防止路徑穿越攻擊。該函數會移除文件名中的路徑信息，只保留有效的文件名部分。不過由于secure_filename函數獲取中文文件名時，中文會被省略，所以文件名都要使用英文格式。

限制上傳文件大小

Flask 默認情況下會接受任意大小的文件上傳，但可以通過設置 MAX_CONTENT_LENGTH 配置鍵來限制最大允許的文件大?。?/p>

app.config['MAX_CONTENT_LENGTH'] = 16 * 1000 * 1000  # 限制為16MB

一般會搭配nginx反向代理來使用，可以通過Nginx的 client_max_body_size 指令控制上傳文件大小。

Docker部署應用

編寫Dockerfile

FROM python:3

WORKDIR /app/fileserver

RUN pip install flask Werkzeug waitress --index-url http://mirrors.cloud.aliyuncs.com/pypi/simple/ --trusted-host mirrors.cloud.aliyuncs.com

COPY fs.py ./

CMD [ "python", "./fs.py" ]

構建鏡像

docker build -t flask-upload-app .

啟動容器

docker run -d --name flask-upload-app -v /data/file_server/upload/:/data/file_server/upload/ -p 9090:9090 flask-upload-app

配置Nginx反向代理

server {
    listen       31503 ;
    # flask上傳    
    location / {
        proxy_set_header Host $http_host;
        proxy_set_header X-Real-IP $remote_addr;
        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
        proxy_pass http://localhost:9090/;
    }
}

測試上傳

結語

通過上述步驟，你可以編寫Flask應用輕松實現文件上傳功能，希望對大家有所幫助。

到此這篇關于使用Python Flask實現簡易文件上傳功能的文章就介紹到這了,更多相關Python Flask文件上傳內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

使用python下載大型文件顯示進度條和下載時間的操作代碼
大家都知道下載大型文件時存在一個問題，那就是內存使用量迅速上升，可能會造成電腦卡死，所以我們需要換一個方式進行下載，這篇文章主要介紹了使用python下載大型文件的方法顯示進度條和下載時間,需要的朋友可以參考下
2022-11-11
Python正則表達式re.sub()用法詳解
re.sub用于替換字符串中的匹配項,下面這篇文章主要給大家介紹了關于Python正則表達式re.sub()用法的相關資料,文中通過實例代碼以及圖文介紹的非常詳細,需要的朋友可以參考下
2022-09-09
python如何利用matplotlib繪制并列雙柱狀圖并標注數值
Python之中最好的圖表庫叫matplotlib,matplotlib,顧名思義就是提供了一整套和matlab相似的API,它的文檔相當完備,下面這篇文章主要給大家介紹了關于python如何利用matplotlib繪制并列雙柱狀圖并標注數值的相關資料,需要的朋友可以參考下
2022-04-04
詳解如何從TensorFlow的mnist數據集導出手寫體數字圖片
這篇文章主要介紹了詳解如何從TensorFlow的mnist數據集導出手寫體數字圖片，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2019-08-08
Python爬蟲爬取博客實現可視化過程解析
這篇文章主要介紹了Python爬蟲爬取博客實現可視化,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-06-06
Python操作JSON實現網絡數據交換
這篇文章主要介紹了Python操作JSON實現網絡數據交換，JSON的全稱是 JavaScript Object Notation，是一種輕量級的數據交換格式，關于JSON的更多相關內容感興趣的小伙伴可以參考一下
2022-06-06
Tornado源碼分析之HTTP服務請求解析
這篇文章主要為大家介紹了Tornado源碼分析之HTTP服務請求解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
2023-09-09
Python第三方庫安裝緩慢的解決方法
這篇文章主要給大家介紹了關于Python第三方庫安裝緩慢的解決方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2021-02-02
基于Python輕松制作一個股票K線圖網站
在當今這個人手一個?Web?服務的年代，GUI?程序還是沒有?Web?服務來的香啊。所以本文將用Python制作一個簡單的股票K線圖網站，感興趣的可以了解一下
2022-09-09
Python實現分段線性插值
這篇文章主要為大家詳細介紹了Python實現分段線性插值，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2018-12-12