在 Django ORM 中按查询分组

Question

我对如何编写 Django 查询来获取我的数据感到困惑。我有 2 tables 'ticket' 和 'ticket_details'。以下是它们的架构。

Ticket(id, name, type, user)
TicketDetails(ticket_id, message, created_time)

注意：一个工单id可以关联多条消息。

而ticket_id是Ticket的外键table.

我想从 table 中获取所有列，其中只有来自 TicketDetails table 的最新消息应该为特定的工单 ID 选择。

Example:
Ticket
id, name, type, user
1,install, application, usr1

TicketDetails
ticket_id, message, creted_time
1, <message1>, 12:00 PM
1, <message2>, 04:00 PM
2, <message3>, 05:00 PM -->latest entry

Expected Output:
id, name, type, user, message, created_time
1, install, application, usr1, <message3>, 05:00PM

提前致谢

Answer 1

我对你的模型做了一些假设，你没有提供：

class Ticket(models.Model):
    name = models.CharField(max_length=50)
    type = models.CharField(max_length=50)
    user = models.ForeignKey('auth.User', on_delete=models.CASCADE)


# Model names should NEVER end with "s"
class TicketDetail(models.Model):
    ticket = models.ForeignKey(Ticket, on_delete=models.CASCADE)
    message = models.CharField(max_length=50)
    created_time = models.DateTimeField(auto_now_add=True)

您有 2 个选择：

你可以纯写sql，你就失去了过滤的能力

sql = """
SELECT ticket.id, ticket.name, ticket.type, ticket.user_id, detail.message
  FROM {ticket} ticket
  LEFT JOIN (
    SELECT detail.ticket_id, detail.message 
      FROM {detail} detail
     INNER JOIN (
        SELECT MAX(id) id, ticket_id 
          FROM {detail} 
         GROUP BY ticket_id
    ) detail_message ON detail.id = detail_message.id
) detail ON detail.ticket_id = ticket.id
""".format(ticket=Ticket._meta.db_table, detail=TicketDetail._meta.db_table)
tickets = Ticket.objects.raw(sql)

for ticket in tickets:
    print(ticket.id, ticket.message)

用“django”的方式写

latest_messages = TicketDetail.objects.values('ticket_id').annotate(id=models.Max('id')).values('id')
tickets = Ticket.objects.prefetch_related(models.Prefetch('ticketdetail_set', TicketDetail.objects.filter(id__in=latest_messages))).order_by('id')

for ticket in tickets:
    print(ticket.id)
    # this iteration will only ever yield 1 result.. or nothing.
    for detail in ticket.ticketdetail_set.all():
        print(detail.message)

测试如下：

 # uses factoryboy and faker to fill in the data

 class UserFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = auth.models.User
        django_get_or_create = ('username',)

    first_name = fake.first_name()
    last_name = fake.last_name()
    email = factory.LazyAttribute(lambda obj: "{}.{}@gmail.com".format(obj.last_name, obj.first_name).lower())
    username = factory.Sequence(lambda n: 'user' + str(n))


class SimpleTestCase(TestCase):
    
    def setUp(self):
        ticket1 = Ticket.objects.create(user=UserFactory(), type='A', name='Number 1')
        TicketDetail.objects.create(ticket=ticket1, message='you wont see this')
        TicketDetail.objects.create(ticket=ticket1, message='you wont see this either')
        TicketDetail.objects.create(ticket=ticket1, message='YES!!')

        ticket2 = Ticket.objects.create(user=UserFactory(), type='B', name='Number 2')
        TicketDetail.objects.create(ticket=ticket2, message='you also wont see this')
        TicketDetail.objects.create(ticket=ticket2, message='you also wont see this either')
        TicketDetail.objects.create(ticket=ticket2, message='also YES!!')

    def test_flatten_pure_sql(self):
        sql = """
        SELECT ticket.id, ticket.name, ticket.type, ticket.user_id, detail.message
          FROM {ticket} ticket
          LEFT JOIN (
            SELECT detail.ticket_id, detail.message 
              FROM {detail} detail
             INNER JOIN (
                SELECT MAX(id) id, ticket_id 
                  FROM {detail} 
                 GROUP BY ticket_id
            ) detail_message ON detail.id = detail_message.id
        ) detail ON detail.ticket_id = ticket.id
        """.format(ticket=Ticket._meta.db_table, detail=TicketDetail._meta.db_table)
        self.assertEquals(['YES!!', 'also YES!!'], [x.message for x in Ticket.objects.raw(sql)])

    def test_orm_way(self):
        latest_messages = TicketDetail.objects.values('ticket_id').annotate(id=models.Max('id')).values('id')
        tickets = Ticket.objects.prefetch_related(models.Prefetch('ticketdetail_set', TicketDetail.objects.filter(id__in=latest_messages))).order_by('id')
        self.assertEquals(['Number 1', 'Number 2'], [x.name for x in tickets])
        self.assertEquals(['YES!!'], [x.message for x in tickets[0].ticketdetail_set.all()])
        self.assertEquals(['also YES!!'], [x.message for x in tickets[1].ticketdetail_set.all()])

在 Django ORM 中按查询分组

Group by query in Django ORM

python

django

orm