macie beacona

BeHub/Hub/views.py

@@ -56,8 +56,10 @@ class UuidBeaconViewSet(APIView):
                 site = beacon_manager.sites.get(category=category)
             except ObjectDoesNotExist:
                 return Response(status=rest_framework.status.HTTP_400_BAD_REQUEST, data={"Any link do not have this category"})
-            serializer = SiteSerializer(site, context={'request': request})
-            return Response(serializer.data)
+            serializer_sites = SiteSerializer(site, context={'request': request})
+        else:
+            serializer_sites = SiteSerializer(beacon_manager.sites, many=True, context={'request': request})
+
         serializer = BeaconManagerSerializer(beacon_manager, context={'request': request})
         serializer_data = serializer.data
         serializer_sites = SiteSerializer(beacon_manager.sites, many=True, context={'request': request})