Revision 304acb60 snf-astakos-app/astakos/im/management/commands/user_add.py
b/snf-astakos-app/astakos/im/management/commands/user_add.py | ||
---|---|---|
41 | 41 |
from django.core.management.base import BaseCommand, CommandError |
42 | 42 |
from django.core.validators import validate_email |
43 | 43 |
from django.core.exceptions import ValidationError |
44 |
from django.contrib.auth.models import Group |
|
45 | 44 |
|
46 |
from astakos.im.models import AstakosUser |
|
45 |
from astakos.im.models import AstakosUser, AstakosGroup
|
|
47 | 46 |
from astakos.im.util import reserved_email |
48 | 47 |
|
49 | 48 |
from ._common import add_user_permission |
... | ... | |
121 | 120 |
groupname = options.get('add-group') |
122 | 121 |
if groupname is not None: |
123 | 122 |
try: |
124 |
group = Group.objects.get(name=groupname) |
|
125 |
user.groups.add(group) |
|
123 |
group = AstakosGroup.objects.get(name=groupname)
|
|
124 |
user.astakos_groups.add(group)
|
|
126 | 125 |
self.stdout.write('Group: %s added successfully\n' % groupname) |
127 |
except Group.DoesNotExist, e: |
|
126 |
except AstakosGroup.DoesNotExist, e:
|
|
128 | 127 |
self.stdout.write('Group named %s does not exist\n' % groupname) |
129 | 128 |
|
130 | 129 |
pname = options.get('add-permission') |
Also available in: Unified diff