Merge pull request #386 from MySecondLanguage/invoice-query-optimized

Invoice query optimized

Author: MySecondLanguage

nxtbn/core/admin_queries.py

@@ -4,15 +4,16 @@
 
 from nxtbn.core import CurrencyTypes
 from nxtbn.core.admin_permissions import gql_store_admin_required
-from nxtbn.core.admin_types import AdminCurrencyTypesEnum, CurrencyExchangeType
-from nxtbn.core.models import CurrencyExchange
+from nxtbn.core.admin_types import AdminCurrencyTypesEnum, CurrencyExchangeType, InvoiceSettingsType
+from nxtbn.core.models import CurrencyExchange, InvoiceSettings
 from graphene_django.filter import DjangoFilterConnectionField
 
 
 class AdminCoreQuery(graphene.ObjectType):
     currency_exchanges = DjangoFilterConnectionField(CurrencyExchangeType)
     currency_exchange = graphene.Field(CurrencyExchangeType, id=graphene.ID(required=True))
     allowed_currency_list = graphene.List(AdminCurrencyTypesEnum)
+    default_invoice_settings = graphene.Field(InvoiceSettingsType)
 
     @gql_store_admin_required
     def resolve_currency_exchanges(self, info, **kwargs):
@@ -36,3 +37,8 @@ def resolve_allowed_currency_list(self, info):
             for currency in CurrencyTypes.choices
             if currency[0] in allowed_currency_list
         ]
+    
+    @gql_store_admin_required
+    def resolve_default_invoice_settings(self, info):
+        site_id = getattr(settings, 'SITE_ID', 1)
+        return InvoiceSettings.objects.get(site__id=site_id)

nxtbn/core/admin_types.py

@@ -3,7 +3,7 @@
 from graphene_django.types import DjangoObjectType
 
 from nxtbn.core import CurrencyTypes
-from .models import CurrencyExchange
+from nxtbn.core.models import CurrencyExchange, InvoiceSettings, SiteSettings
 
 class CurrencyExchangeType(DjangoObjectType):
     db_id = graphene.ID(source='id')
@@ -17,6 +17,13 @@ class Meta:
             'exchange_rate': ['exact', 'icontains'],
         }
 
+
+class InvoiceSettingsType(DjangoObjectType):
+    db_id = graphene.ID(source='id')
+    class Meta:
+        model = InvoiceSettings
+        fields = "__all__"
+
 class AdminCurrencyTypesEnum(graphene.ObjectType):
     value = graphene.String()
     label = graphene.String()

nxtbn/order/admin_queries.py

@@ -1,15 +1,19 @@
+from django.conf import settings
 import graphene
 from graphene_django.filter import DjangoFilterConnectionField
 
 from nxtbn.core.admin_permissions import gql_store_admin_required
-from nxtbn.order.admin_types import OrderType
+from nxtbn.core.models import SiteSettings
+from nxtbn.order.admin_types import OrderInvoiceType, OrderType
 from nxtbn.order.models import Address, Order
 from nxtbn.users import UserRole
 
 
 class AdminOrderQuery(graphene.ObjectType):
     orders = DjangoFilterConnectionField(OrderType)
     order = graphene.Field(OrderType, id=graphene.Int(required=True))
+    order_invoice = graphene.Field(OrderInvoiceType, order_id=graphene.Int(required=True))
+    order_invoices = DjangoFilterConnectionField(OrderInvoiceType)
 
 
 
@@ -24,4 +28,25 @@ def resolve_order(self, info, id):
         except Order.DoesNotExist:
             raise Exception("Order not found")
 
+        return order
+    
+   
+    @gql_store_admin_required
+    def resolve_order_invoice(self, info, order_id):
+        order = Order.objects.select_related(
+            'billing_address', 'shipping_address'
+        ).prefetch_related(
+            'line_items__variant'
+        ).get(id=order_id)
+
+        return order
+    
+    @gql_store_admin_required
+    def resolve_order_invoices(self, info, **kwargs):
+        order = Order.objects.select_related(
+            'billing_address', 'shipping_address'
+        ).prefetch_related(
+            'line_items__variant'
+        ).all()
+
         return order

nxtbn/order/admin_types.py

@@ -1,7 +1,9 @@
+from django.conf import settings
 import graphene
 from graphene_django import DjangoObjectType
 from graphene import relay
-from nxtbn.order.models import Address, Order
+from nxtbn.core.models import SiteSettings
+from nxtbn.order.models import Address, Order, OrderLineItem
 
 from nxtbn.order.admin_filters import OrderFilter
 
@@ -58,4 +60,65 @@ class Meta:
             'comment',
         )
         interfaces = (relay.Node,)
+        filterset_class = OrderFilter
+
+
+
+class OrderInvoiceLineItemType(DjangoObjectType):
+    total_price = graphene.String()
+    price_per_unit = graphene.String()
+    name = graphene.String()
+
+    class Meta:
+        model = OrderLineItem
+        fields = ['id', 'quantity']
+
+    def resolve_total_price(self, info):
+        return self.humanize_total_price()
+
+    def resolve_price_per_unit(self, info):
+        return self.humanize_price_per_unit()
+
+    def resolve_name(self, info):
+        return self.variant.get_descriptive_name_minimal()
+
+class OrderInvoiceType(DjangoObjectType):
+    db_id = graphene.Int(source='id')
+    humanize_total_price = graphene.String()
+    items = graphene.List(OrderInvoiceLineItemType)
+    total_price = graphene.String()
+
+    def resolve_humanize_total_price(self, info):
+        return self.humanize_total_price()
+    
+    def resolve_items(self, info):
+        return self.line_items.all()
+
+    def resolve_total_price(self, info):
+        return self.humanize_total_price()
+    
+    class Meta:
+        model = Order
+        fields = (
+            'alias',
+            'id',
+            'status',
+            'shipping_address',
+            'billing_address',
+            'created_at',
+            'total_price_without_tax',
+            'total_shipping_cost',
+            'total_discounted_amount',
+            'total_tax',
+            'customer_currency',
+            'currency_conversion_rate',
+            'authorize_status',
+            'charge_status',
+            'payment_term',
+            'due_date',
+            'preferred_payment_method',
+            'reservation_status',
+            'note',
+        )
+        interfaces = (relay.Node,)
         filterset_class = OrderFilter