CODEWITHSUNDEEP
pythondjangodjango-admindjango-authentication
billrichards
billrichards

Reputation: 2060

django - set user permissions when user is automatically created using get_or_create

Django 1.5, python 2.6

The model automatically creates a user under certain conditions:

User.objects.get_or_create(username=new_user_name, is_staff=True) 
u = User.objects.get(username=new_user_name)
u.set_password('temporary')

In addition to setting the username, password, and is_staff status, I would like to set the user's permissions - something like:

u.user_permissions('Can view poll')

or

u.set_permissions('Can change poll')

Is this possible? Thank you!

Upvotes: 43

Views: 58993

Answers (3)

alko
alko

Reputation: 48337

Use add and remove methods:

 from django.contrib.auth.models import Permission
 permission = Permission.objects.get(name='Can view poll')
 u.user_permissions.add(permission)

Upvotes: 79

BiswajitPaloi
BiswajitPaloi

Reputation: 641

Using codename filed:-

from django.contrib.auth import get_user_model
from django.contrib.auth.models import Permission

user = get_user_model().objects.get(name="test")
permission = Permission.objects.get(codename='view_poll')
user.user_permissions.add(permission)

# Check permission 
print(user.has_perm("<app_name).view_poll"))

Upvotes: 1

juanmhidalgo
juanmhidalgo

Reputation: 1546

Andrew M. Farrell's answer is correct. I only add the use of get_user_model() and a full example.

from django.contrib.auth import get_user_model
from django.contrib.auth.models import Permission
u = get_user_model().objects.get(username=new_user_name)

To get the permission you can use

permission = Permission.objects.get(name='Can view poll')

or

permission = Permission.objects.get(codename='can_view_poll')

then add it to the user permissions set

u.user_permissions.add(permission)

Upvotes: 28

Related Questions