如何为外键中的特定字段指定默认外部字段值?

时间:2012-07-11 05:00:17

标签: django django-models

我希望ForeignKey的每个实例具有不同的默认值,具体取决于它在父模型中的属性。

插图可能会更清楚:

class Day(models.Model):
    date = models.DateField(unique=True)
    manager_shift = models.ForeignKey('Shift') # Shift.shift_type should be "M"
    lead_shift = models.ForeignKey('Shift') # Shift.shift_type should be "L"
    early_day_shift = models.ForeignKey('Shift') # Shift.shift_type should be "ED"
    late_day_shift = models.ForeignKey('Shift') # Shift.shift_type should be "LD"
    swing_shift = models.ForeignKey('Shift') # Shift.shift_type should be "SW"
    early_evening_shift = models.ForeignKey('Shift') # Shift.shift_type should be "EM"
    late_evening_shift = models.ForeignKey('Shift') # Shift.shift_type should be "LM"

class Shift(models.Model):
    user = models.ForeignKey('User', limit_choices_to={'is_staff': True}, blank=False)
    shift_type = models.CharField(max_length=2, choices=SHIFT_TYPE_CHOICES, blank=False) 

SHIFT_TYPE_CHOICES = { 
    ('M', 'Manager'),
    ('L', 'Lead'),
    ('ED', 'Early Day'),
    ('LD', 'Early Day'),
    ('SW', 'Swing'),
    ('EM', 'Early Evening'),
    ('LM', 'Late Evening'),
}

在这种情况下,

所以在做的时候:

Day.manager_shift = Shift() 

Day.manager_shift.shift_type应自动为“M”

3 个答案:

答案 0 :(得分:1)

我不相信有任何内置的方式,不应该有。如果你这样做会发生什么:

>>> my_shift = Shift()
>>> Day.manager_shift = my_shift
>>> Day.lead_shift = my_shift
>>> my_shift.shift_type
???

更好的选择是覆盖save的{​​{1}}方法来检查已保存对象的每个字段,并且:a)如果没有Day,则添加正确的字段; b)如果有一个转换时间,但是错误的那个,引发错误(不是最干净的解决方案,但我不知道更好的事情)。

更新:我只是read编辑:和Rohan也指出)您可以给出默认值shift_type,但您应该以函数的形式执行此操作:

ForeignKey

答案 1 :(得分:1)

您可以将默认值设置为可调用,因此您可以执行此操作:

def get_m_shift():
    return Shift.objects.get(shitf_type='M')

    class Day(models.Model):
        date = models.DateField(unique=True)
        manager_shift = models.ForeignKey('Shift', default=get_m_shift)

我认为你必须编写函数来获得每种班次类型。我不确定默认的callable是否参数。

答案 2 :(得分:1)

为什么你有一个单独的'Day'模型?

SHIFT_TYPE_CHOICES = { 
    ('M', 'Manager'),
    ('L', 'Lead'),
    ('ED', 'Early Day'),
    ('LD', 'Early Day'),
    ('SW', 'Swing'),
    ('EM', 'Early Evening'),
    ('LM', 'Late Evening'),
}

class Shift(models.Model):
    user = models.ForeignKey('User',
                             limit_choices_to={'is_staff': True},
                             blank=False)
    shift_type = models.CharField(max_length=2,
                                  choices=SHIFT_TYPE_CHOICES,
                                  blank=False) 
    shift_date = models.DateField()

    class Meta:
        unique_together = (('shift_date','shift_user','shift_type'),)

这将防止同一用户在同一日期重复移位,并且只能移位类型和用户的一个组合。您可以根据需要进行调整。因此,如果您需要在同一天进行多次轮班,请移除shift_datedocumentation for unique_together解释了这是如何运作的。

现在,要了解特定日期的经理是谁:

for s in Shift.objects.filter(shift_type='M',
                              shift_date=datetime.datetime.today()):
    print s.user

或者打印整个名单的日子:

for s in Shift.objects.filter(shift_date=datetime.datetime.today()):
    print "%s: %s\n" % (s.get_shift_type_display(),s.user)