chore: fix generic carrier configration parsing

karrioapi · Aug 22, 2024 · 3ef15ca · 3ef15ca
1 parent 3ff9167
commit 3ef15ca
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 11 deletions.
diff --git a/modules/core/karrio/server/core/dataunits.py b/modules/core/karrio/server/core/dataunits.py
@@ -78,11 +78,10 @@ def contextual_reference(request: Request = None, reduced: bool = True):
 
     def _get_generic_carriers():
         system_custom_carriers = [
-            c.settings
-            for c in gateway.Carriers.list(system_only=True, carrier_name="generic")
+            c for c in gateway.Carriers.list(system_only=True, carrier_name="generic")
         ]
         custom_carriers = [
-            c.settings
+            c
             for c in (
                 gateway.Carriers.list(context=request, carrier_name="generic").exclude(
                     is_system=True
@@ -93,13 +92,14 @@ def _get_generic_carriers():
         ]
 
         extra_carriers = {
-            c.custom_carrier_name: c.display_name for c in custom_carriers
+            f"{c.credentials.get('custom_carrier_name') or "generic"}": c.display_name for c in custom_carriers
         }
         system_carriers = {
-            c.custom_carrier_name: c.display_name for c in system_custom_carriers
+            f"{c.credentials.get('custom_carrier_name') or "generic"}": c.display_name
+            for c in system_custom_carriers
         }
         extra_services = {
-            c.custom_carrier_name: {
+            f"{c.credentials.get('custom_carrier_name') or "generic"}": {
                 s.service_code: s.service_code for s in c.services.all()
             }
             for c in custom_carriers

diff --git a/modules/core/karrio/server/core/gateway.py b/modules/core/karrio/server/core/gateway.py
@@ -1,7 +1,7 @@
 import uuid
 import typing
 import logging
-from datetime import datetime
+import datetime
 
 from django.db.models import Q
 from django.conf import settings
@@ -198,7 +198,7 @@ def process_meta(parent) -> dict:
                 "ext": carrier.ext,
                 "carrier": rate_provider,
                 "service_name": service_name,
-                "rate_provider": rate_provider,  # TODO: deprecate rate_provider
+                "rate_provider": rate_provider,  # TODO: deprecate 'rate_provider' in favor of 'carrier'
             }
 
         def process_selected_rate() -> dict:
@@ -263,7 +263,9 @@ def process_parcel_refs(parcels: typing.List[dict]) -> list:
                 "parcels": process_parcel_refs(payload["parcels"]),
                 "tracking_url": process_tracking_url(shipment_rate),
                 "status": serializers.ShipmentStatus.purchased.value,
-                "created_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f%z"),
+                "created_at": datetime.datetime.now().strftime(
+                    "%Y-%m-%d %H:%M:%S.%f%z"
+                ),
                 "meta": process_meta(shipment),
                 "messages": messages,
             },
@@ -361,10 +363,10 @@ def track(
             tracking_number=tracking_number,
             events=[
                 datatypes.TrackingEvent(
-                    date=datetime.now().strftime("%Y-%m-%d"),
+                    date=datetime.datetime.now().strftime("%Y-%m-%d"),
                     description="Awaiting update from carrier...",
                     code="UNKNOWN",
-                    time=datetime.now().strftime("%H:%M"),
+                    time=datetime.datetime.now().strftime("%H:%M"),
                 )
             ],
             delivered=False,