1# use some undocumented Django tricks to execute custom logic after syncdb
2
3from django.db.models import signals
4from django.contrib import auth
5from django.conf import settings
6# In this file, it is critical that we import models *just like this*.  In
7# particular, we *cannot* do import common; from autotest_lib... import models.
8# This is because when we pass the models module to signal.connect(), it
9# calls id() on the module, and the id() of a module can differ depending on how
10# it was imported.  For that reason, we must import models as Django does -- not
11# through the autotest_lib magic set up through common.py.  If you do that, the
12# connection won't work and the dispatcher will simply never call the method.
13from frontend.afe import models
14
15BASIC_ADMIN = 'Basic admin'
16
17def create_admin_group(app, created_models, verbosity, **kwargs):
18    """\
19    Create a basic admin group with permissions for managing basic autotest
20    objects.
21    """
22    admin_group, created = auth.models.Group.objects.get_or_create(
23        name=BASIC_ADMIN)
24    admin_group.save() # must save before adding permissions
25    PermissionModel = auth.models.Permission
26    have_permissions = list(admin_group.permissions.all())
27    for model_name in ('host', 'label', 'test', 'aclgroup', 'profiler',
28                       'atomicgroup', 'hostattribute'):
29        for permission_type in ('add', 'change', 'delete'):
30            codename = permission_type + '_' + model_name
31            permissions = list(PermissionModel.objects.filter(
32                codename=codename))
33            if len(permissions) == 0:
34                if verbosity:
35                    print '  No permission ' + codename
36                continue
37            for permission in permissions:
38                if permission not in have_permissions:
39                    if verbosity:
40                        print '  Adding permission ' + codename
41                    admin_group.permissions.add(permission)
42    if verbosity:
43        if created:
44            print 'Created group "%s"' % BASIC_ADMIN
45        else:
46            print 'Group "%s" already exists' % BASIC_ADMIN
47
48if settings.AUTOTEST_CREATE_ADMIN_GROUPS:
49    signals.post_syncdb.connect(create_admin_group, sender=models)
50