import django_filters
from rest_framework.filters import BaseFilterBackend

from dpdapi.models import Alias


class AliasRegexFilter(django_filters.FilterSet):
    class Meta:
        model = Alias
        fields = {"source": ["iregex", "regex"], "destination": ["iregex", "regex"]}


class AliasRegexFilterBackend(BaseFilterBackend):
    def filter_queryset(self, request, queryset, view):
        return AliasRegexFilter(request.query_params, queryset=queryset).qs