karrio-server-graph 2026.1__py3-none-any.whl → 2026.1.3__py3-none-any.whl

karrio/server/graph/schemas/base/mutations.py

@@ -8,7 +8,7 @@ from django.contrib.contenttypes.models import ContentType
 from django_email_verification import confirm as email_verification
 from django_otp.plugins.otp_email import models as otp
 from django.utils.translation import gettext_lazy as _
-from django.db import transaction
+from django.db import transaction, models
 
 from karrio.server.core.utils import ConfirmationToken, send_email
 from karrio.server.user.serializers import TokenSerializer
@@ -490,7 +490,12 @@ class DeleteMutation(utils.BaseMutation):
         if validator:
             validator(instance, context=info.context)
 
+        # Get the shipment FK if it exists and is a direct relationship
+        # (not a GenericRelation/RelatedManager)
         shipment = getattr(instance, "shipment", None)
+        if shipment is not None and not isinstance(shipment, manager.Shipment):
+            shipment = None
+
         instance.delete(keep_parents=True)
 
         if shipment is not None:
@@ -514,6 +519,7 @@ class CreateRateSheetMutation(utils.BaseMutation):
         zones_data = data.pop("zones", [])
         surcharges_data = data.pop("surcharges", [])
         service_rates_data = data.pop("service_rates", [])
