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