diff --git a/companies/views.py b/companies/views.py index c92ab46..436bd58 100644 --- a/companies/views.py +++ b/companies/views.py @@ -19,6 +19,7 @@ from companies.models import Company from companies.serializers import CompanySerializer from utils.tag_filters import CompanyTagFilter from back_latienda.permissions import IsCreator +from rest_framework import filters from utils import woocommerce @@ -28,6 +29,9 @@ class CompanyViewSet(viewsets.ModelViewSet): serializer_class = CompanySerializer permission_classes = [IsAuthenticatedOrReadOnly, IsCreator] filterset_class = CompanyTagFilter + filter_backends = [filters.SearchFilter] + search_fields = ['company_name__unaccent__icontains', 'short_name__unaccent__icontains'] + def perform_create(self, serializer): serializer.save(creator=self.request.user)