2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

auto prefetch_related on LocationSlug

Browse source
This commit is contained in:
Laura Klünder 2017-06-18 18:42:30 +02:00
parent f992cd75ed
commit 279d9bab52
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/c3nav/mapdata/models/locations.py
View file

@ -11,6 +11,15 @@ from c3nav.mapdata.models.base import EditorFormMixin, SerializableMixin
LOCATION_MODELS = []
class LocationSlugManager(models.Manager):
def get_queryset(self):
result = super().get_queryset()
if self.model == LocationSlug:
result = result.select_related(*(model._meta.default_related_name
for model in LOCATION_MODELS+[LocationRedirect]))
return result
class LocationSlug(SerializableMixin, models.Model):
LOCATION_TYPE_CODES = {
'Level': 'l',
@ -22,9 +31,7 @@ class LocationSlug(SerializableMixin, models.Model):
LOCATION_TYPE_BY_CODE = {code: model_name for model_name, code in LOCATION_TYPE_CODES.items()}
slug = models.SlugField(_('Slug'), unique=True, null=True, max_length=50)
@classmethod
def get_prefetch_names(cls):
return [model._meta.default_related_name for model in LOCATION_MODELS+[LocationRedirect]]
objects = LocationSlugManager()
def get_child(self):
# todo: cache this