shopinvader_api_delivery_pickup: add with_dropoff_site in schema

shopinvader · Sep 6, 2024 · 6b93a80 · 6b93a80
1 parent fc80b3c
commit 6b93a80
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/shopinvader_api_delivery_pickup/schemas/__init__.py b/shopinvader_api_delivery_pickup/schemas/__init__.py
@@ -5,4 +5,7 @@
     DeliveryPickupSearch,
 )
 from .address import DeliveryAddress
-from .delivery_carrier import DeliveryCarrierWithPrice
+from .delivery_carrier import (
+    DeliveryCarrierWithPrice,
+    DeliveryCarrier,
+)
diff --git a/shopinvader_api_delivery_pickup/schemas/delivery_carrier.py b/shopinvader_api_delivery_pickup/schemas/delivery_carrier.py
@@ -12,3 +12,13 @@ def from_delivery_carrier(cls, odoo_rec, cart=None):
         res = super().from_delivery_carrier(odoo_rec, cart=cart)
         res.with_dropoff_site = odoo_rec.with_dropoff_site or None
         return res
+
+
+class DeliveryCarrier(delivery_carrier.DeliveryCarrier, extends=True):
+    with_dropoff_site: bool | None = None
+
+    @classmethod
+    def from_delivery_carrier(cls, odoo_rec):
+        res = super().from_delivery_carrier(odoo_rec)
+        res.with_dropoff_site = odoo_rec.with_dropoff_site or None
+        return res