Django模型方法与自我实例

时间:2012-11-09 22:51:49

标签: django django-models

我正在尝试将一些功能逻辑移动到模型的方法中(而不是在视图中),我认为它属于:

class Spans(models.Model):
    snow = models.IntegerField()
    wind = models.IntegerField()
    exposure = models.CharField(max_length=1)
    module_length = models.IntegerField()
    max_roof_angle = models.IntegerField()
    building_height = models.IntegerField()
    importance_factor = models.IntegerField()
    seismic = models.DecimalField(max_digits=4, decimal_places=2)
    zone = models.IntegerField()
    profile = models.CharField(max_length=55)
    tilt_angle = models.IntegerField()
    span = models.DecimalField(max_digits=18, decimal_places=15)

    def get_spans(self, snow_load, wind_speed, module_length):
        """
        Function to get spans.
        """
        spans = self.objects.filter(
            snow=snow_load,
            wind=wind_speed,
            exposure='B',
            module_length__gte=module_length,
            max_roof_angle=45,
            building_height=30,
            importance_factor=1,
            seismic=1.2,
            zone=1,
            profile='worst'
        ).order_by('snow', 'module_length', 'span')
        return spans

但是,我有点不确定如何调用它。在我试过的shell中:

>>> possible_spans = Spans.get_spans(0, 85, 54)

>>> possible_spans = Spans.get_spans(Spans, 0, 85, 54)

但是我收到了错误:

TypeError: unbound method get_spans() must be called with Spans instance as first argument (got int instance instead) 

TypeError: unbound method get_spans() must be called with Spans instance as first argument (got ModelBase instance instead)  

我知道我在这里缺少一些基本的python逻辑,但我不确定它是什么。

非常感谢任何帮助。

1 个答案:

答案 0 :(得分:4)

方法需要使用实例调用,因为方法调用将其作为默认的自身arg。例如,

instance = Spans.objects.get(id=1)
instance.get_spans(0, 85, 54)

或者如果您想要使用类本身进行调用。您必须手动传递实例

instance = Spans.objects.get(id=1)
Spans.get_spans(instance, 0, 85, 54)

更新: 您正在寻找的是使用静态方法

@staticmethod
def get_spans(snow_load, wind_speed, module_length):
        """
        Function to get spans.
        """
        spans = Spans.objects.filter(
            snow=snow_load,
            wind=wind_speed,
            exposure='B',
            module_length__gte=module_length,
            max_roof_angle=45,
            building_height=30,
            importance_factor=1,
            seismic=1.2,
            zone=1,
            profile='worst'
        ).order_by('snow', 'module_length', 'span')
        return spans

现在你可以直接做,

Spans.get_spans(0, 85, 54)