mirror of
https://github.com/retspen/webvirtcloud
synced 2024-12-24 23:25:24 +00:00
47 lines
1.9 KiB
Python
47 lines
1.9 KiB
Python
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)
|