✨ Feat: [#22] category 기준 filter class 구현
This commit is contained in:
17
portfolios/filters.py
Normal file
17
portfolios/filters.py
Normal file
@@ -0,0 +1,17 @@
|
||||
from django_filters import rest_framework as filters
|
||||
from django.db.models import Q
|
||||
from .models import Portfolio
|
||||
|
||||
class PortfolioFilter(filters.FilterSet):
|
||||
category = filters.CharFilter(method='filter_category')
|
||||
|
||||
def filter_category(self, queryset, name, value):
|
||||
categories = value.split(',')
|
||||
q = Q()
|
||||
for c in categories:
|
||||
q |= Q(category__contains=[c])
|
||||
return queryset.filter(q)
|
||||
|
||||
class Meta:
|
||||
model = Portfolio
|
||||
fields = ['category']
|
||||
Reference in New Issue
Block a user