From 297b52330bafc3f6006f37025e8df62513c2eeb2 Mon Sep 17 00:00:00 2001
From: Dario Lodeiros <dario@commitsun.com>
Date: Wed, 30 Jun 2021 08:34:06 +0200
Subject: [PATCH] [IMP] computes reservation customer fields

---
 pms/models/pms_availability_plan.py |  1 +
 pms/models/pms_folio.py             | 30 ++++++++++++++++++++++++++---
 pms/models/pms_reservation.py       |  8 +++++++-
 3 files changed, 35 insertions(+), 4 deletions(-)

diff --git a/pms/models/pms_availability_plan.py b/pms/models/pms_availability_plan.py
index 3ae04ded4..419da062b 100644
--- a/pms/models/pms_availability_plan.py
+++ b/pms/models/pms_availability_plan.py
@@ -187,6 +187,7 @@ class PmsAvailabilityPlan(models.Model):
             ("room_type_id", "=", room_type_id),
             ("pms_property_id", "=", pms_property_id),
         ]
+        pricelist = False
         if pricelist_id:
             pricelist = self.env["product.pricelist"].browse(pricelist_id)
         if pricelist and pricelist.availability_plan_id:
diff --git a/pms/models/pms_folio.py b/pms/models/pms_folio.py
index b3006352c..ce8e19c1f 100644
--- a/pms/models/pms_folio.py
+++ b/pms/models/pms_folio.py
@@ -765,27 +765,51 @@ class PmsFolio(models.Model):
             else:
                 order.invoice_status = "no"
 
-    @api.depends("partner_id", "partner_id.name")
+    @api.depends("partner_id", "partner_id.name", "reservation_ids.partner_name")
     def _compute_partner_name(self):
         for record in self:
             if record.partner_id and not record.partner_name:
                 record.partner_name = record.partner_id.name
+            # if there is only one customer name in the folio reservations
+            # we update the partner name of the folio when the partner name of
+            # the reservations is modified
+            elif (
+                len(record.reservation_ids.mapped("partner_name")) == 1
+                and not record.partner_name
+            ):
+                record.partner_name = record.reservation_ids[0].partner_name
             elif not record.partner_name:
                 record.partner_name = False
 
-    @api.depends("partner_id", "partner_id.email")
+    @api.depends("partner_id", "partner_id.email", "reservation_ids.partner_email")
     def _compute_email(self):
         for record in self:
             if record.partner_id and not record.email:
                 record.email = record.partner_id.email
+            # if there is only one customer email in the folio reservations
+            # we update the partner email of the folio when the partner email of
+            # the reservations is modified
+            elif (
+                len(record.reservation_ids.mapped("partner_email")) == 1
+                and not record.email
+            ):
+                record.email = record.reservation_ids[0].partner_email
             elif not record.email:
                 record.email = False
 
-    @api.depends("partner_id", "partner_id.mobile")
+    @api.depends("partner_id", "partner_id.mobile", "reservation_ids.partner_mobile")
     def _compute_mobile(self):
         for record in self:
             if record.partner_id and not record.mobile:
                 record.mobile = record.partner_id.mobile
+            # if there is only one customer mobile in the folio reservations
+            # we update the partner mobile of the folio when the partner mobile of
+            # the reservations is modified
+            elif (
+                len(record.reservation_ids.mapped("partner_mobile")) == 1
+                and not record.mobile
+            ):
+                record.mobile = record.reservation_ids[0].partner_mobile
             elif not record.mobile:
                 record.mobile = False
 
diff --git a/pms/models/pms_reservation.py b/pms/models/pms_reservation.py
index 155b329f8..2d7fb57f4 100644
--- a/pms/models/pms_reservation.py
+++ b/pms/models/pms_reservation.py
@@ -1210,11 +1210,13 @@ class PmsReservation(models.Model):
             else:
                 record.shared_folio = False
 
-    @api.depends("partner_id", "partner_id.name")
+    @api.depends("partner_id", "partner_id.name", "folio_id.partner_name")
     def _compute_partner_name(self):
         for record in self:
             if record.partner_id and not record.partner_name:
                 record.partner_name = record.partner_id.name
+            elif record.folio_id.partner_name and not record.partner_name:
+                record.partner_name = record.folio_id.partner_name
             elif not record.partner_name:
                 record.partner_name = False
 
@@ -1223,6 +1225,8 @@ class PmsReservation(models.Model):
         for record in self:
             if record.partner_id and not record.partner_email:
                 record.partner_email = record.partner_id.email
+            elif record.folio_id.email and not record.partner_email:
+                record.partner_email = record.folio_id.partner_email
             elif not record.partner_email:
                 record.partner_email = False
 
@@ -1231,6 +1235,8 @@ class PmsReservation(models.Model):
         for record in self:
             if record.partner_id and not record.partner_mobile:
                 record.partner_mobile = record.partner_id.mobile
+            elif record.folio_id.mobile and not record.partner_mobile:
+                record.partner_mobile = record.folio_id.partner_mobile
             elif not record.partner_mobile:
                 record.partner_mobile = False