mirror of https://github.com/jumpserver/jumpserver
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
from django_filters import rest_framework as filters
|
|
from django.db.models.functions import Concat
|
|
from django.db.models import Subquery, OuterRef, Value, F, Q
|
|
|
|
from common.drf.filters import BaseFilterSet
|
|
|
|
from tickets.models import (
|
|
Ticket, TicketStep, ApplyAssetTicket, ApplyApplicationTicket,
|
|
ApplyLoginTicket, ApplyLoginAssetTicket, ApplyCommandTicket
|
|
)
|
|
|
|
|
|
class TicketFilter(BaseFilterSet):
|
|
assignees__id = filters.UUIDFilter(method='filter_assignees_id')
|
|
relevant_app = filters.CharFilter(method='filter_relevant_app')
|
|
relevant_asset = filters.CharFilter(method='filter_relevant_asset')
|
|
relevant_system_user = filters.CharFilter(method='filter_relevant_system_user')
|
|
relevant_command = filters.CharFilter(method='filter_relevant_command')
|
|
|
|
class Meta:
|
|
model = Ticket
|
|
fields = (
|
|
'id', 'title', 'type', 'status', 'state', 'applicant', 'assignees__id'
|
|
)
|
|
|
|
def filter_assignees_id(self, queryset, name, value):
|
|
step_qs = TicketStep.objects.filter(
|
|
level=OuterRef("approval_step")
|
|
).values_list('id', flat=True)
|
|
return queryset.filter(
|
|
ticket_steps__id__in=Subquery(step_qs),
|
|
ticket_steps__ticket_assignees__assignee__id=value
|
|
)
|
|
|
|
def filter_relevant_asset(self, queryset, name, value):
|
|
asset_ids = ApplyAssetTicket.objects.annotate(
|
|
asset_str=Concat(
|
|
F('apply_assets__hostname'), Value('('),
|
|
F('apply_assets__ip'), Value(')')
|
|
)
|
|
).filter(
|
|
asset_str__icontains=value
|
|
).values_list('id', flat=True)
|
|
|
|
login_asset_ids = ApplyLoginAssetTicket.objects.annotate(
|
|
asset_str=Concat(
|
|
F('apply_login_asset__hostname'), Value('('),
|
|
F('apply_login_asset__ip'), Value(')')
|
|
)
|
|
).filter(
|
|
asset_str__icontains=value
|
|
).values_list('id', flat=True)
|
|
|
|
command_ids = ApplyCommandTicket.objects.filter(
|
|
apply_run_asset__icontains=value
|
|
).values_list('id', flat=True)
|
|
|
|
ticket_ids = list(set(list(asset_ids) + list(login_asset_ids) + list(command_ids)))
|
|
return queryset.filter(id__in=ticket_ids)
|
|
|
|
def filter_relevant_app(self, queryset, name, value):
|
|
app_ids = ApplyApplicationTicket.objects.filter(
|
|
apply_applications__name__icontains=value
|
|
).values_list('id', flat=True)
|
|
|
|
command_ids = ApplyCommandTicket.objects.filter(
|
|
apply_run_asset__icontains=value
|
|
).values_list('id', flat=True)
|
|
|
|
ticket_ids = list(set(list(app_ids) + list(command_ids)))
|
|
return queryset.filter(id__in=ticket_ids)
|
|
|
|
def filter_relevant_system_user(self, queryset, name, value):
|
|
system_user_query = Q(apply_system_users__name__icontains=value)
|
|
asset_ids = ApplyAssetTicket.objects.filter(
|
|
system_user_query
|
|
).values_list('id', flat=True)
|
|
|
|
app_ids = ApplyApplicationTicket.objects.filter(
|
|
system_user_query
|
|
).values_list('id', flat=True)
|
|
|
|
login_asset_ids = ApplyLoginAssetTicket.objects.filter(
|
|
apply_login_system_user__name__icontains=value
|
|
).values_list('id', flat=True)
|
|
|
|
command_ids = ApplyCommandTicket.objects.filter(
|
|
apply_run_system_user__name__icontains=value
|
|
).values_list('id', flat=True)
|
|
ticket_ids = list(
|
|
set(list(asset_ids) + list(app_ids) + list(login_asset_ids) + list(command_ids))
|
|
)
|
|
return queryset.filter(id__in=ticket_ids)
|
|
|
|
def filter_relevant_command(self, queryset, name, value):
|
|
command_ids = ApplyCommandTicket.objects.filter(
|
|
apply_run_command__icontains=value
|
|
).values_list('id', flat=True)
|
|
return queryset.filter(id__in=list(command_ids))
|
|
|
|
|
|
class ApplyAssetTicketFilter(BaseFilterSet):
|
|
class Meta:
|
|
model = ApplyAssetTicket
|
|
fields = ('id',)
|
|
|
|
|
|
class ApplyApplicationTicketFilter(BaseFilterSet):
|
|
class Meta:
|
|
model = ApplyApplicationTicket
|
|
fields = ('id',)
|
|
|
|
|
|
class ApplyLoginTicketFilter(BaseFilterSet):
|
|
class Meta:
|
|
model = ApplyLoginTicket
|
|
fields = ('id',)
|
|
|
|
|
|
class ApplyLoginAssetTicketFilter(BaseFilterSet):
|
|
class Meta:
|
|
model = ApplyLoginAssetTicket
|
|
fields = ('id',)
|
|
|
|
|
|
class ApplyCommandTicketFilter(BaseFilterSet):
|
|
class Meta:
|
|
model = ApplyCommandTicket
|
|
fields = ('id',)
|