Question

如果当前用户角色= admin，则显示表中的所有记录。如果不是，则按创建的用户限制行。

如果我在View Class中定义一个函数，则可以获取用户名，但是在构造列表之前需要它。请参见下面的源代码。

from flask_appbuilder.models.sqla.filters import FilterEqualFunction
from app import appbuilder, db
from app.models import Language
from wtforms import validators, TextField
from flask import g
from flask_appbuilder.security.sqla.models import User

def get_user():
    return g.user

class LanguageView(ModelView):
    datamodel = SQLAInterface(Language)
    list_columns = ["id", "name"]
    base_order = ("name", "asc")
    page_size = 50

    #This is the part that does not work - unable to import app Error: Working outside of application context
    #If the user role is admin show all, if not filter only to the specific user
    if g.user.roles != "admin":
        base_filters = [['created_by', FilterEqualFunction, get_user]]

这是我得到的错误：

无法导入应用程序错误：在应用程序上下文之外工作。

这通常意味着您尝试使用所需的功能以某种方式与当前应用程序对象交互。解决为此，使用app.app_context（）设置应用程序上下文。见文档以获取更多信息。

Answer 1

在这种情况下，最好创建两个不同的 ModelView，一个用于具有 base_filters = [['created_by', FilterEqualFunction, get_user]] 的用户，第二个仅用于没有任何过滤的管理员。

并且不要忘记为两者指定正确的权限。

Answer 2

就我而言，我通过处理 FilterStartsWith 创建了新的过滤器 FilterStartsWithFunction（您可以轻松地在 Flask-Appbuilder 包中找到源代码。）。看代码

from flask_appbuilder.models.sqla.filters import get_field_setup_query

class FilterStartsWithFunction(BaseFilter):
    name = "Filter view with a function"
    arg_name = "eqf"

    def apply(self, query, func):
        query, field = get_field_setup_query(query, self.model, self.column_name)
        return query.filter(field.ilike(func() + "%"))

def get_user():
    if 'Admin' in [r.name for r in g.user.roles]:
        return ''
    else:
        return g.user.username

...
...

    base_filters = [['created_by',FilterStartsWithFunction,get_user]]

Flask-Appbuilder-View中需要用户安全角色

2 个答案: