show group categories in api

2017-07-10 16:30:38 +02:00 · 2017-07-10 16:30:38 +02:00 · 3762ebf37d
commit 3762ebf37d
parent 782666bcaf
3 changed files with 13 additions and 7 deletions
--- a/src/c3nav/mapdata/models/locations.py
+++ b/src/c3nav/mapdata/models/locations.py
@ -144,7 +144,10 @@ class SpecificLocation(Location, models.Model):
    def _serialize(self, detailed=True, **kwargs):
        result = super()._serialize(detailed=detailed, **kwargs)
        if detailed:
-            result['groups'] = list(g.pk for g in self.groups.all())
+            groups = {}
+            for group in self.groups.all():
+                groups.setdefault(group.category.name, []).append(group.pk)
+            result['groups'] = groups
        return result