Полиморфизм модели django с наследованием прокси

Моя модель Discount описывает общие поля для всех типов скидок в системе. У меня есть несколько прокси-моделей, которые описывают конкретный алгоритм подсчета суммы. Базовый класс Discount имеет поле-член с именем type, представляющее собой строку, идентифицирующую его тип и связанный с ним класс.

class Discount(models.Model):
  TYPE_CHOICES = (
    ('V', 'Value'),
    ('P', 'Percentage'),
  )

  name = models.CharField(max_length=32)
  code = models.CharField(max_length=32)
  quantity = models.PositiveIntegerField()
  value = models.DecimalField(max_digits=4, decimal_places=2)
  type = models.CharField(max_length=1, choices=TYPE_CHOICES)

  def __unicode__(self):
    return self.name

  def __init__(self, *args, **kwargs):
    if self.type:
      self.__class__ = getattr(sys.modules[__name__], self.type + 'Discount')
    super(Discount, self).__init__(*args, **kwargs)

class ValueDiscount(Discount):
  class Meta:
    proxy = True

  def total(self, total):
    return total - self.value

Но я продолжаю получать исключение AttributeError, говоря, что у себя нет типа. Как это исправить или есть другой способ добиться этого?

aambrozkiewicz 23.09.2011 источник

Ответы (2)

arrow_upward
12
arrow_downward

Вместо этого ваш метод инициализации должен выглядеть так:

def __init__(self, *args, **kwargs):
    super(Discount, self).__init__(*args, **kwargs)
    if self.type:
        self.__class__ = getattr(sys.modules[__name__], self.type + 'Discount')

Вам нужно позвонить суперсуперу __init__, прежде чем вы сможете получить доступ к self.type.

Будьте осторожны с вызовом вашего поля type, так как type также является встроенной функцией Python, хотя вы можете не столкнуться с какими-либо проблемами.

См.: http://docs.python.org/library/functions.html#type

Derek Kwok 23.09.2011

comment

Спасибо. Еще один вопрос, зачем мне вызывать super init перед доступом к свойствам объектов? Я думал, что это python, который делает все члены, объявленные в объекте, доступными в конструкторе, а не в суперклассе (models.Model от Django). Почему? - aambrozkiewicz; 23.09.2011