2018-06-28 13:48:12 -04:00
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
|
|
|
from django.db.models import Q, QuerySet
|
|
|
|
|
|
|
|
|
|
|
|
class ConfigContextQuerySet(QuerySet):
|
|
|
|
|
|
|
|
def get_for_object(self, obj):
|
|
|
|
"""
|
|
|
|
Return all applicable ConfigContexts for a given object. Only active ConfigContexts will be included.
|
|
|
|
"""
|
|
|
|
|
|
|
|
# `device_role` for Device; `role` for VirtualMachine
|
|
|
|
role = getattr(obj, 'device_role', None) or obj.role
|
|
|
|
|
|
|
|
return self.filter(
|
2018-07-03 13:40:47 -04:00
|
|
|
Q(regions=getattr(obj.site, 'region', None)) | Q(regions=None),
|
2018-06-28 13:48:12 -04:00
|
|
|
Q(sites=obj.site) | Q(sites=None),
|
|
|
|
Q(roles=role) | Q(roles=None),
|
|
|
|
Q(tenants=obj.tenant) | Q(tenants=None),
|
|
|
|
Q(platforms=obj.platform) | Q(platforms=None),
|
|
|
|
is_active=True,
|
|
|
|
).order_by('weight', 'name')
|