拨开荷叶行,寻梦已然成。仙女莲花里,翩翩白鹭情。
IMG-LOGO
主页 文章列表 我应该写哪个DjangoORM或原始SQL来获得我需要的东西

我应该写哪个DjangoORM或原始SQL来获得我需要的东西

白鹭 - 2022-02-12 2134 0 0

我正在使用 Postgresql。我有模型:

class EventsList(CreatedUpdatedMixin):
    start = models.DateTimeField()
    end = models.DateTimeField()
    is_inner = models.BooleanField()

假设我有这些数据库条目:

开始 结尾 is_inner
2021-12-09 14:30:12 2021-12-09 15:00:21 真的
2021-12-09 14:00:05 2021-12-10 21:00:15 错误的
2021-12-10 09:00:39 2021-12-10 09:30:50 真的
2021-12-10 14:00:00 2021-12-11 15:00:00 真的
2021-12-14 10:00:00 2021-12-14 11:00:00 真的
2021-12-13 13:30:00 2021-12-16 14:30:00 错误的
2021-12-14 13:10:00 2021-12-15 00:30:00 真的
2021-12-14 10:30:00 2021-12-16 13:34:00 错误的
2021-12-15 13:30:00 2021-12-15 18:30:00 真的

这是我需要的结果:

[
    {"2021-12-09": {"external_events": 1, "internal_events": 1}},
    {"2021-12-10": {"external_events": 0, "internal_events": 2}},
    {"2021-12-11": {"external_events": 0, "internal_events": 1}},
    {"2021-12-13": {"external_events": 1, "internal_events": 0}},
    {"2021-12-14": {"external_events": 2, "internal_events": 2}},
    {"2021-12-15": {"external_events": 2, "internal_events": 2}},
    {"2021-12-16": {"external_events": 2, "internal_events": 0}},
]

所以我想获取所有现有日期,并为每个日期获取外部事件的计数(其中 is_inner == False)和内部事件的计数(其中 is_inner == True)。如何使用 Django ORM 或原始 SQL 来实作?现在,我想出了

EventsList.objects.annotate(
    start_day=Cast("start", output_field=DateField())
).values("start_day").annotate(
    external_events=Count("id", filter=Q(is_inner=False)),
    internal_events=Count("id", filter=Q(is_inner=True)),
).values(
    "start_day", "external_events", "internal_events"
)

此代码回传几乎正确的结果(但仅适用于“开始”日期)。我需要包括所有日期(开始日期、结束日期以及它们之间的所有日期)。将不胜感激任何帮助。

uj5u.com热心网友回复:

使用原始 SQL,首先使用扩展日期串列,generate_series然后lateral join进行条件聚合。所以在这里,有点冗长,但我希望易于阅读。 SQL小提琴

with t as
(
 select is_inner, d::date from the_table 
 cross join lateral generate_series
 (
  date_trunc('day', "start"), 
  date_trunc('day', "end"),  
  interval '1 day'
 ) as d
)
select d as event_date, 
       count(*) filter (where not is_inner) external_events,   
       count(*) filter (where is_inner) internal_events
from t
group by d order by d; 

您也可以使用以下方式塑造问题中的 JSON 结构jsonb_build_object

with t as
(
 select is_inner, d::date from the_table 
 cross join lateral generate_series
 (
  date_trunc('day', "start"), 
  date_trunc('day', "end"),  
  interval '1 day'
 ) as d
)
select jsonb_build_object
(
  d::text, 
  jsonb_build_object('external_events', count(*) filter (where not is_inner),   
                     'internal_events', count(*) filter (where is_inner))
) as date_info   
from t
group by d order by d; 
日期信息
{“2021-12-09”:{“external_events”:1,“internal_events”:1}}
{“2021-12-10”:{“external_events”:1,“internal_events”:2}}
{“2021-12-11”:{“external_events”:0,“internal_events”:1}}
{“2021-12-13”:{“external_events”:1,“internal_events”:0}}
{“2021-12-14”:{“external_events”:2,“internal_events”:2}}
{“2021-12-15”:{“external_events”:2,“internal_events”:2}}
{“2021-12-16”:{“external_events”:2,“internal_events”:0}}

uj5u.com热心网友回复:

首先,要获取不在您的数据库中的所有日期,您将在最大和最小日期之间回圈以获取所有日期,并为每个日期运行查询以匹配您想要的过滤器。

标签:

0 评论

发表评论

您的电子邮件地址不会被公开。 必填的字段已做标记 *