+        # Note: origin_countries stays in data - saved via serializer
         services_data = [
             (svc.copy() if isinstance(svc, dict) else dict(svc))
             for svc in data.get("services", [])
@@ -542,7 +548,7 @@ class CreateRateSheetMutation(utils.BaseMutation):
 
         # Link carriers
         if any(carriers):
-            providers.Carrier.access_by(info.context.request).filter(
+            providers.CarrierConnection.access_by(info.context.request).filter(
                 carrier_code=rate_sheet.carrier_name,
                 id__in=carriers,
             ).update(rate_sheet=rate_sheet)
@@ -573,6 +579,7 @@ class UpdateRateSheetMutation(utils.BaseMutation):
         )
         data = input.copy()
         carriers = data.pop("carriers", [])
+        # Note: origin_countries stays in data - saved via serializer
 
         serializer = serializers.RateSheetModelSerializer(
             instance,
@@ -595,7 +602,7 @@ class UpdateRateSheetMutation(utils.BaseMutation):
 
         # Link/unlink carriers
         if any(carriers):
-            carrier_qs = providers.Carrier.access_by(info.context.request).filter(
+            carrier_qs = providers.CarrierConnection.access_by(info.context.request).filter(
                 carrier_code=rate_sheet.carrier_name
             )
             carrier_qs.filter(id__in=carriers).update(rate_sheet=rate_sheet)
@@ -987,72 +994,348 @@ class ChangeShipmentStatusMutation(utils.BaseMutation):
         return ChangeShipmentStatusMutation(shipment=shipment)  # type:ignore
 
 
-def create_template_mutation(name: str, template_type: str) -> typing.Type:
-    _type: typing.Any = dict(
-        address=types.AddressTemplateType,
-        customs=types.CustomsTemplateType,
-        parcel=types.ParcelTemplateType,
-    ).get(template_type)
-
-    @strawberry.type
-    class _Mutation(utils.BaseMutation):
-        template: typing.Optional[_type] = None
-
-        @staticmethod
-        @utils.authentication_required
-        @transaction.atomic
-        def mutate(info: Info, **input) -> name:  # type:ignore
-            data = input.copy()
-            instance = (
-                graph.Template.access_by(info.context.request).get(id=input["id"])
-                if "id" in input
-                else None
+def _clear_default_address_templates(info, exclude_id=None):
+    """Clear is_default flag on all address templates except the specified one."""
+    queryset = manager.Address.access_by(info.context.request).filter(
+        meta__is_default=True,
+        meta__label__isnull=False,
+    )
+    if exclude_id:
+        queryset = queryset.exclude(id=exclude_id)
+
+    for address in queryset:
+        address.meta = {**(address.meta or {}), "is_default": False}
+        address.save(update_fields=["meta"])
+
+
+def _clear_default_parcel_templates(info, exclude_id=None):
+    """Clear is_default flag on all parcel templates except the specified one."""
+    queryset = manager.Parcel.access_by(info.context.request).filter(
+        meta__is_default=True,
+        meta__label__isnull=False,
+    )
+    if exclude_id:
+        queryset = queryset.exclude(id=exclude_id)
+
+    for parcel in queryset:
+        parcel.meta = {**(parcel.meta or {}), "is_default": False}
+        parcel.save(update_fields=["meta"])
+
+
+@strawberry.type
+class CreateAddressMutation(utils.BaseMutation):
+    """Create a saved address using Address model with meta.label."""
+
+    address: typing.Optional[types.AddressTemplateType] = None
+
+    @staticmethod
+    @utils.authentication_required
+    @transaction.atomic
+    def mutate(info: Info, **input) -> "CreateAddressMutation":
+        data = input.copy()
+
+        # Extract meta from input (flat structure)
+        meta_input = data.pop("meta", {})
+        meta = {
+            k: v
+            for k, v in meta_input.items()
+            if not utils.is_unset(v)
+        }
+
+        # If setting as default, clear existing default
+        if meta.get("is_default"):
+            _clear_default_address_templates(info)
+
+        # Ensure is_default has a value
+        if "is_default" not in meta:
+            meta["is_default"] = False
+
+        serializer = serializers.AddressModelSerializer(
+            data={**data, "meta": meta},
+            context=info.context.request,
+        )
+        serializer.is_valid(raise_exception=True)
+        address = serializer.save()
+
+        return CreateAddressMutation(address=address)  # type:ignore
+
+
+@strawberry.type
+class UpdateAddressMutation(utils.BaseMutation):
+    """Update a saved address."""
+
+    address: typing.Optional[types.AddressTemplateType] = None
+
+    @staticmethod
+    @utils.authentication_required
+    @transaction.atomic
+    def mutate(info: Info, **input) -> "UpdateAddressMutation":
+        data = input.copy()
+        id = data.pop("id")
+
+        # Extract meta from input (flat structure)
+        meta_input = data.pop("meta", {})
+        meta_updates = {
+            k: v
+            for k, v in meta_input.items()
+            if not utils.is_unset(v)
+        }
+
+        instance = manager.Address.access_by(info.context.request).get(id=id)
+
+        # Verify this is a template (has meta.label)
+        if not (instance.meta or {}).get("label"):
+            raise utils.ValidationError(
+                "This address is not a template. Only templates can be updated here."
             )
-            customs_data = data.get("customs", {})
-
-            if "commodities" in customs_data and instance is not None:
-                save_many_to_many_data(
-                    "commodities",
-                    serializers.CommodityModelSerializer,
-                    getattr(instance, "customs", None),
-                    payload=customs_data,
-                    context=info.context.request,
-                )
 
-            serializer = serializers.TemplateModelSerializer(
-                instance,
-                data=data,
-                context=info.context.request,
-                partial=(instance is not None),
+        # Update meta field
+        meta = {**(instance.meta or {}), **meta_updates}
+
+        # If setting as default, clear existing default
+        if meta_updates.get("is_default"):
+            _clear_default_address_templates(info, exclude_id=id)
+
+        serializer = serializers.AddressModelSerializer(
+            instance,
+            data={**data, "meta": meta},
+            context=info.context.request,
+            partial=True,
+        )
+        serializer.is_valid(raise_exception=True)
+        address = serializer.save()
+
+        return UpdateAddressMutation(address=address)  # type:ignore
+
+
+@strawberry.type
+class CreateParcelMutation(utils.BaseMutation):
+    """Create a saved parcel using Parcel model with meta.label."""
+
+    parcel: typing.Optional[types.ParcelTemplateType] = None
+
+    @staticmethod
+    @utils.authentication_required
+    @transaction.atomic
+    def mutate(info: Info, **input) -> "CreateParcelMutation":
+        data = input.copy()
+
+        # Extract meta from input (flat structure)
+        meta_input = data.pop("meta", {})
+        meta = {
+            k: v
+            for k, v in meta_input.items()
+            if not utils.is_unset(v)
+        }
+
+        # If setting as default, clear existing default
+        if meta.get("is_default"):
+            _clear_default_parcel_templates(info)
+
+        # Ensure is_default has a value
+        if "is_default" not in meta:
+            meta["is_default"] = False
+
+        serializer = serializers.ParcelModelSerializer(
+            data={**data, "meta": meta},
+            context=info.context.request,
+        )
+        serializer.is_valid(raise_exception=True)
+        parcel = serializer.save()
+
+        return CreateParcelMutation(parcel=parcel)  # type:ignore
+
+
+@strawberry.type
+class UpdateParcelMutation(utils.BaseMutation):
+    """Update a saved parcel."""
+
+    parcel: typing.Optional[types.ParcelTemplateType] = None
+
+    @staticmethod
+    @utils.authentication_required
+    @transaction.atomic
+    def mutate(info: Info, **input) -> "UpdateParcelMutation":
+        data = input.copy()
+        id = data.pop("id")
+
+        # Extract meta from input (flat structure)
+        meta_input = data.pop("meta", {})
+        meta_updates = {
+            k: v
+            for k, v in meta_input.items()
+            if not utils.is_unset(v)
+        }
+
+        instance = manager.Parcel.access_by(info.context.request).get(id=id)
+
+        # Verify this is a template (has meta.label)
+        if not (instance.meta or {}).get("label"):
+            raise utils.ValidationError(
+                "This parcel is not a template. Only templates can be updated here."
             )
 
-            serializer.is_valid(raise_exception=True)
-            template = serializer.save()
+        # Update meta field
+        meta = {**(instance.meta or {}), **meta_updates}
 
-            return _Mutation(template=template)  # type:ignore
+        # If setting as default, clear existing default
+        if meta_updates.get("is_default"):
+            _clear_default_parcel_templates(info, exclude_id=id)
 
-    return strawberry.type(type(name, (_Mutation,), {}))
+        serializer = serializers.ParcelModelSerializer(
+            instance,
+            data={**data, "meta": meta},
+            context=info.context.request,
+            partial=True,
+        )
+        serializer.is_valid(raise_exception=True)
+        parcel = serializer.save()
 
+        return UpdateParcelMutation(parcel=parcel)  # type:ignore
 
-CreateAddressTemplateMutation = create_template_mutation(
-    "CreateAddressTemplateMutation", "address"
-)
-UpdateAddressTemplateMutation = create_template_mutation(
-    "UpdateAddressTemplateMutation",
-    "address",
-)
-CreateCustomsTemplateMutation = create_template_mutation(
-    "CreateCustomsTemplateMutation", "customs"
-)
-UpdateCustomsTemplateMutation = create_template_mutation(
-    "UpdateCustomsTemplateMutation", "customs"
-)
-CreateParcelTemplateMutation = create_template_mutation(
-    "CreateParcelTemplateMutation", "parcel"
-)
-UpdateParcelTemplateMutation = create_template_mutation(
-    "UpdateParcelTemplateMutation", "parcel"
-)
+
+@strawberry.type
+class DeleteAddressMutation(utils.BaseMutation):
+    """Delete a saved address."""
+
+    id: str = strawberry.UNSET
+
+    @staticmethod
+    @utils.authentication_required
+    def mutate(info: Info, **input) -> "DeleteAddressMutation":
+        id = input.get("id")
+        instance = manager.Address.access_by(info.context.request).get(id=id)
+
+        # Verify this is a saved address (has meta.label)
+        if not (instance.meta or {}).get("label"):
+            raise utils.ValidationError(
+                "This address is not a saved address. Only saved addresses can be deleted here."
+            )
+
+        instance.delete(keep_parents=True)
+        return DeleteAddressMutation(id=id)  # type:ignore
+
+
+@strawberry.type
+class DeleteParcelMutation(utils.BaseMutation):
+    """Delete a saved parcel."""
+
+    id: str = strawberry.UNSET
+
+    @staticmethod
+    @utils.authentication_required
+    def mutate(info: Info, **input) -> "DeleteParcelMutation":
+        id = input.get("id")
+        instance = manager.Parcel.access_by(info.context.request).get(id=id)
+
+        # Verify this is a saved parcel (has meta.label)
+        if not (instance.meta or {}).get("label"):
+            raise utils.ValidationError(
+                "This parcel is not a saved parcel. Only saved parcels can be deleted here."
+            )
+
+        instance.delete(keep_parents=True)
+        return DeleteParcelMutation(id=id)  # type:ignore
+
+
+def _clear_default_product_templates(info, exclude_id=None):
+    """Clear is_default flag on all product templates except the specified one."""
+    queryset = manager.Commodity.access_by(info.context.request).filter(
+        meta__is_default=True,
+        meta__label__isnull=False,
+    )
+    if exclude_id:
+        queryset = queryset.exclude(id=exclude_id)
+
+    for commodity in queryset:
+        commodity.meta = {**(commodity.meta or {}), "is_default": False}
+        commodity.save(update_fields=["meta"])
+
+
+@strawberry.type
+class CreateProductMutation(utils.BaseMutation):
+    """Create a saved product using Commodity model with meta.label."""
+
+    product: typing.Optional[types.ProductTemplateType] = None
+
+    @staticmethod
+    @utils.authentication_required
+    @transaction.atomic
+    def mutate(info: Info, **input) -> "CreateProductMutation":
+        data = input.copy()
+
+        # Extract meta from input (flat structure)
+        meta_input = data.pop("meta", {})
+        meta = {
+            k: v
+            for k, v in meta_input.items()
+            if not utils.is_unset(v)
+        }
+
+        # If setting as default, clear existing default
+        if meta.get("is_default"):
+            _clear_default_product_templates(info)
+
+        # Ensure is_default has a value
+        if "is_default" not in meta:
+            meta["is_default"] = False
+
+        serializer = serializers.CommodityModelSerializer(
+            data={**data, "meta": meta},
+            context=info.context.request,
+        )
+        serializer.is_valid(raise_exception=True)
+        product = serializer.save()
+
+        return CreateProductMutation(product=product)  # type:ignore
+
+
+@strawberry.type
+class UpdateProductMutation(utils.BaseMutation):
+    """Update a saved product."""
+
+    product: typing.Optional[types.ProductTemplateType] = None
+
+    @staticmethod
+    @utils.authentication_required
+    @transaction.atomic
+    def mutate(info: Info, **input) -> "UpdateProductMutation":
+        data = input.copy()
+        id = data.pop("id")
+
+        # Extract meta from input (flat structure)
+        meta_input = data.pop("meta", {})
+        meta_updates = {
+            k: v
+            for k, v in meta_input.items()
+            if not utils.is_unset(v)
+        }
+
+        instance = manager.Commodity.access_by(info.context.request).get(id=id)
+
+        # Verify this is a template (has meta.label)
+        if not instance.is_template:
+            raise utils.ValidationError(
+                "This commodity is not a template. Only templates can be updated here."
+            )
+
+        # Update meta field
+        meta = {**(instance.meta or {}), **meta_updates}
+
+        # If setting as default, clear existing default
+        if meta_updates.get("is_default"):
+            _clear_default_product_templates(info, exclude_id=id)
+
+        serializer = serializers.CommodityModelSerializer(
+            instance,
+            data={**data, "meta": meta},
+            context=info.context.request,
+            partial=True,
+        )
+        serializer.is_valid(raise_exception=True)
+        product = serializer.save()
+
+        return UpdateProductMutation(product=product)  # type:ignore
 
 
 @strawberry.type
@@ -1089,7 +1372,7 @@ class UpdateCarrierConnectionMutation(utils.BaseMutation):
     def mutate(info: Info, **input) -> "UpdateCarrierConnectionMutation":
         data = input.copy()
         id = data.get("id")
-        instance = providers.Carrier.access_by(info.context.request).get(id=id)
+        instance = providers.CarrierConnection.access_by(info.context.request).get(id=id)
         connection = lib.identity(
             providers_serializers.CarrierConnectionModelSerializer.map(
                 instance,
@@ -1115,37 +1398,35 @@ class SystemCarrierMutation(utils.BaseMutation):
     def mutate(
         info: Info, **input: inputs.SystemCarrierMutationInput
     ) -> "SystemCarrierMutation":
+        from karrio.server.providers.serializers import BrokeredConnectionModelSerializer
+
         pk = input.get("id")
         context = info.context.request
-        carrier = providers.Carrier.system_carriers.get(pk=pk)
+        system_connection = providers.SystemConnection.objects.get(pk=pk)
+
+        # Build serializer data from input
+        # Map 'config' to 'config_overrides' and 'enable' to 'is_enabled'
+        data = {"system_connection_id": pk}
 
         if "enable" in input:
-            if input.get("enable"):
-                if hasattr(carrier, "active_orgs"):
-                    carrier.active_orgs.add(info.context.request.org)
-                else:
-                    carrier.active_users.add(info.context.request.user)
-            else:
-                if hasattr(carrier, "active_orgs"):
-                    carrier.active_orgs.remove(info.context.request.org)
-                else:
-                    carrier.active_users.remove(info.context.request.user)
+            data["is_enabled"] = input.get("enable")
 
         if "config" in input:
-            config = providers.Carrier.resolve_config(carrier, is_user_config=True)
-            serializers.CarrierConfigModelSerializer.map(
-                instance=config,
+            data["config_overrides"] = input.get("config") or {}
+
+        # Use the serializer to create or update the BrokeredConnection
+        # The @owned_model_serializer decorator handles org linking automatically
+        brokered = (
+            BrokeredConnectionModelSerializer.map(
+                data=data,
                 context=context,
-                data={
-                    "carrier": carrier.pk,
-                    "config": process_dictionaries_mutations(
-                        ["config"], (input["config"] or {}), config
-                    ),
-                },
-            ).save()
+            )
+            .save()
+            .instance
+        )
 
         return SystemCarrierMutation(
-            carrier=providers.Carrier.system_carriers.get(pk=pk)
+            carrier=system_connection
         )  # type: ignore