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)