41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.db.models.signals import post_save
|
|
|
|
from navigation.api import register_links, register_multi_item_links
|
|
from project_setup.api import register_setup
|
|
|
|
from .models import Role
|
|
from .links import (permission_grant, permission_revoke, role_create,
|
|
role_delete, role_edit, role_list, role_members,
|
|
role_permissions)
|
|
from .settings import DEFAULT_ROLES
|
|
|
|
register_links(Role, [role_delete, role_edit, role_members, role_permissions])
|
|
register_links([Role, 'permissions:role_create', 'permissions:role_list'], [role_create, role_list], menu_name='secondary_menu')
|
|
register_multi_item_links(['permissions:role_permissions'], [permission_grant, permission_revoke])
|
|
|
|
permission_views = ['permissions:role_list', 'permissions:role_create', 'permissions:role_edit', 'permissions:role_members', 'permissions:role_permissions', 'permissions:role_delete']
|
|
|
|
|
|
def user_post_save(sender, instance, **kwargs):
|
|
if kwargs.get('created', False):
|
|
for default_role in DEFAULT_ROLES:
|
|
if isinstance(default_role, Role):
|
|
# If a model is passed, execute method
|
|
default_role.add_member(instance)
|
|
else:
|
|
# If a role name is passed, lookup the corresponding model
|
|
try:
|
|
role = Role.objects.get(name=default_role)
|
|
role.add_member(instance)
|
|
except ObjectDoesNotExist:
|
|
pass
|
|
|
|
|
|
post_save.connect(user_post_save, sender=User)
|
|
|
|
register_setup(role_list)
|