From 6b93a80c7c5b36c3bdda8328e076cec870634302 Mon Sep 17 00:00:00 2001
From: "chafique.delli" <chafique.delli@akretion.com>
Date: Fri, 6 Sep 2024 14:46:25 +0200
Subject: [PATCH] shopinvader_api_delivery_pickup: add with_dropoff_site in
 schema

---
 shopinvader_api_delivery_pickup/schemas/__init__.py    |  5 ++++-
 .../schemas/delivery_carrier.py                        | 10 ++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/shopinvader_api_delivery_pickup/schemas/__init__.py b/shopinvader_api_delivery_pickup/schemas/__init__.py
index 80f4b16..3714183 100644
--- 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
index 81bd1a8..dfc3e5f 100644
--- 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