mirror of https://github.com/jumpserver/jumpserver
28 lines
927 B
Python
28 lines
927 B
Python
from django.dispatch import receiver
|
|
from django.db.models.signals import post_migrate, post_save
|
|
from django.apps import apps
|
|
|
|
from .models import SystemRole, OrgRole
|
|
from .builtin import BuiltinRole
|
|
|
|
|
|
@receiver(post_migrate)
|
|
def after_migrate_update_builtin_role_permissions(sender, app_config, **kwargs):
|
|
# 最后一个 app migrations 后执行, 更新内置角色的权限
|
|
last_app = list(apps.get_app_configs())[-1]
|
|
if app_config.name == last_app.name:
|
|
print("\nAfter migration, update builtin role permissions")
|
|
BuiltinRole.sync_to_db()
|
|
|
|
|
|
@receiver(post_save, sender=SystemRole)
|
|
def on_system_role_update(sender, instance, created, **kwargs):
|
|
from users.models import User
|
|
User.expire_users_rbac_perms_cache()
|
|
|
|
|
|
@receiver(post_save, sender=OrgRole)
|
|
def on_org_role_update(sender, instance, created, **kwargs):
|
|
from users.models import User
|
|
User.expire_users_rbac_perms_cache()
|