Wie schreibe ich eine Update-Postgres-Abfrage als Abfragesatz in Django 1.9?

Question

Situation

Ich habe die folgende Abfrage in Postgres 9.5.

update warehouse_shelf 
    set package_count = counter.result 
from (
    select count(id) as result, shelf_id 
    from warehouse_package 
    group by shelf_id 
) counter 
where counter.shelf_id = warehouse_shelf.id; 

Dies sind meine Tabellen.

warehouse_shelf 
+----+------+---------------+ 
| ID | NAME | PACKAGE_COUNT | 
+----+------+---------------+ 
| 1 | S1 |    3 | 
| 2 | S2 |    1 | 
| 3 | S3 |    0 | 
+----+------+---------------+ 

warehouse_package 
+----+------+---------------+ 
| ID | NAME | SHELF_ID  | 
+----+------+---------------+ 
| 1 | P1 |    1 | 
| 2 | P2 |    1 | 
| 3 | P3 |    1 | 
| 4 | P4 |    2 | 
+----+------+---------------+ 

Frage

Wie führe ich die Abfrage über, wenn ich eine Änderung an einem einzelnen Paket zu machen (zum Beispiel speichern, löschen, erstellen, aktualisieren usw.) über django-Modell?

Ich möchte, wenn möglich mit Django-Abfrage-Set ausführen und vermeiden, es als unbearbeitete Abfrage auszuführen.

Answer 1

Betrachtet man Modelle und ihre Fremdschlüsselbeziehungen:

from django.db.models import Count 

shelves = WhShelf.objects.all()  

for shelf in shelves: 
     count = WhPackage.objects.filter(shelf_id=shelf.id).aggregate(Count('shelf')) 
     shelf.update(package_count=count[shelf__count']) 

Alterantively, können Sie eine einzelne Abfrage:

WhShelf.objects.annotate(package_count=WhPackage.objects. 
       filter(shelf_id=shelf.id).aggregate(Count('shelf'))['shelf__count'])