flask - 请求后初始化 flask 管理员的文件管理员

Question

考虑以下我正在尝试实现的用例。我正在使用 Flask、Flask-SQLAlchemy 将用户数据存储在 SQLite 数据库中，并使用 Flask-Admin 来实现其文件管理功能。

流程是这样的:当用户登录时，他会被带到管理模块的索引 View 。但是当他登录时，他的用户名被添加到初始化 FileAdmin 类时使用的路径中——这个想法是拥有用户目录，这些目录是在用户注册时由一个简单的 shell 命令创建的。因此，如果 Flask 中定义的根路径是“/files/”，而我的用户名是“rudimk”，那么在初始化文件 admin 时应该使用的路径应该是“/files/rudimk”。

问题是，当应用程序启动时，没有 session ，没有用户登录 - 但文件 admin 已初始化，默认路径，在我们的例子中是“/files/”。当用户登录时，无法重新初始化文件管理类，因为它在请求上下文之外。我查看了 flask-admin 文档中的 get_base_path() ，但我无法弄清楚它在这里有什么用处。

非常感激。谢谢!

编辑 - 添加了一些我正在使用的代码。请注意，在这里，我已经删除了身份验证，并通过将用户名添加到 session 来模拟经过身份验证的 session 。

import os
import os.path as op

from flask import Flask, session

from flask.ext import admin
from flask.ext.admin.contrib import fileadmin


# Create flask app
app = Flask(__name__, template_folder='templates', static_folder='files')

# Create dummy secrey key so we can use flash
app.config['SECRET_KEY'] = '123456790'


# Flask views
@app.route('/')
def index():
    # Simulating a logged-in user by storing his/her username in the session.
    session["username"]
    return '<a href="/admin/">Click me to get to Admin!</a>'


if __name__ == '__main__':
    # Create directory
    username = session["username"]
    path = op.join(op.dirname(__file__), 'files/%s' %username)
    try:
        os.mkdir(path)
    except OSError:
        pass

    # Create admin interface
    admin = admin.Admin(app)
    admin.add_view(fileadmin.FileAdmin(path, '/files/%s' %username, name='Files'))

    # Start app
    app.run(debug=True)

Answer 1

get_base_path方法在您需要做的事情中非常明确..

Override to customize behavior (per-user directories, etc)

所以，这就是你需要做的。您需要创建一个继承自 FileAdmin 的类，覆盖 get_base_path()方法，并使用那个类而不是基本的 FileAdmin。如果您不熟悉什么是“覆盖”方法，您应该查找 Python 面向对象的教程。

您创建的这个新类需要覆盖 get_base_path()方法是每个用户。所以，像...

class MyFileAdmin(FileAdmin):
    def get_base_path(self):
        path = FileAdmin.get_base_path(self)

        if not current_user.is_anonymous():
            return os.path.join(path, current_user.custom_path)
        else:
            return path

现在使用 MyFileAdmin代替 FileAdmin .我还没有测试过这段代码，但希望这能让你走上正轨。

请注意，您在实际 FileAdmin 实例上更改路径的原始计划将不起作用。您创建一个 FileAdmin 实例，所有用户的请求都将使用相同的 View 进行路由。如果您尝试这样做，您可能会遇到多线程问题。