1
0
Fork 0
mirror of https://github.com/retspen/webvirtcloud synced 2025-07-31 12:41:08 +00:00

format python code with black

This commit is contained in:
catborise 2022-11-02 08:54:35 +03:00
parent ea409ca863
commit 217e106c8b
55 changed files with 2510 additions and 1454 deletions

View file

@ -3,17 +3,19 @@ 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', [])
plan = kwargs.get("plan", [])
for migration, rolled_back in plan:
if migration.app_label == 'instances' and migration.name == '0002_permissionset' and not rolled_back:
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')
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:
@ -22,25 +24,26 @@ def migrate_can_clone_instances(sender, **kwargs):
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', [])
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')
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')
permission = Permission.objects.get(codename="passwordless_console")
for user in users:
user.user_permissions.add(permission)
class InstancesConfig(AppConfig):
name = 'instances'
verbose_name = 'Instances'
name = "instances"
verbose_name = "Instances"
def ready(self):
post_migrate.connect(migrate_can_clone_instances, sender=self)