webvirtcloud/instances/apps.py

from django.apps import AppConfig
from django.db.models.signals import post_migrate


def migrate_can_clone_instances(sender, **kwargs):
    '''
    Migrate can clone instances user attribute to permission
    '''
    from django.contrib.auth.models import Permission, User

    plan = kwargs.get('plan', [])
    for migration, rolled_back in plan:
        if migration.app_label == 'instances' and migration.name == '0002_permissionset' and not rolled_back:
            users = User.objects.all()
            permission = Permission.objects.get(codename='clone_instances')
            print('\033[1m* \033[92mMigrating can_clone_instaces user attribute to permission\033[0m')
            for user in users:
                if user.userattributes:
                    if user.userattributes.can_clone_instances:
                        user.user_permissions.add(permission)
            break


def apply_passwordless_console(sender, **kwargs):
    '''
    Apply new passwordless_console permission for all users
    '''
    from django.contrib.auth import get_user_model
    from django.contrib.auth.models import Permission
    User = get_user_model()
    plan = kwargs.get('plan', [])
    for migration, rolled_back in plan:
        if migration.app_label == 'instances' and migration.name == '0009_auto_20200717_0524' and not rolled_back:
            print('\033[1m* \033[92mApplying permission passwordless_console for all users\033[0m')
            users = User.objects.all()
            permission = Permission.objects.get(codename='passwordless_console')
            for user in users:
                user.user_permissions.add(permission)


class InstancesConfig(AppConfig):
    name = 'instances'
    verbose_name = 'Instances'

    def ready(self):
        post_migrate.connect(migrate_can_clone_instances, sender=self)
        post_migrate.connect(apply_passwordless_console, sender=self)