j3d1/webvirtcloud
1
0
Fork 0
mirror of https://github.com/retspen/webvirtcloud synced 2025-07-31 12:41:08 +00:00
Code Issues Projects Releases Wiki Activity

More instances tests

Browse source
This commit is contained in:
Real-Gecko 2020-10-15 11:58:42 +06:00
parent 9e56378682
commit 0f66187e80
7 changed files with 769 additions and 390 deletions
Download patch file Download diff file Expand all files Collapse all files

13
instances/apps.py
View file

@ -6,9 +6,7 @@ def migrate_can_clone_instances(sender, **kwargs):
'''
Migrate can clone instances user attribute to permission
'''
from django.conf import settings
from django.contrib.auth.models import User, Permission
from accounts.models import UserAttributes
from django.contrib.auth.models import Permission, User
plan = kwargs['plan']
for migration, rolled_back in plan:
@ -22,22 +20,23 @@ def migrate_can_clone_instances(sender, **kwargs):
user.user_permissions.add(permission)
break
def apply_passwordless_console(sender, **kwargs):
'''
Apply new passwordless_console permission for all users
'''
from django.conf import settings
from django.contrib.auth.models import User, Permission
from django.contrib.auth.models import Permission, User
print('\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)
user.user_permissions.add(permission)
class InstancesConfig(AppConfig):
name = 'instances'
verbose_name = 'instances'
verbose_name = 'Instances'
def ready(self):
post_migrate.connect(migrate_can_clone_instances, sender=self)