Fix select_related

This commit is contained in:
Joseph Kennedy 2017-08-30 20:56:41 -04:00
parent abafe47d46
commit 2dd97896da

View File

@ -622,14 +622,14 @@ class InterfaceListFilter(django_filters.FilterSet):
def filter_site(self, queryset, name, value): def filter_site(self, queryset, name, value):
try: try:
site = Site.objects.select_related('device_id').get(**{name: value}) site = Device.objects.select_related('device_id','site').get(**{name: value})
return queryset return queryset
except Device.DoesNotExist: except Device.DoesNotExist:
return queryset.none() return queryset.none()
def filter_role(self, queryset, name, value): def filter_role(self, queryset, name, value):
try: try:
site = DeviceRole.objects.select_related('device_id').get(**{name: value}) site = Device.objects.select_related('device_id','device_role').get(**{name: value})
return queryset return queryset
except Device.DoesNotExist: except Device.DoesNotExist:
return queryset.none() return queryset.none()