菜单 学习猿地 - LMONKEY

VIP

开通学习猿地VIP

尊享10项VIP特权 持续新增

知识通关挑战

打卡带练!告别无效练习

接私单赚外块

VIP优先接,累计金额超百万

学习猿地私房课免费学

大厂实战课仅对VIP开放

你的一对一导师

每月可免费咨询大牛30次

领取更多软件工程师实用特权

入驻
1831
0

Flask——文件上传

原创
05/13 14:22
阅读数 2840

导入所需模块

from app.admin import admin
import os
from flask import Flask, request, redirect, url_for
from werkzeug import secure_filename
from app import app

定义修改文件名函数

def change_filename(filename):
    filename = os.path.splitext(filename)
    new_file = time.strftime('%Y%m%d%H%M%S', time.localtime())+str(uuid.uuid4().hex)+filename[-1]
    return new_file

在配置项中添加配置

#允许上传文件后缀名
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
#文件保存目录
UP_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'app/static/admin/uploads/')

过滤不允许上传文件

def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS

上传方法

@admin.route('/upload', methods=['POST'])
def upload_file():
    file = request.files.get('file')
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
    if not os.path.exists(app.config['UP_DIR']):
        os.makedirs(app.config['UP_DIR'],755)
        filename = change_filename(filename)
        file.save(os.path.join(app.config['UP_DIR'], filename))

发表评论

0/200
1831 点赞
0 评论
收藏
为你推荐 换一批