team-3/src/c3nav/mapdata/models/base.py

from collections import OrderedDict
from django.db import models
from django.db.models.base import ModelBase
from django.utils.translation import get_language

from c3nav.mapdata.lastupdate import set_last_mapdata_update

FEATURE_TYPES = OrderedDict()


class FeatureBase(ModelBase):
    def __new__(mcs, name, bases, attrs):
        cls = super().__new__(mcs, name, bases, attrs)
        if not cls._meta.abstract and name != 'Source':
            FEATURE_TYPES[name.lower()] = cls
        return cls


class Feature(models.Model, metaclass=FeatureBase):
    EditorForm = None

    @property
    def title(self):
        if not hasattr(self, 'titles'):
            return self.name
        lang = get_language()
        if lang in self.titles:
            return self.titles[lang]
        return next(iter(self.titles.values())) if self.titles else self.name

    def save(self, *args, **kwargs):
        with set_last_mapdata_update():
            super().save(*args, **kwargs)

    def delete(self, *args, **kwargs):
        with set_last_mapdata_update():
            super().delete(*args, **kwargs)

    class Meta:
        abstract = True