From c77f2436a106a5f6f1aef936ae9565ccde7a5598 Mon Sep 17 00:00:00 2001
From: Daniel K <danielk.developer@gmail.com>
Date: Wed, 16 Oct 2024 15:39:53 +0200
Subject: [PATCH] feat: add support for additional seko_reference_*

---
 .../seko/karrio/providers/seko/shipment/create.py         | 8 ++++----
 modules/connectors/seko/karrio/providers/seko/units.py    | 2 ++
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/modules/connectors/seko/karrio/providers/seko/shipment/create.py b/modules/connectors/seko/karrio/providers/seko/shipment/create.py
index 01a222024..c85f1b566 100644
--- a/modules/connectors/seko/karrio/providers/seko/shipment/create.py
+++ b/modules/connectors/seko/karrio/providers/seko/shipment/create.py
@@ -71,7 +71,7 @@ def shipment_request(
     shipper = lib.to_address(payload.shipper)
     recipient = lib.to_address(payload.recipient)
     packages = lib.to_packages(payload.parcels)
-    service = provider_units.ShippingService.map(payload.service)
+    service = provider_units.ShippingService.map(payload.service).value_or_key
     options = lib.to_shipping_options(
         payload.options,
         package_options=packages.options,
@@ -95,8 +95,8 @@ def shipment_request(
     # map data to convert karrio model to seko specific type
     request = seko.ShippingRequestType(
         DeliveryReference=payload.reference,
-        Reference2=None,
-        Reference3=None,
+        Reference2=options.seko_reference_2.state,
+        Reference3=options.seko_reference_3.state,
         Origin=seko.DestinationType(
             Id=options.seko_origin_id.state,
             Name=shipper.company_name,
@@ -174,7 +174,7 @@ def shipment_request(
         ),
         IncludeLineDetails=True,
         Carrier=options.seko_carrier.state,
-        Service=service.value_or_key,
+        Service=service,
         CostCentreName=settings.connection_config.cost_center.state,
         CodValue=options.cash_on_delivery.state,
         TaxCollected=lib.identity(
diff --git a/modules/connectors/seko/karrio/providers/seko/units.py b/modules/connectors/seko/karrio/providers/seko/units.py
index 7ac5ede8b..67482c761 100644
--- a/modules/connectors/seko/karrio/providers/seko/units.py
+++ b/modules/connectors/seko/karrio/providers/seko/units.py
@@ -66,6 +66,8 @@ class ShippingOption(lib.Enum):
     seko_amount_collected = lib.OptionEnum("AmountCollected", float)
     seko_tax_collected = lib.OptionEnum("TaxCollected", bool)
     seko_cod_amount = lib.OptionEnum("CODAmount", float)
+    seko_reference_2 = lib.OptionEnum("Reference2")
+    seko_reference_3 = lib.OptionEnum("Reference3")
 
     """ Unified Option type mapping """
     saturday_delivery = seko_is_saturday_delivery