diff --git a/src/c3nav/mapdata/models/locations.py b/src/c3nav/mapdata/models/locations.py index 5a13a42f..4a5bc8f3 100644 --- a/src/c3nav/mapdata/models/locations.py +++ b/src/c3nav/mapdata/models/locations.py @@ -171,11 +171,18 @@ class LocationGroupCategory(TitledMixin, models.Model): return result +class LocationGroupManager(models.Manager): + def get_queryset(self): + return super().get_queryset().select_related('category') + + class LocationGroup(Location, models.Model): category = models.ForeignKey(LocationGroupCategory, related_name='groups', on_delete=models.PROTECT, verbose_name=_('Category')) priority = models.IntegerField(default=0, db_index=True) + objects = LocationGroupManager() + class Meta: verbose_name = _('Location Group') verbose_name_plural = _('Location Groups')