jumpserver/apps/tickets/filters.py

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',)