Fixed invoice logo upload

Author: MySecondLanguage

nxtbn/core/admin_mutation.py

@@ -3,8 +3,8 @@
 
 from nxtbn.core import CurrencyTypes
 from nxtbn.core.admin_permissions import gql_required_perm
-from nxtbn.core.admin_types import CurrencyExchangeType
-from nxtbn.core.models import CurrencyExchange
+from nxtbn.core.admin_types import CurrencyExchangeType, InvoiceSettingsInput, InvoiceSettingsType
+from nxtbn.core.models import CurrencyExchange, InvoiceSettings
 from nxtbn.users import UserRole
 
 class CurrencyExchangeInput(graphene.InputObjectType):
@@ -77,9 +77,33 @@ def mutate(root, info, id):
 
         currency_exchange.delete()  # Delete the record
         return DeleteCurrencyExchange(success=True)  # Return success
+    
+
+
+class CreateInvoiceSettings(graphene.Mutation):
+    class Arguments:
+        input = InvoiceSettingsInput()
+
+    invoice_settings = graphene.Field(InvoiceSettingsType)
+
+    @gql_required_perm(InvoiceSettings, 'add_invoicesettings')
+    @staticmethod
+    def mutate(root, info, input):
+        invoice_settings = InvoiceSettings.objects.create(
+            store_name=input.store_name,
+            store_address=input.store_address,
+            city=input.city,
+            country=input.country,
+            postal_code=input.postal_code,
+            contact_email=input.contact_email,
+            contact_phone=input.contact_phone,
+            is_default=input.is_default,
+        )
+        return CreateInvoiceSettings(invoice_settings=invoice_settings)
 
 
 class CoreMutation(graphene.ObjectType):
     update_exchange_rate = UpdateCurrencyExchange.Field()
     create_exchange_rate = CreateCurrencyExchange.Field()
-    delete_exchange_rate = DeleteCurrencyExchange.Field() 
+    delete_exchange_rate = DeleteCurrencyExchange.Field()
+    create_invoice_settings = CreateInvoiceSettings.Field()

nxtbn/core/admin_types.py

@@ -20,6 +20,12 @@ class Meta:
 
 class InvoiceSettingsType(DjangoObjectType):
     db_id = graphene.ID(source='id')
+    logo = graphene.String()
+
+    def resolve_logo(self, info):
+        if self.logo:
+            return info.context.build_absolute_uri(self.logo.url)
+        return None
     class Meta:
         model = InvoiceSettings
         fields = "__all__"
@@ -28,6 +34,18 @@ class Meta:
             'store_name': ['exact', 'icontains'],
         }
 
+
+class InvoiceSettingsInput(graphene.InputObjectType):
+    store_name = graphene.String()
+    store_address = graphene.String()
+    city = graphene.String()
+    country = graphene.String()
+    postal_code = graphene.String()
+    contact_email = graphene.String()
+    contact_phone = graphene.String()
+    is_default = graphene.Boolean()
+
+
 class AdminCurrencyTypesEnum(graphene.ObjectType):
     value = graphene.String()
     label = graphene.String()

nxtbn/core/api/dashboard/serializers.py

@@ -26,4 +26,10 @@ class Meta:
             'is_default',
         ]
 
-    
+    
+
+class InvoiceSettingsLogoSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = InvoiceSettings
+        fields = ['id', 'logo']
+        read_only_fields = ['id']

nxtbn/core/api/dashboard/urls.py

@@ -5,6 +5,8 @@
 urlpatterns = [
     path('site-settings/', core_views.SiteSettingsView.as_view(), name='site-settings'),
     path('invoice-settings/<int:pk>/', core_views.InvoiceSettingsView.as_view(), name='invoice-settings'),
+    path('invoice-settings/upload-logo/<int:id>/', core_views.InvoiceSettingsLogoUploadAPIView.as_view(), name='upload-logo'),
+
     path('system-status/', status_views.SystemStatusAPIView.as_view(), name='system-status'),
     path('db-tables-details/', status_views.DatabaseTableInfoAPIView.as_view(), name='db-details'),
     path('language-list/', core_views.LanguageChoicesAPIView.as_view(), name='language-list'),

nxtbn/core/api/dashboard/views.py

@@ -27,7 +27,7 @@
 
 from nxtbn.core import LanguageChoices
 from nxtbn.core.admin_permissions import GranularPermission, IsStoreAdmin, IsStoreStaff
-from nxtbn.core.api.dashboard.serializers import InvoiceSettingsSerializer, SiteSettingsSerializer
+from nxtbn.core.api.dashboard.serializers import InvoiceSettingsLogoSerializer, InvoiceSettingsSerializer, SiteSettingsSerializer
 from nxtbn.core.models import InvoiceSettings, SiteSettings
 from nxtbn.users import UserRole
 
@@ -58,6 +58,17 @@ class InvoiceSettingsView(generics.RetrieveUpdateAPIView):
 
 
 
+class InvoiceSettingsLogoUploadAPIView(generics.UpdateAPIView):
+    permission_classes = (IsStoreAdmin,)
+    queryset = InvoiceSettings.objects.all()
+    serializer_class = InvoiceSettingsLogoSerializer
+    lookup_field = 'id'
+
+    def partial_update(self, request, *args, **kwargs):
+        kwargs['partial'] = True
+        return self.update(request, *args, **kwargs)
+
+
 class LanguageChoicesAPIView(APIView):
     permission_classes = (IsStoreStaff,)
     def get(self, request, *args, **kwargs):