Django 中间件确定会话中的用户组

Question

我有一个使用 django.contrib.auth 但不使用 Django 内置权限系统的应用程序。相反，视图具有 @login_required 装饰器，然后检查用户属于哪个组，并根据组在视图内遵循不同的代码执行分支。

一个用户只能属于一个组。

每次检查用户的组似乎太多了，所以我尝试编写一个 Django 中间件，让我知道会话中的用户组。

看看下面的代码，我的中间件会像我想要的那样工作吗？

class SetGroupMiddleware(object):
    def process_request(self, request):
        check_if_already_set = request.session.get('thegroup', 'notset')
        if check_if_already_set == 'notset':
            if request.user.id: # User is not AnonymousUser
                groups = request.user.groups.all()
                if groups: # actually this will always be True
                    request.session['thegroup'] = str(groups[0].name) # flowchart of the app ensures that the logged in user will only have one group, and that the user will always have a group
            else:
                request.session['thegroup'] = 'nogroup' # for completeness

然后我打算在需要的地方检查 request.session['thegroup'] 。

需要您的建议和意见。如果这样处理，会话安全吗？这会起作用吗？我对 Django、Python 和一般编程都是新手。

谢谢。

Answer 1

总的来说看起来不错。不过你可以让它更Pythonic一点：

class SetGroupMiddleware(object):
    def process_request(self, request):
        if 'thegroup' not in request.session:
            if not request.user.is_anonymous():
                groups = request.user.groups.all()
                if groups:
                    request.session['thegroup'] = str(groups[0].name)
            else:
                request.session['thegroup'] = None # for completeness

Answer 2

它看起来大致正确（没有测试过）。需要注意的一件事是，您的中间件必须出现在 MIDDLEWARE_CLASSES 列表中的 之后 django.contrib.sessions.middleware.SessionMiddleware ，否则会话不会为您设置当您尝试引用它时。

Answer 3

好吧，正如我在史蒂夫·洛什的回答<中评论的那样< /a> ，该代码无法按预期工作。

我修改如下，目前看来还可以：-

class SetGroupMiddleware(object):
    def process_request(self, request):
        if not request.user.is_anonymous():
            if 'thegroup' not in request.session:
                groups = request.user.groups.all()
                if groups:
                    request.session['thegroup'] = str(groups[0].name)