主页 / dba / 问题 / 130397
Accepted
PythonEnthusiast
Asked: 2016-02-25 21:52:17 +0800 CST

SQL 到 DjangoORM 查询

  • 772

我有模型，

class ModelA(models.Model):
    device_id = models.IntegerField()
    deal_sent = ArrayField(models.IntegerField)

对于给定的 device_id，我们可以有一个重复的 deal_sent 列表。例如：- 考虑下表

device_id         deals_sent
    1       [4,5,1]
    2       [1,2,3,4,2,3,4,3,4,5,1]
    3       [1,4,2,3,4,3,4,5,1]
    4       [1,4,3,4,5,1]
    5       [4,5]
    6       [1,3,4,2,3,4,3,4,5,1]

我想获得 top N device_id，他们1在deal_sentArrayField 中的计数最少。例如：- 如果我想要前 2 个 device_id 的1in最少deals_sent，那么 device_id1和5将被返回，因为 in 的计数1在device_id和deals_sent中最少。15

我已经关注了这个问题，这似乎是我所期待的一个接近的解决方案。

with elements (element) as (
   select unnest(ARRAY['a','b','c','a','a'])
)
select count(*)
from elements
where element = 'a';

如何将上述查询转换为 DjangoORM 查询？或者在不在应用程序级别使用 for 循环的情况下，数据库级别本身的替代完美查询有什么用？

postgresql orm

1 个回答

  1. Best Answer

    进行查询的一种方法是：

    with norm (device_id, element) as (
    select device_id, unnest(deals_sent) from modelA
)
select device_id
from norm·
group by device_id
order by count(element = 1)
limit 2;

    Django ORM 无法生成相同的查询。你只能使用原始的。然而，仍然有一些选项可以保留 ORM 的好处。

    1）使用视图

    You can create a VIEW and an unmanaged model in your django app pointing to that view.

CREATE VIEW modelA_norm AS (
   select device_id, unnest(deals_sent) as element from modelA
);

    然后在 Django 中：

    class ModelA_norm(models.Model):
    device_id = models.IntegerField()
    element = models.IntegerField()

     class Meta:
         managed = False

    查询看起来像这样：

    ModelA_norm.objects.extra(
    select={'counter': 'COUNT(element=%s)'}, 
    select_params=(1, )
).order_by('counter')[:2]

    2）在where中使用subselect

    ModelA.objects.extra(
    where=[
        'device_id IN ( ... here comes the select to get the device ids ...)'
    ]
)

    请注意，在这种情况下，您将以任意顺序返回结果，而不必按子选择生成的 ID 顺序返回。

    3）直接运行select

    您可以将 ID 从数据库中获取到列表中，并使用它们从您的模型中进行查询。

    from django.db import connection
_c = connection.cursor() 
_c.execute('... the query above with placeholders ...', [1])
_ids = [_d[0] for _d in _c.fetchall()]
objects = modalA.objects.in_bulk(_ids)
sorted_objects = [objects[id] for id in _ids]

    我倾向于根据实际情况同时使用1）和3）。

    1）对于我可以从原始模型继承模型或者我实际上不需要原始模型的情况，我可以拥有一个新模型。

    3）当我需要原始模型时。我倾向于将查询放入应用程序 SQL 目录中的文件中（就像模板一样）。所以我有一个明确的指示，我正在做一些依赖于数据库的事情，如果需要的话，我可以稍后轻松地切换/重写它们。

    我希望这有帮助。

    • 0

相关问题