Flask文件上傳（四）：文件管理與多文件上傳

01-30

這篇文章介紹文件的管理和文件名的處理以及多文件上傳，文章內容基於本系列前三篇文章。

Flask文件上傳系列目錄

Flask文件上傳（一）：原生實現
Flask文件上傳（二）：使用擴展實現
Flask文件上傳（三）：完整實現
Flask文件上傳（四）：文件管理與多文件上傳
Flask文件上傳（五）：拖拽上傳和進度條

文件管理

其實這部分並不難，出於完整性，我還是簡單的說一下。

列出全部文件

使用os模塊的listdir方法列出所有文件：

photos_list = os.listdir(app.config[UPLOADED_PHOTOS_DEST])n

在模板里列出文件：

刪除文件

1、使用Flask-Uploads的set提供的path方法獲取文件的絕對路徑（用文件名作為參數）：

file_path = photos.path(filename)n

2、使用os的remove方法刪除文件，同時在資料庫里刪除相應的內容：

os.remove(file_path)n

多個set或多個文件夾

需求較小的話，比如用作個人網盤，你可以為不同的文件創建不同的set。每個set都有不同的存儲位置。需求量大的話，可以為每一個用戶創建一個文件夾，用用戶名的md5值的前五位作為文件夾名稱。

保存時設置文件夾和文件名

使用Flask-Uploads時，可以在保存時傳入文件夾和文件名參數：

photos.save(form.photo.data, folder=demo_dir name=demo.)n

文件名以『.』結尾會自動加上後綴。n
文件夾也可以直接寫在name值里，比如：n

photos.save(form.photo.data, name=demo_dir/demo.)n

文件名處理

直接存儲文件名會造成安全問題，詳情見這裡。如果不使用Flask-Uploads的話，需要使用Werkzeug提供的secure_filename()函數來處理文件名。

中文文件名

secure_filename()函數只返回ASCII字元，非ASCII字元會被過濾掉。所以中文文件名需要轉換成英文，可以使用pypinyin來轉換（使用pip安裝），比如：

from pypinyin import lazy_pinyinn...nnfilename = 例子.jpgnname = filename.split(.)[0]next = filename.split(.)[1]nnew_filename = _.join(lazy_pinyin(name)) + . + extn

這裡要注意的是，如果傳入lazy_pinyin()的不是中文會報錯。所以可以添加一個判斷，當secure_filename()返回空白文件名時轉換：

filename = secure_filename(files.filename)nif filename.startswith(.):n name = files.filename.split(.)[0]n ext = files.filename.split(.)[1]n filename = _.join(lazy_pinyin(name)) + . + extn

但是文件名除了中文和英文，還可能是日文。所以出於完善的考慮，建議使用下面的方法處理。

統一處理

如果有大量的文件要處理，可以使用用戶名+時間戳的md5值作為文件名，使用用戶名的md5值作為文件夾（長度適當選擇）：

import timenimport hashlibnnname = hashlib.md5(admin + str(time.time())).hexdigest()[:15]nfilename = photos.save(form.photo.data, name=name+.)n

對Python3來說，md5()中的字元串需要先進行編碼：

(admin + str(time.time())).encode(UTF-8)

因為用來檢查文件名的secure_filename()函數只返回ASCII字元，如果文件名為全中文會返回空白，這裡需要修改一處源碼（flask_uploads.py#L80）：

把這裡的extension()方法修改為：

def extension(filename):n ext = os.path.splitext(filename)[1]n if ext == :n ext = os.path.splitext(filename)[0]n if ext.startswith(.):n # os.path.splitext retains . separatorn ext = ext[1:]n return extn

多文件上傳

兩步：

1、給input添加multiple屬性（獲得多選能力）：

{{ form.photo(multiple="multiple")}}n

2、在獲取文件時從request對象獲取，并迭代保存：

from flask import requestn...nfor filename in request.files.getlist(photo):n photos.save(filename)n...n

request.files.getlist(photo)返回一個FileStorage對象的列表（具體見：Data Structures - Werkzeug Documentation (0.11)），比如這樣：n[<FileStorage: ucat1478345000718.jpg (image/jpeg)>, <FileStorage: ucat1478345401052.jpg (image/jpeg)>]
上面的photo是input欄位的name值，如果使用Flask-WTF，即在表單類里定義的欄位名稱。
調用save方法的對象photos是Flask-Uploads的set名稱（你可以自定義set的名稱），詳情參見這個系列的前三篇。n

完整實現

這篇文章涉及的內容比較多，就不直接貼出來了。

具體見Gist：https://gist.github.com/greyli/ca74d71f13c52d089a8da8d2b758d519