python flask：模仿werkzeug FileStorage對象

Question

我在Flask應用程序中具有以下路線，該應用程序接受上載的文件並將文件對象扔到函數中進行驗證。 這是一個准系統示例：

def is_file_valid(file):
    if file.filename == 'test':
        return True
    return False

@app.route('/validate', method=['POST'])
def validate():
    file = request.files['file']
    if is_file_valid(file):
        return redirect(url_for('somewhere'))
    return redirect(url_for('somewhere_else'))

我正在嘗試創建一個單元測試來測試is_file_valid函數，但是在創建FileStorage對象flask的使用時遇到了麻煩，它看起來與標准python文件對象（ docs ）非常相似。

到目前為止，這是我嘗試過的方法：

import io
with io.open('/path/to/file', 'rb') as f:
    print(f.filename)

但我收到以下錯誤：

AttributeError: 'io.TextIOWrapper' object has no attribute 'filename'

知道如何在常規python腳本中模仿werkzeug的FileStorage對象嗎？

Answer 1

在這里回答。 只需將python文件對象包裝在werkzeug FileStorage類中即可。