Commit cbc95b4d authored by Rafal's avatar Rafal

Merge remote-tracking branch 'origin/master'

parents 2e7a4ab7 e863e41a
...@@ -32,7 +32,7 @@ class Beacon(models.Model): ...@@ -32,7 +32,7 @@ class Beacon(models.Model):
class BeaconManager(models.Model): class BeaconManager(models.Model):
beacon = models.OneToOneField('Beacon') beacon = models.OneToOneField('Beacon')
sites = models.ManyToManyField('Site') sites = models.ManyToManyField('Site', blank=True)
def __str__(self): def __str__(self):
return str(self.beacon) return str(self.beacon)
...@@ -56,10 +56,16 @@ class UuidBeaconViewSet(APIView): ...@@ -56,10 +56,16 @@ class UuidBeaconViewSet(APIView):
site = beacon_manager.sites.get(category=category) site = beacon_manager.sites.get(category=category)
except ObjectDoesNotExist: except ObjectDoesNotExist:
return Response(status=rest_framework.status.HTTP_400_BAD_REQUEST, data={"Any link do not have this category"}) return Response(status=rest_framework.status.HTTP_400_BAD_REQUEST, data={"Any link do not have this category"})
serializer = SiteSerializer(site, context={'request': request}) serializer_sites = SiteSerializer(site, context={'request': request})
return Response(serializer.data) else:
serializer = SiteSerializer(beacon_manager.sites, many=True, context={'request': request}) serializer_sites = SiteSerializer(beacon_manager.sites, many=True, context={'request': request})
return Response(serializer.data)
serializer = BeaconManagerSerializer(beacon_manager, context={'request': request})
serializer_data = serializer.data
serializer_beacon = BeaconSerializer(beacon_manager.beacon, context={'request': request})
serializer_data['sites'] = serializer_sites.data
serializer_data['beacon'] = serializer_beacon.data
return Response(serializer_data)
class CategoryViewSet(viewsets.ModelViewSet): class CategoryViewSet(viewsets.ModelViewSet):
""" """
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment