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