mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
2.9 KiB
89 lines
2.9 KiB
from django.db.models import Subquery, OuterRef, Value, F, Q |
|
from django_filters import rest_framework as filters |
|
from django.db.models.functions import Concat |
|
|
|
from common.drf.filters import BaseFilterSet |
|
|
|
from tickets.models import ( |
|
Ticket, TicketStep, ApplyAssetTicket, |
|
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__name'), Value('('), |
|
F('apply_assets__address'), Value(')') |
|
) |
|
).filter( |
|
asset_str__icontains=value |
|
).values_list('id', flat=True) |
|
|
|
login_asset_ids = ApplyLoginAssetTicket.objects.annotate( |
|
asset_str=Concat( |
|
F('apply_login_asset__name'), Value('('), |
|
F('apply_login_asset__address'), 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_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 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',)
|
|
|