Merge pull request #47 from subodh30/apartments_search_functionality

added apartments search functionality

housing/tests.py

@@ -53,6 +53,7 @@ def test_delete_owner(self):
         self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
         self.assertEqual(Owner.objects.count(), 0)
 
+
 class InterestedTests(APITestCase, TestCase):
 
     @classmethod
@@ -162,4 +163,4 @@ def test_delete_flat(self):
         response = self.client.delete(url, data, format='json')
 
         self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
-        self.assertEqual(Flat.objects.count(), 0)
+        self.assertEqual(Flat.objects.count(), 0)

housing/urls.py

@@ -8,7 +8,7 @@
 # router.register(r'flats', views.FlatViewSet, basename="flat_viewset")
 # router.register(r'owners', views.OwnerViewSet, basename="owner_viewset")
 # router.register(r'interests', views.InterestedViewSet, basename="interests_viewset")
-router.register(r'apartments', views.ApartmentViewSet, basename="apartment_viewset")
+# router.register(r'apartments', views.ApartmentViewSet, basename="apartment_viewset")
 # router.register(r'lease', views.LeaseViewSet, basename='lease_viewset')
 
 urlpatterns = [
@@ -21,5 +21,7 @@
     path('lease/<str:pk>', views.LeaseViewSet.as_view()),
     path('interests', views.InterestedViewSet.as_view()),
     path('interests/<str:pk>', views.InterestedViewSet.as_view()),
-
+    path('apartments', views.ApartmentViewSet.as_view()),
+    path('apartments/<str:pk>', views.ApartmentViewSet.as_view()),
+
 ]

housing/views.py

@@ -37,7 +37,9 @@ class LeaseViewSet(generics.ListCreateAPIView, generics.RetrieveUpdateDestroyAPI
     serializer_class = serializers.LeaseSerializer
 
 
-class ApartmentViewSet(viewsets.ModelViewSet):
+class ApartmentViewSet(generics.ListCreateAPIView, generics.RetrieveUpdateDestroyAPIView):
+    search_fields = ['address', 'facilities', 'owner_id']
+    filter_backends = (filters.SearchFilter,)
     queryset = models.Apartment.objects.all()
     serializer_class = serializers.ApartmentSerializer