Revision ffb1e7a8 snf-astakos-app/astakos/im/management/commands/group_add.py

b/snf-astakos-app/astakos/im/management/commands/group_add.py
39 39
from os.path import abspath
40 40

  
41 41
from django.core.management.base import BaseCommand, CommandError
42
from django.contrib.auth.models import Group
42

  
43
from astakos.im.models import AstakosGroup
43 44

  
44 45
from ._common import add_group_permission
45 46

  
......
54 55
        name = args[0].decode('utf8')
55 56
        
56 57
        try:
57
            Group.objects.get(name=name)
58
            AstakosGroup.objects.get(name=name)
58 59
            raise CommandError("A group with this name already exists")
59
        except Group.DoesNotExist, e:
60
            group = Group(name=name)
60
        except AstakosGroup.DoesNotExist, e:
61
            group = AstakosGroup(name=name)
61 62
            group.save()
62 63
            msg = "Created group id %d" % (group.id,)
63 64
            self.stdout.write(msg + '\n')

Also available in: Unified diff