Django ORM 相当于此 SQL..从相关表派生的计算字段

Question

我的模型结构如下：

class Master(models.Model): 
    name = models.CharField(max_length=50)          
    mounting_height = models.DecimalField(max_digits=10,decimal_places=2)   

class MLog(models.Model):                  
    date = models.DateField(db_index=True)
    time = models.TimeField(db_index=True)      
    sensor_reading = models.IntegerField()      
    m_master = models.ForeignKey(Master)

目标是生成一个查询集，该查询集返回 MLog 中的所有字段以及基于 Master 中的相关数据的计算字段 (item_height)，

使用 Django 的原始 sql：

querySet = MLog.objects.raw('''
    SELECT a.id,
           date,
           time,
           sensor_reading,
           mounting_height,
          (sensor_reading - mounting_height) as item_height 
    FROM db_mlog a JOIN db_master b 
                     ON a.m_master_id = b.id
''')                                

如何使用 Django 的 ORM 对其进行编码？

Answer 1

我可以想到两种不依赖 raw() 的方法来解决这个问题。第一个与 @tylerl 建议。像这样的事情：

class Master(models.Model):
    name = models.CharField(max_length=50)
    mounting_height = models.DecimalField(max_digits=10,decimal_places=2)

class MLog(models.Model):
    date = models.DateField(db_index=True)
    time = models.TimeField(db_index=True)
    sensor_reading = models.IntegerField()
    m_master = models.ForeignKey(Master)

    def _get_item_height(self):
        return self.sensor_reading - self.m_master.mounting_height
    item_height = property(_get_item_height)

在本例中，我为 MLog 定义一个名为 item_height 的自定义（派生）属性。此属性的计算方式为实例的 sensor_reading 与其相关主实例的 Mounting_height 之差。有关属性的更多信息此处。

然后，您可以执行以下操作：

In [4]: q = MLog.objects.all()

In [5]: q[0]
Out[5]: <MLog: 2010-09-11 8>

In [6]: q[0].item_height
Out[6]: Decimal('-2.00')

第二种方法是使用 extra() 方法并拥有数据库为你计算一下。

In [14]: q = MLog.objects.select_related().extra(select = 
          {'item_height': 'sensor_reading - mounting_height'})

In [16]: q[0]
Out[16]: <MLog: 2010-09-11 8>

In [17]: q[0].item_height
Out[17]: Decimal('-2.00')

您会注意到 select_lated() 的使用。如果没有这个，Master 表将不会与查询连接，并且您将收到错误。

Answer 2

我总是在应用程序中而不是在数据库中进行计算。

class Thing(models.Model):
    foo = models.IntegerField()
    bar = models.IntegerField()     
    @Property
    def diff():
        def fget(self):
            return self.foo - self.bar
        def fset(self,value):
            self.bar = self.foo - value

然后，您可以像操作任何其他字段一样操作它，并且它会执行您使用基础数据定义的任何操作。例如：

obj = Thing.objects.all()[0]
print(obj.diff)  # prints .foo - .bar
obj.diff = 4     # sets .bar to .foo - 4

顺便说一句，Property只是一个标准的属性装饰器，在本例中编码如下（我不记得它来自哪里）：

def Property(function):
    keys = 'fget', 'fset', 'fdel'
    func_locals = {'doc':function.__doc__}
    def probeFunc(frame, event, arg):
        if event == 'return':
            locals = frame.f_locals
            func_locals.update(dict((k,locals.get(k)) for k in keys))
            sys.settrace(None)
        return probeFunc
    sys.settrace(probeFunc)
    function()
    return property(**func_locals)