flask保存 文件到本地

本篇隊長介紹一下如何 把前端上傳的文件保存 到 后端flask項目目錄

首先講一下上傳、保存文件的思路：

　　第一步：前端通過post請求方式提交上傳的文件

　　　　 <input id="file" type="file" name="fileName" style="display: inline-block"/>

　　　　<input type="submit" value="上傳文件" />

　　第二步： 后端flask通過 request.files['fileName'] 獲取文件對象

　　　　f = request.files['fileName']

　　第三步：指定要保存的文件夾的絕對路徑 (此案例把上傳文件放到task_files文件夾內)

　　　　UPLOAD_FOLDER = 'app/controls/task_mgm/task_files'
　　　　file_dir = os.path.join(os.getcwd(), UPLOAD_FOLDER)
　第四步：分別 保存文件路徑到本地 和數據庫以便查詢　　　　

　　　　file_path = os.path.join(file_dir, f.filename)
　　　　f.save(file_path)
　　　　task.fileName = f.filename

代碼Demo

@duty_mgm.route('/upload', methods=['POST'])
@login_required
def upload():
     #  保存上傳的文檔
        UPLOAD_FOLDER = 'app/controls/task_mgm/task_files'
        file_dir = os.path.join(os.getcwd(), UPLOAD_FOLDER)
        if not os.path.exists(file_dir):
            os.makedirs(file_dir)
        f =request.file.get('fileName')  # 獲取上傳的文檔對象
        if f:
            file_path = os.path.join(file_dir, f.filename)  # filename是f的固有屬性
            f.save(file_path)                 # 保存到指定目錄
            task.fileName = f.filename   # 保存到數據庫
        try:
            db.session.add(task)
            db.session.commit()
        except Exception as e:
            flash('添加失敗')
        flash('添加成功')        
    return render_template('/task_mgm/taskinfo_add.html', form=form, header=header)    # 返回到文件上傳頁面

ALLOWED_EXTENSIONS = set(['txt', 'png', 'jpg', 'xls', 'JPG', 'PNG', 'xlsx', 'gif', 'GIF', 'ppt', 'pptx', 'doc', 'docx', 'csv', 'sql', 'py'])

# 用於判斷文件后綴，可調用
def allowed_file(filename):
 return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS