1e8819cdf80ca0e0602d22551a50f970aa68e108dmbligh# use some undocumented Django tricks to execute custom logic after syncdb 2e8819cdf80ca0e0602d22551a50f970aa68e108dmbligh 3e8819cdf80ca0e0602d22551a50f970aa68e108dmblighfrom django.db.models import signals 4e8819cdf80ca0e0602d22551a50f970aa68e108dmblighfrom django.contrib import auth 562cf0477206aa81c0163de63b3c86c26ed72eb79Aviv Keshetfrom django.conf import settings 6043abc1c38db968c11d7a5f8b5830e53ceb0fc8cshoward# In this file, it is critical that we import models *just like this*. In 7043abc1c38db968c11d7a5f8b5830e53ceb0fc8cshoward# particular, we *cannot* do import common; from autotest_lib... import models. 8a5288b4bb2b09aafe914d0b7d5aab79a7e433eafshoward# This is because when we pass the models module to signal.connect(), it 9043abc1c38db968c11d7a5f8b5830e53ceb0fc8cshoward# calls id() on the module, and the id() of a module can differ depending on how 10043abc1c38db968c11d7a5f8b5830e53ceb0fc8cshoward# it was imported. For that reason, we must import models as Django does -- not 11043abc1c38db968c11d7a5f8b5830e53ceb0fc8cshoward# through the autotest_lib magic set up through common.py. If you do that, the 12043abc1c38db968c11d7a5f8b5830e53ceb0fc8cshoward# connection won't work and the dispatcher will simply never call the method. 13043abc1c38db968c11d7a5f8b5830e53ceb0fc8cshowardfrom frontend.afe import models 14e8819cdf80ca0e0602d22551a50f970aa68e108dmbligh 15e8819cdf80ca0e0602d22551a50f970aa68e108dmblighBASIC_ADMIN = 'Basic admin' 16e8819cdf80ca0e0602d22551a50f970aa68e108dmbligh 17e8819cdf80ca0e0602d22551a50f970aa68e108dmblighdef create_admin_group(app, created_models, verbosity, **kwargs): 180afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski """\ 190afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski Create a basic admin group with permissions for managing basic autotest 200afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski objects. 210afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski """ 220afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski admin_group, created = auth.models.Group.objects.get_or_create( 230afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski name=BASIC_ADMIN) 240afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski admin_group.save() # must save before adding permissions 250afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski PermissionModel = auth.models.Permission 260afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski have_permissions = list(admin_group.permissions.all()) 27d1677ebabfa318980cc6fbe3cd08990d800dcbacshoward for model_name in ('host', 'label', 'test', 'aclgroup', 'profiler', 280739d68fcbd84d02839a85ce3c594e774e82c602Simran Basi 'atomicgroup', 'hostattribute'): 290afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski for permission_type in ('add', 'change', 'delete'): 300afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski codename = permission_type + '_' + model_name 310afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski permissions = list(PermissionModel.objects.filter( 320afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski codename=codename)) 330afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski if len(permissions) == 0: 343986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet if verbosity: 353986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet print ' No permission ' + codename 360afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski continue 370afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski for permission in permissions: 380afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski if permission not in have_permissions: 393986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet if verbosity: 403986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet print ' Adding permission ' + codename 410afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181jadmanski admin_group.permissions.add(permission) 423986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet if verbosity: 433986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet if created: 443986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet print 'Created group "%s"' % BASIC_ADMIN 453986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet else: 463986bac27734a8c3cd864d9ade775131a2261413Aviv Keshet print 'Group "%s" already exists' % BASIC_ADMIN 47e8819cdf80ca0e0602d22551a50f970aa68e108dmbligh 4862cf0477206aa81c0163de63b3c86c26ed72eb79Aviv Keshetif settings.AUTOTEST_CREATE_ADMIN_GROUPS: 4962cf0477206aa81c0163de63b3c86c26ed72eb79Aviv Keshet signals.post_syncdb.connect(create_admin_group, sender=models) 50