W Django, jak sprawdzić, czy użytkownik jest w określonej grupie?

Utworzyłem niestandardową grupę w serwisie administratora Django.

W moim kodzie chcę sprawdzić, czy użytkownik jest w tej grupie. W jaki sposób mogę to zrobić?

Dostęp do grup można uzyskać po prostu za pomocą groupsatrybutu on User.

from django.contrib.auth.models import User, Group

group = Group(name = "Editor")
group.save()                    # save this new group for this example
user = User.objects.get(pk = 1) # assuming, there is one initial user 
user.groups.add(group)          # user is now in the "Editor" group

potem user.groups.all()wraca [<Group: Editor>].

Alternatywnie, i bardziej bezpośrednio, możesz sprawdzić, czy użytkownik jest w grupie:

if django_user.groups.filter(name = groupname).exists():

    ...

Zauważ, że groupnamemoże to być również rzeczywisty obiekt Django Group.

Twój obiekt użytkownika jest połączony z obiektem Group za pośrednictwem relacji ManyToMany .

W ten sposób możesz zastosować metodę filtru do user.groups .

Aby więc sprawdzić, czy dany Użytkownik należy do określonej grupy (na przykład „Członek”), wykonaj następujące czynności:

def is_member(user):
    return user.groups.filter(name='Member').exists()

Jeśli chcesz sprawdzić, czy dany użytkownik należy do więcej niż jednej podanej grupy, użyj operatora __in w następujący sposób:

def is_in_multiple_groups(user):
    return user.groups.filter(name__in=['group1', 'group2']).exists()

Pamiętaj, że te funkcje mogą być używane z dekoratorem @user_passes_test do zarządzania dostępem do twoich widoków:

from django.contrib.auth.decorators import login_required, user_passes_test
@login_required
@user_passes_test(is_member) # or @user_passes_test(is_in_multiple_groups)
def myview(request):
    # Do your processing

Mam nadzieję, że to pomoże

Jeśli potrzebujesz listy użytkowników należących do grupy, możesz to zrobić:

from django.contrib.auth.models import Group
users_in_group = Group.objects.get(name="group name").user_set.all()

a następnie sprawdź

 if user in users_in_group:
     # do something

aby sprawdzić, czy użytkownik jest w grupie.

Jeśli nie potrzebujesz instancji użytkownika na miejscu (tak jak ja), możesz to zrobić za pomocą

User.objects.filter(pk=userId, groups__name='Editor').exists()

Spowoduje to wygenerowanie tylko jednego żądania do bazy danych i zwrócenie wartości logicznej.

Jeśli użytkownik należy do określonej grupy lub nie, można to sprawdzić w szablonach django za pomocą:

{% if group in request.user.groups.all %} "some action" {% endif %}

Potrzebujesz tylko jednej linii:

from django.contrib.auth.decorators import user_passes_test  

@user_passes_test(lambda u: u.groups.filter(name='companyGroup').exists())
def you_view():
    return HttpResponse("Since you're logged in, you can see this text!")

Na wszelki wypadek, jeśli chcesz sprawdzić, czy grupa użytkownika należy do predefiniowanej listy grup:

def is_allowed(user):
    allowed_group = set(['admin', 'lead', 'manager'])
    usr = User.objects.get(username=user)
    groups = [ x.name for x in usr.groups.all()]
    if allowed_group.intersection(set(groups)):
       return True
    return False

Mam podobną sytuację, chciałem sprawdzić, czy użytkownik jest w określonej grupie. Dlatego utworzyłem nowy plik utils.py, w którym umieściłem wszystkie moje małe narzędzia, które pomagają mi w całej aplikacji. Mam taką definicję:

utils.py

def is_company_admin(user):
    return user.groups.filter(name='company_admin').exists()

więc zasadniczo sprawdzam, czy użytkownik jest w grupie company_admin i dla jasności nazwałem tę funkcję is_company_admin .

Kiedy chcę sprawdzić, czy użytkownik jest na koncie company_admin , po prostu robię to:

views.py

from .utils import *

if is_company_admin(request.user):
        data = Company.objects.all().filter(id=request.user.company.id)

Teraz, jeśli chcesz przetestować to samo w swoim szablonie, możesz dodać is_user_admin w swoim kontekście, coś takiego:

views.py

return render(request, 'admin/users.html', {'data': data, 'is_company_admin': is_company_admin(request.user)})

Teraz możesz ocenić swoją odpowiedź w szablonie:

users.html

{% if is_company_admin %}
     ... do something ...
{% endif %}

Proste i przejrzyste rozwiązanie, oparte na odpowiedziach, które można znaleźć wcześniej w tym wątku, ale zrobiono inaczej. Mam nadzieję, że to komuś pomoże.

Przetestowane w Django 3.0.4.

W jednej linii:

'Groupname' in user.groups.values_list('name', flat=True)

Ocenia się albo Truealbo False.

Zrobiłem to w następujący sposób. Wydaje się nieefektywne, ale nie miałem innego wyjścia:

@login_required
def list_track(request):

usergroup = request.user.groups.values_list('name', flat=True).first()
if usergroup in 'appAdmin':
    tracks = QuestionTrack.objects.order_by('pk')
    return render(request, 'cmit/appadmin/list_track.html', {'tracks': tracks})

else:
    return HttpResponseRedirect('/cmit/loggedin')

User.objects.filter(username='tom', groups__name='admin').exists()

To zapytanie poinformuje użytkownika: "tom", czy należy do grupy "admin", czy nie

Zrobiłem to w ten sposób. Dla grupy o nazwie Editor.

# views.py
def index(request):
    current_user_groups = request.user.groups.values_list("name", flat=True)
    context = {
        "is_editor": "Editor" in current_user_groups,
    }
    return render(request, "index.html", context)

szablon

# index.html
{% if is_editor %}
  <h1>Editor tools</h1>
{% endif %}