Merge pull request #35 from ARYAN-NIKNEZHAD/perf/api-query

Update queryset of apis
sageteamorg · Dec 5, 2024 · df49b8b · df49b8b
2 parents 4cc3284 + 778bd27
commit df49b8b
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 5 deletions.
diff --git a/sage_invoice/api/views/category.py b/sage_invoice/api/views/category.py
@@ -7,7 +7,7 @@
 
 
 class CategoryViewSet(ErrorHandlingMixin, viewsets.ModelViewSet):
-    queryset = Category.objects.all()
+    queryset = Category.objects.prefetch_related("invoices").all()
     serializer_class = CategorySerializer
     lookup_field = "slug"
 

diff --git a/sage_invoice/api/views/column.py b/sage_invoice/api/views/column.py
@@ -6,7 +6,7 @@
 
 
 class ColumnViewSet(viewsets.ModelViewSet):
-    queryset = Column.objects.all()
+    queryset = Column.objects.select_related("invoice", "item").all()
     serializer_class = ColumnSerializer
     lookup_field = "id"
     lookup_url_kwarg = "id"
diff --git a/sage_invoice/api/views/expense.py b/sage_invoice/api/views/expense.py
@@ -6,7 +6,7 @@
 
 
 class ExpenseViewSet(viewsets.ModelViewSet):
-    queryset = Expense.objects.all()
+    queryset = Expense.objects.select_related("invoice").all()
     serializer_class = ExpenseSerializer
     lookup_field = "id"
     lookup_url_kwarg = "id"
diff --git a/sage_invoice/api/views/invoice.py b/sage_invoice/api/views/invoice.py
@@ -16,7 +16,9 @@
 
 
 class InvoiceViewSet(ErrorHandlingMixin, viewsets.ModelViewSet):
-    queryset = Invoice.objects.all()
+    queryset = Invoice.objects.select_related("category", "expense", "customer").prefetch_related("category__invoices",
+        "items__invoice", "columns__item"
+    ).all()
     serializer_class = InvoiceSerializer
     lookup_field = "slug"
     versioning_class = HeaderVersioning

diff --git a/sage_invoice/api/views/item.py b/sage_invoice/api/views/item.py
@@ -6,7 +6,7 @@
 
 
 class ItemViewSet(viewsets.ModelViewSet):
-    queryset = Item.objects.all()
+    queryset = Item.objects.select_related("invoice").all()
     serializer_class = ItemSerializer
     lookup_field = "id"
     lookup_url_kwarg = "id"