（二）Flask 實(shí)現(xiàn)多個文件上傳

這篇文章介紹文件的管理和文件名的處理以及多文件上傳，文章內(nèi)容基于上一篇文章。

具體實(shí)現(xiàn)demo結(jié)構(gòu)，可以復(fù)制體驗(yàn)，下面有詳解

image.png
browser.html
<!DOCTYPE html>
<title>File Browser</title>
<h1>File Browser</h1>
<a href="{{ url_for('upload_file') }}">Upload</a> /
<a href="{{ url_for('manage_file') }}">Manage</a>
<hr>

<p>URL: {{ file_url }}</p>
<img src="{{ file_url }}">
index.html
<!DOCTYPE html>
<title>Upload File</title>
<h1>Upload File</h1>
<a href="{{ url_for('upload_file') }}">Upload</a> /
<a href="{{ url_for('manage_file') }}">Manage</a>
<hr>

<form method="POST" enctype="multipart/form-data">
    {{ form.hidden_tag() }}
    {{ form.photo(multiple="multiple")}}
    {% for error in form.photo.errors %}
        <span style="color: red;">{{ error }}</span>
    {% endfor %}
    {{ form.submit }}
</form>

{% if success %}
<br>
<p>Upload Success!</p>
{% endif %}
manage.html
<!DOCTYPE html>
<title>File Manager</title>
<h1>File Manager</h1>
<a href="{{ url_for('upload_file') }}">Upload</a> /
<a href="{{ url_for('manage_file') }}">Manage</a>
<hr>

{% for photo in files_list %}
    - {{ photo }}
    <a href="{{ url_for('open_file', filename=photo) }}">open</a>
    <a href="{{ url_for('delete_file', filename=photo) }}">del</a><br>
{% endfor %}
app.py
# -*- coding: utf-8 -*-
import os
import time
import hashlib

from flask import Flask, render_template, redirect, url_for, request
from flask_uploads import UploadSet, configure_uploads, IMAGES, patch_request_class
from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired, FileAllowed
from wtforms import SubmitField

app = Flask(__name__)
app.config['SECRET_KEY'] = 'I have a dream'
app.config['UPLOADED_PHOTOS_DEST'] = os.getcwd() + '/static'

photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)
patch_request_class(app)  # set maximum file size, default is 16MB


class UploadForm(FlaskForm):
    photo = FileField(validators=[FileAllowed(photos, u'Image Only!'), FileRequired(u'Choose a file!')])
    submit = SubmitField(u'Upload')


@app.route('/', methods=['GET', 'POST'])
def upload_file():
    form = UploadForm()
    if form.validate_on_submit():
        for filename in request.files.getlist('photo'):
            f = ('admin' + str(time.time())).encode('UTF-8')
            name = hashlib.md5(f).hexdigest()[:15]
            photos.save(filename, name=name + '.')
        success = True
    else:
        success = False
    return render_template('index.html', form=form, success=success)


@app.route('/manage')
def manage_file():
    files_list = os.listdir(app.config['UPLOADED_PHOTOS_DEST'])
    return render_template('manage.html', files_list=files_list)


@app.route('/open/<filename>')
def open_file(filename):
    file_url = photos.url(filename)
    return render_template('browser.html', file_url=file_url)


@app.route('/delete/<filename>')
def delete_file(filename):
    file_path = photos.path(filename)
    os.remove(file_path)
    return redirect(url_for('manage_file'))


if __name__ == '__main__':
    app.run(debug=True)
運(yùn)行結(jié)果
image.png

文件管理

列出全部文件

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

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

在模板里列出文件：

image

刪除文件

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

file_path = photos.path(filename)

2、使用os的remove方法刪除文件，同時在數(shù)據(jù)庫里刪除相應(yīng)的內(nèi)容：

os.remove(file_path)

多個set或多個文件夾

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

保存時設(shè)置文件夾和文件名

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

photos.save(form.photo.data, folder='demo_dir' name='demo.')
  • 文件名以‘.’結(jié)尾會自動加上后綴。
  • 文件夾也可以直接寫在name值里，比如：
photos.save(form.photo.data, name='demo_dir/demo.')

文件名處理

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

中文文件名

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

from pypinyin import lazy_pinyin
...

filename = '例子.jpg'
name = filename.split('.')[0]
ext = filename.split('.')[1]
new_filename = '_'.join(lazy_pinyin(name)) + '.' + ext

這里要注意的是，如果傳入lazy_pinyin()的不是中文會報(bào)錯。所以可以添加一個判斷，當(dāng)secure_filename()返回空白文件名時轉(zhuǎn)換：

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

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

統(tǒng)一處理

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

import time
import hashlib

name = hashlib.md5('admin' + str(time.time())).hexdigest()[:15]
filename = photos.save(form.photo.data, name=name+'.')

對Python3來說，md5()中的字符串需要先進(jìn)行編碼：
('admin' + str(time.time())).encode('UTF-8')

因?yàn)橛脕頇z查文件名的secure_filename()函數(shù)只返回ASCII字符，如果文件名為全中文會返回空白，這里需要修改一處源碼（flask_uploads.py#L80）：

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

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

多文件上傳

兩步：

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

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

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

from flask import request
...
for filename in request.files.getlist('photo'):
    photos.save(filename)
...
  • request.files.getlist('photo')返回一個FileStorage對象的列表（具體見：Data Structures - Werkzeug Documentation (0.11)），比如這樣： [<FileStorage: u'cat1478345000718.jpg' ('image/jpeg')>, <FileStorage: u'cat1478345401052.jpg' ('image/jpeg')>]
  • 上面的'photo'是input字段的name值，如果使用Flask-WTF，即在表單類里定義的字段名稱。
  • 調(diào)用save方法的對象photos是Flask-Uploads的set名稱（你可以自定義set的名稱），詳情參見這個系列的前三篇。
image
最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡書系信息發(fā)布平臺，僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容