Flask Principal:按需提供细粒度资源

7
我一直在研究这篇文章:http://pythonhosted.org/Flask-Principal/#granular-resource-protection。现在它的工作没有问题,但我认为并不是很实用,因为在登录时所有帖子都会被读取,并将EditBlogPostNeed添加到标识中。如果我写了超过正常数量的帖子,那么长期来看这不是一个很好的策略,因为我想在访问/posts/<post_id>视图时检查帖子。有没有办法使用Flask Principal在每个请求视图时进行此检查?当然,我可以通过惰性关系查询和过滤轻松解决它,但我想使用Flask Principal

嘿,你找到解决方案了吗? - Avi Das
不,我停止使用Flask了,因为在Flask中有一些其他问题很难解决。现在我改用WebOb,以获得更多的控制权。 - Asken
这是关于Flask-Principal的相关问题,链接在Github上:https://github.com/mattupstate/flask-principal/issues/6 - Vlad Frolov
4个回答

1
这个主题的所有内容似乎都过于晦涩难懂。虽然不是我最初想要的,但我已决定在我的视图函数中手动处理它。这更加明确,并减少了对数据库的额外查询。请注意,我仍然使用flask-security进行其开箱即用的基于角色的身份验证(仍通过其@roles_accepted('role')装饰器实现)。
@app.route('/my_accounts/', methods = ['GET'])
@app.route('/my_accounts/<int:id>/', methods = ['GET'])
@roles_accepted('client')
def my_accounts(id=None):

    if id:
        account = Account.query.get_or_404(id)

        if account.owner == current_user:
            return render_template("my_account.html",
                                   title = "Account: {0}".format(account.name),
                                   account = account)
        else:
            abort(403)

    accounts = Account.query.filter_by(owner=current_user).all()

    return render_template("my_accounts.html",
                           title = 'My Accounts',
                           accounts = accounts)

1
你触动了我的心。 - userx

1

虽然Flask-Principal是最受欢迎的插件,但它过于复杂,而且在大多数情况下都无法满足我的需求。我一直试图强制它按照我喜欢的方式工作,但从未成功过。幸运的是,我找到了一个非常简单和轻量级的模块 - permission

用法

首先,您需要通过子类化Rule来定义自己的规则,然后重写check()deny()方法:

# rules.py
from flask import session, flash, redirect, url_for
from permission import Rule

class UserRule(Rule):
    def check(self):
        """Check if there is a user signed in."""
        return 'user_id' in session

    def deny(self):
        """When no user signed in, redirect to signin page."""
        flash('Sign in first.')
        return redirect(url_for('signin'))

然后,您可以通过子类化Permission并覆盖rule()来定义权限:
# permissions.py
from permission import Permission
from .rules import UserRule

class UserPermission(Permission):
    """Only signin user has this permission."""
    def rule(self):
        return UserRule()

有4种方法可以使用上面定义的UserPermission

1. 作为视图装饰器使用

from .permissions import UserPermission

@app.route('/settings')
@UserPermission()
def settings():
    """User settings page, only accessable for sign-in user."""
    return render_template('settings.html')

2. 在视图代码中使用

from .permissions import UserPermission

@app.route('/settions')
def settings():
    permission = UserPermission()
    if not permission.check()
        return permission.deny()
    return render_template('settings.html')

3. 在视图代码中使用(使用 with 语句)

from .permissions import UserPermission

@app.route('/settions')
def settings():
    with UserPermission():
        return render_template('settings.html')

4.在Jinja2模板中使用

首先,您需要将定义的权限注入到模板上下文中:

from . import permissions

@app.context_processor
def inject_vars():
    return dict(
        permissions=permissions
    )

然后在模板中:
{% if permissions.UserPermission().check() %}
    <a href="{{ url_for('new') }}">New</a>
{% endif %}

1

我不确定完全理解你的问题,但这可能有所帮助。在Flask应用程序中,我使用Flask-Principal进行角色权限控制,例如管理员和编辑器,并且我还将其用于精细的资源保护,如 Flask-Principal文档中所述。在我的情况下,我正在检查用户是否有权限访问特定帐户。在每个视图中,都会加载标识并检查权限。

视图中:

@login_required
def call_list(id, page=1):

    dept = models.Department.query.get_or_404(id)
    view_permission = auth.ViewAccountPermission(dept.office.account_id)

    if view_permission.can():
        # do something

自定义权限:

ViewAccount = namedtuple('View', ['method', 'value'])
ViewAccountNeed = partial(ViewAccount, 'view')

class ViewAccountPermission(Permission):
    def __init__(self, account_id):
        need = ViewAccountNeed(unicode(account_id))
        super(ViewAccountPermission, self).__init__(need)

在身份加载器函数中:
if hasattr(current_user, 'assigned_accounts'):
    for account_id in current_user.assigned_accounts():
        identity.provides.add(auth.ViewAccountNeed(unicode(account_id)))

0

我的回答基于您已经了解了Flask Principal的工作原理以及它如何与数据库集成。

首先,我们只需要在数据库中存储需求, 如果您不知道为什么,我不建议您阅读下面的答案

然后,回到您的问题,我们需要编辑一篇文章,如何进行管控?

@app.route('/article/edit/<id>'):
@Permission(Need('edit', 'article')).require()
def article(id):
    pass

用户的身份

id = identity(user.id)
id.provide.add(Need('edit','article'))

用户有编辑文章的权限。@Permission(Need('edit', 'article')).require()将为每篇文章返回true,即使用户不是文章的作者,这是你的问题,对吧?

以下是我解决这个问题的方法

因为默认的Permission.require()没有提供任何参数传递,所以我定义了自己的Permission和IdentityContext,并传入文章ID和文章模型,然后检查文章的user_id是否与Flask登录的current_user.id相匹配。

class MyPermission(Permission):
    pass


class MyIdentityContext():

    pass

如果用户是文章的作者,那么我返回True,用户可以编辑文章。如果不是,则返回False,这样就可以工作了。
--------稍后我会更新更多细节------------

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接