Django许多人抱怨PrimaryKey

时间:2012-02-28 03:31:56

标签: python django django-models many-to-many

我有三个相关的模型。第一个叫DayOfWeek,它是一天的标签和编号。它看起来像这样:

class DayOfWeek(models.Model):
  day = models.IntegerField()
  label = models.CharField(max_length='20')

  def __str__(self):
    return self.label

每次syncdb都会使用fixture填充此类。接下来,我有一个事件模型,它看起来像这样:

class Event(AnnouncementBase, Location):
  cost = CurrencyField(decimal_places=2, max_digits=10, blank=True, default=0.00)
  start_date = models.DateField(default = datetime.now().date())
  start_time = models.TimeField(default = datetime.now().time())
  end_date = models.DateField(blank=True, default=None, null = True)
  end_time = models.TimeField(blank=True, default=None, null = True)

最后,再次发生。它有一个事件,用于为重复事件安排事件。它看起来像这样:

class Recurrence(models.Model):

  event = models.ForeignKey(Event, related_name='event')
  repeats  = models.CharField(max_length = 50, choices = EVENT_REPEAT_CHOICES)
  repeat_every = models.IntegerField(default = 1)

  repeat_on = models.ManyToManyField(DayOfWeek, blank=True, null=True)

  repeat_by = models.CharField(max_length = 50, choices = EVENT_REPEAT_BY_CHOICES, blank=True)
  repeat_by_day_of_month = models.IntegerField(default = 0, blank=True)

  repeat_ends = models.CharField(max_length = 50, choices = EVENT_REPEAT_END_CHOICES)
  end_occurrences = models.IntegerField(default = 0, blank=True)
  repeat_end_date = models.DateField(blank=True, default=None, null = True)

  past_event_count = models.IntegerField(default=0, blank=True)
  scheduled_events = models.ManyToManyField(Event, blank=True, default=None, related_name = 'scheduled_events')
  is_active = models.BooleanField(blank=True, default=True)

   def save(self, force_insert=False, force_update=False, using=None):
      """Overridden to create events the first time."""

      self.full_clean()
      #First do  normal save so the data is there for the even scheduler.
      self.save_base(force_insert=force_insert, force_update=force_update, using=using)

      #If nothing is scheduled yet, schedule the first batch
      if self.scheduled_events.count() == 0 and self.past_event_count == 0:
          self.scheduleEvents()


  def clean(self):
      #repeat on weekly
      if self.repeat_every < 1:
          raise ValidationError('Repeat every must be at least 1.')


      #weekly
      if self.repeats == EVENT_REPEAT_CHOICES[1][0]:
        #look for missing stuff
        if not self.repeat_on:
            raise ValidationError('Missing repeat on.')

最后,我有一个单元测试,检查这是否正常,好像它看起来像这样:

def test_weekly_mon_wed_fri_occurrence(self):
    event = Event()
    event.start_date = date(year=2012, month=1, day=2)
    event.start_time = time(hour=13, minute=30)
    event.save()

    recurrence = Recurrence()
    recurrence.repeats = EVENT_REPEAT_CHOICES[1][0]
    recurrence.repeat_on = (EVENT_DAY_CHOICES[1][0], EVENT_DAY_CHOICES[3][0], EVENT_DAY_CHOICES[5][0])
    recurrence.repeat_ends = EVENT_REPEAT_END_CHOICES[0][0]
    recurrence.event = event

    nextEvent = recurrence.getNextEvent(event)

    self.assertEquals(date(year=2012, month=1, day=4), nextEvent.start_date)
    self.assertEquals(event.start_time, nextEvent.start_time)

    nextNextEvent = recurrence.getNextEvent(nextEvent)

    self.assertEquals(date(year=2012, month=1, day=6), nextNextEvent.start_date)
    self.assertEquals(event.start_time, nextNextEvent.start_time)

每当运行测试时,它都会失败,但会出现以下异常。     ValueError:'Recurrence'实例需要具有主键值才能使用多对多关系。

如果在clean方法中使用self.repeat_on,则该行发生错误。

我希望repeat_on是可选的,只有某些类型的重复需要它。我该如何工作?我错过了什么导致它失败?

1 个答案:

答案 0 :(得分:4)

在分配Many2Many关系之前,您需要调用recurrence.save()。在你的代码中你做

recurrence.repeat_on = (EVENT_DAY_CHOICES[1][0], EVENT_DAY_CHOICES[3][0], EVENT_DAY_CHOICES[5][0])

首先不保存重复。因为它没有保存,所以重复没有生成主键,并且Django ORM不知道将什么作为外键插入到M2M表中。