Django เทียบเท่าสำหรับการนับและจัดกลุ่มตาม

ฉันมีโมเดลที่มีลักษณะดังนี้:

class Category(models.Model):
    name = models.CharField(max_length=60)

class Item(models.Model):
    name = models.CharField(max_length=60)
    category = models.ForeignKey(Category)

ฉันต้องการเลือกจำนวน (เฉพาะจำนวน) ของรายการสำหรับแต่ละหมวดหมู่ดังนั้นใน SQL จะง่ายดังนี้:

select category_id, count(id) from item group by category_id

มีวิธีเทียบเท่ากับการทำ "ทาง Django" หรือไม่? หรือ SQL ธรรมดาเป็นตัวเลือกเดียว? ฉันคุ้นเคยกับวิธีการcount ()ใน Django แต่ฉันไม่เห็นว่ากลุ่มตามจะเหมาะสมกับที่นั่นอย่างไร

ตามที่ฉันเพิ่งค้นพบคือวิธีดำเนินการกับ Django 1.1 aggregation API:

from django.db.models import Count
theanswer = Item.objects.values('category').annotate(Count('category'))

( อัปเดต : ตอนนี้การสนับสนุนการรวม ORM แบบเต็มรวมอยู่ในDjango 1.1จริงตามคำเตือนด้านล่างเกี่ยวกับการใช้ API ส่วนตัววิธีการที่บันทึกไว้ที่นี่ไม่สามารถใช้งานได้ใน Django เวอร์ชันหลัง 1.1 อีกต่อไปฉันไม่ได้ขุดเพื่อหาสาเหตุ หากคุณใช้ 1.1 หรือใหม่กว่าคุณควรใช้API การรวมจริงอยู่ดี)

การสนับสนุนการรวมหลักมีอยู่แล้วใน 1.0; มันเป็นเพียงเอกสารที่ไม่ได้รับการสนับสนุนและยังไม่มี API ที่เป็นมิตรอยู่ด้านบน แต่นี่คือวิธีที่คุณสามารถใช้งานได้จนกว่า 1.1 จะมาถึง (ยอมรับความเสี่ยงของคุณเองและด้วยความรู้ทั้งหมดว่าแอตทริบิวต์ query.group_by ไม่ได้เป็นส่วนหนึ่งของ API สาธารณะและสามารถเปลี่ยนแปลงได้):

query_set = Item.objects.extra(select={'count': 'count(1)'}, 
                               order_by=['-count']).values('count', 'category')
query_set.query.group_by = ['category_id']

หากคุณวนซ้ำใน query_set ค่าที่ส่งคืนแต่ละค่าจะเป็นพจนานุกรมที่มีคีย์ "หมวดหมู่" และคีย์ "count"

คุณไม่จำเป็นต้องสั่งซื้อโดย -count ที่นี่ซึ่งรวมไว้เพื่อแสดงให้เห็นว่ามันทำอย่างไร (ต้องทำในการเรียก. extra () ไม่ใช่ที่อื่นในห่วงโซ่การก่อสร้างแบบสอบถาม) นอกจากนี้คุณสามารถพูด count (id) แทน count (1) ได้เช่นกัน แต่ข้อหลังอาจมีประสิทธิภาพมากกว่า

โปรดทราบว่าเมื่อตั้งค่า .query.group_by ค่าต้องเป็นชื่อคอลัมน์ DB จริง ('category_id') ไม่ใช่ชื่อฟิลด์ Django ('category') เนื่องจากคุณกำลังปรับแต่งการสืบค้นภายในในระดับที่ทุกอย่างอยู่ในเงื่อนไข DB ไม่ใช่เงื่อนไขของ Django

เนื่องจากฉันสับสนเล็กน้อยเกี่ยวกับวิธีการจัดกลุ่มใน Django 1.1 ฉันคิดว่าฉันจะอธิบายรายละเอียดที่นี่ว่าคุณใช้มันอย่างไร ก่อนอื่นให้ทำซ้ำสิ่งที่ไมเคิลพูด:

ตามที่ฉันเพิ่งค้นพบคือวิธีดำเนินการกับ Django 1.1 aggregation API:

from django.db.models import Count
theanswer = Item.objects.values('category').annotate(Count('category'))

โปรดทราบว่าคุณต้องfrom django.db.models import Count!

category__countนี้จะเลือกเฉพาะหมวดหมู่และเพิ่มคำอธิบายประกอบที่เรียกว่า ทั้งนี้ขึ้นอยู่กับการสั่งซื้อเริ่มต้นนี้อาจจะเป็นสิ่งที่คุณต้องแต่ถ้าสั่งซื้อเริ่มต้นใช้สนามอื่น ๆ กว่าcategoryนี้จะไม่ทำงาน ด้วยเหตุนี้จึงมีการเลือกฟิลด์ที่จำเป็นสำหรับการสั่งซื้อและทำให้แต่ละแถวไม่ซ้ำกันดังนั้นคุณจะไม่ได้รับการจัดกลุ่มตามที่คุณต้องการ วิธีหนึ่งที่รวดเร็วในการแก้ไขปัญหานี้คือรีเซ็ตการสั่งซื้อ:

Item.objects.values('category').annotate(Count('category')).order_by()

สิ่งนี้ควรให้ผลลัพธ์ที่คุณต้องการ ในการตั้งชื่อคำอธิบายประกอบคุณสามารถใช้:

...annotate(mycount = Count('category'))...

จากนั้นคุณจะมีคำอธิบายประกอบที่เรียกmycountในผลลัพธ์

ทุกอย่างเกี่ยวกับการจัดกลุ่มนั้นตรงไปตรงมาสำหรับฉันมาก อย่าลืมตรวจสอบDjango aggregation APIสำหรับข้อมูลโดยละเอียดเพิ่มเติม

วิธีนี้เป็นอย่างไร (นอกเหนือจากช้า)

counts= [ (c, Item.filter( category=c.id ).count()) for c in Category.objects.all() ]

มีข้อดีคือสั้นแม้ว่าจะดึงข้อมูลจำนวนแถวได้มากก็ตาม

แก้ไข.

เวอร์ชันแบบสอบถามเดียว BTW มักเร็วกว่า SELECT COUNT (*) ในฐานข้อมูล ลองไปดู.

counts = defaultdict(int)
for i in Item.objects.all():
    counts[i.category] += 1