原创  python flask flask_admin 上传文件放在 项目目录之外

分类: 2020-06-20T10:58:18    191人阅读   

背景:flask 项目集成flask_admin 模块,上传文件时为了方便flask寻找静态资源默认放在了 项目目录static下。导致每次覆盖发布项目 的时候 上传的文件会丢失。所以需要把上传的文件外置到项目目录外,防止文件丢失。

修改前代码:

class AdminView(sqla.ModelView):
    ......

class FileView(AdminView):
    # Override form field to use Flask-Admin FileUploadField
    form_overrides = {
        'path': form.FileUploadField
    }

    # Pass additional parameters to 'path' to FileUploadField constructor
    form_args = {
        'path': {
            'label': 'File',
            'base_path': 'static/upload/files',
            'allow_overwrite': False
        }
    }

    def _list_thumbnail(view, context, model, name):
        if not model.path:
            return ''

        return Markup(
            '<a href="%s">点击查看<a><br/>' % url_for('static',
                                                  filename='upload/files/' + model.path)) + 'static/upload/files' + model.path

    column_formatters = {
        'path': _list_thumbnail
    }

由代码可以看出,将base_path 的地址改成绝对地址就行,

修改后代码如下:

class AdminView(sqla.ModelView):
    ......

class FileView(AdminView):
    def __init__(self, app, model, session, name=None, category=None, endpoint=None, url=None, static_folder=None,
                 menu_class_name=None, menu_icon_type=None, menu_icon_value=None):

        # Override form field to use Flask-Admin FileUploadField
        self.form_args = {
            'path': {
                'label': 'File',
                'base_path':  app.config['IMG_ROOT'],
                'allow_overwrite': True
            }
        }

        super().__init__(model, session, name, category, endpoint, url, static_folder, menu_class_name,
                         menu_icon_type, menu_icon_value)

    # Override form field to use Flask-Admin FileUploadField
    form_overrides = {
        'path': form.FileUploadField
    }

    def _list_thumbnail(view, context, model, name):
        if not model.path:
            return ''

        return Markup(
            '<a href="%s">点击查看<a><br/>' % url_for('static_img',
                                                  filename= model.path)) + 'blog2/' + model.path

    column_formatters = {
        'path': _list_thumbnail
    }

由代码可以看出修改了构造函数,将flask app传过来 然后获取地址配置(/var/upload/),这样就解决了文件外置的问题。

那么问题又来了,链接静态资源是flask只会去找static下面的资源,外部地址怎么去找呢?

答案是使用send_from_directory函数(将制定目录内的文件发送到客户端)

代码如下:

@app.route('/itclub/<path:filename>')
    def static_img(filename):
        return send_from_directory(app.config['IMG_ROOT'],filename,mimetype='image/png', as_attachment=False)

as_attachment=False文件不以下载方式呈现

至此 完美解决 flask 上传文件外置和查看问题

综上可以看出,只需要将文件流返回给客户端就行,所以其实只需要做一个类似上传下载的功能就可以实现文件外置,和系统默认寻址方式无关。

问题探讨请加页脚q群


分享到: