From 28ed2f8f0f9e3be67e7f72014ad5a80cbffce462 Mon Sep 17 00:00:00 2001 From: pablogg Date: Tue, 16 Mar 2021 13:08:41 +0100 Subject: [PATCH] Search in companies --- companies/views.py | 4 ++++ 1 file changed, 4 insertions(+) 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)