Add a user to a default role only when the user is created

This commit is contained in:
Roberto Rosario
2011-03-29 11:17:04 -04:00
parent 043f8f3502
commit c1bb3a90bb

View File

@@ -27,10 +27,13 @@ register_links(['role_list', 'role_view', 'role_create', 'role_edit', 'role_perm
def user_post_save(sender, instance, **kwargs): def user_post_save(sender, instance, **kwargs):
if kwargs.get('created', False):
for default_role in DEFAULT_ROLES: for default_role in DEFAULT_ROLES:
if isinstance(default_role, Role): if isinstance(default_role, Role):
#If a model is passed, execute method
default_role.add_member(instance) default_role.add_member(instance)
else: else:
#If a role name is passed, lookup the corresponding model
try: try:
role = Role.objects.get(name=default_role) role = Role.objects.get(name=default_role)
role.add_member(instance) role.add_member(instance)