118 lines
3.5 KiB
Python
118 lines
3.5 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import django_filters
|
|
from django.db.models import Q
|
|
|
|
from dcim.models import Platform, Site
|
|
from extras.filters import CustomFieldFilterSet
|
|
from tenancy.models import Tenant
|
|
from utilities.filters import NullableModelMultipleChoiceFilter, NumericInFilter
|
|
from .constants import STATUS_CHOICES
|
|
from .models import Cluster, ClusterGroup, ClusterType, VirtualMachine
|
|
|
|
|
|
class ClusterFilter(CustomFieldFilterSet):
|
|
id__in = NumericInFilter(name='id', lookup_expr='in')
|
|
q = django_filters.CharFilter(
|
|
method='search',
|
|
label='Search',
|
|
)
|
|
group_id = NullableModelMultipleChoiceFilter(
|
|
queryset=ClusterGroup.objects.all(),
|
|
label='Parent group (ID)',
|
|
)
|
|
group = NullableModelMultipleChoiceFilter(
|
|
queryset=ClusterGroup.objects.all(),
|
|
to_field_name='slug',
|
|
label='Parent group (slug)',
|
|
)
|
|
type_id = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=ClusterType.objects.all(),
|
|
label='Cluster type (ID)',
|
|
)
|
|
type = django_filters.ModelMultipleChoiceFilter(
|
|
name='type__slug',
|
|
queryset=ClusterType.objects.all(),
|
|
to_field_name='slug',
|
|
label='Cluster type (slug)',
|
|
)
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=Site.objects.all(),
|
|
label='Site (ID)',
|
|
)
|
|
site = django_filters.ModelMultipleChoiceFilter(
|
|
name='site__slug',
|
|
queryset=Site.objects.all(),
|
|
to_field_name='slug',
|
|
label='Site (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = Cluster
|
|
fields = ['name']
|
|
|
|
def search(self, queryset, name, value):
|
|
if not value.strip():
|
|
return queryset
|
|
return queryset.filter(
|
|
Q(name__icontains=value) |
|
|
Q(comments__icontains=value)
|
|
)
|
|
|
|
|
|
class VirtualMachineFilter(CustomFieldFilterSet):
|
|
id__in = NumericInFilter(name='id', lookup_expr='in')
|
|
q = django_filters.CharFilter(
|
|
method='search',
|
|
label='Search',
|
|
)
|
|
status = django_filters.MultipleChoiceFilter(
|
|
choices=STATUS_CHOICES
|
|
)
|
|
cluster_group_id = NullableModelMultipleChoiceFilter(
|
|
name='cluster__group',
|
|
queryset=ClusterGroup.objects.all(),
|
|
label='Cluster group (ID)',
|
|
)
|
|
cluster_group = NullableModelMultipleChoiceFilter(
|
|
name='cluster__group__slug',
|
|
queryset=ClusterGroup.objects.all(),
|
|
to_field_name='slug',
|
|
label='Cluster group (slug)',
|
|
)
|
|
cluster_id = NullableModelMultipleChoiceFilter(
|
|
queryset=Cluster.objects.all(),
|
|
label='Cluster (ID)',
|
|
)
|
|
tenant_id = NullableModelMultipleChoiceFilter(
|
|
queryset=Tenant.objects.all(),
|
|
label='Tenant (ID)',
|
|
)
|
|
tenant = NullableModelMultipleChoiceFilter(
|
|
queryset=Tenant.objects.all(),
|
|
to_field_name='slug',
|
|
label='Tenant (slug)',
|
|
)
|
|
platform_id = NullableModelMultipleChoiceFilter(
|
|
queryset=Platform.objects.all(),
|
|
label='Platform (ID)',
|
|
)
|
|
platform = NullableModelMultipleChoiceFilter(
|
|
name='platform',
|
|
queryset=Platform.objects.all(),
|
|
to_field_name='slug',
|
|
label='Platform (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = VirtualMachine
|
|
fields = ['name', 'cluster']
|
|
|
|
def search(self, queryset, name, value):
|
|
if not value.strip():
|
|
return queryset
|
|
return queryset.filter(
|
|
Q(name__icontains=value) |
|
|
Q(comments__icontains=value)
|
|
)
|