[REF] refactor _prepare_automated_actions_id(), ceate() and write()

pms/models/pms_automated_mails.py

@@ -88,7 +88,6 @@ class PmsAutomatedMails(models.Model):
         dict_val = self._prepare_automated_actions_id(
             action, time, moment, properties, is_create
         )
-
         action_server_vals = {
             "name": name,
             "state": "email",
@@ -96,28 +95,30 @@ class PmsAutomatedMails(models.Model):
             "model_id": dict_val["model_id"],
         }
         action_server = self.env["ir.actions.server"].create(action_server_vals)
+        automated_actions_vals = {
+            "active": active,
+            "action_server_id": action_server.id,
+            "trigger": dict_val["trigger"],
+            "filter_domain": dict_val["filter_domain"],
+            "filter_pre_domain": dict_val["filter_pre_domain"],
+            "trg_date_range": dict_val["time"],
+            "trg_date_range_type": date_range_type,
+            "template_id": template_id,
+        }
         model_field = dict_val["model_field"]
-        if not model_field:
+        if model_field:
-            automated_actions_vals = {
+            automated_actions_vals.update(
-                "active": active,
+                {
-                "action_server_id": action_server.id,
-                "trigger": dict_val["trigger"],
-                "filter_domain": dict_val["filter_domain"],
-                "trg_date_range": dict_val["time"],
-                "trg_date_range_type": date_range_type,
-                "template_id": template_id,
-            }
-        else:
-            automated_actions_vals = {
-                "active": active,
-                "action_server_id": action_server.id,
-                "trigger": dict_val["trigger"],
                     "trg_date_id": dict_val["model_field"].id,
-                "filter_domain": dict_val["filter_domain"],
-                "trg_date_range": dict_val["time"],
-                "trg_date_range_type": date_range_type,
-                "template_id": template_id,
                 }
+            )
+        trigger_field = dict_val["trigger_fields"]
+        if trigger_field:
+            automated_actions_vals.update(
+                {
+                    "trigger_field_ids": dict_val["trigger_fields"].ids,
+                }
+            )
         automated_action = self.env["base.automation"].create(automated_actions_vals)
         vals.update({"automated_actions_id": automated_action.id})
         return super(PmsAutomatedMails, self).create(vals)
@@ -142,28 +143,30 @@ class PmsAutomatedMails(models.Model):
             "model_id": dict_val["model_id"],
         }
         action_server.write(action_server_vals)
+        automated_actions_vals = {
+            "active": self.active,
+            "action_server_id": action_server.id,
+            "trigger": dict_val["trigger"],
+            "filter_domain": dict_val["filter_domain"],
+            "filter_pre_domain": dict_val["filter_pre_domain"],
+            "trg_date_range": dict_val["time"],
+            "trg_date_range_type": self.time_type,
+            "template_id": self.template_id,
+        }
         model_field = dict_val["model_field"]
-        if not model_field:
+        if model_field:
-            automated_actions_vals = {
+            automated_actions_vals.update(
-                "active": self.active,
+                {
-                "action_server_id": action_server.id,
-                "trigger": dict_val["trigger"],
-                "filter_domain": dict_val["filter_domain"],
-                "trg_date_range": dict_val["time"],
-                "trg_date_range_type": self.time_type,
-                "template_id": self.template_id,
-            }
-        else:
-            automated_actions_vals = {
-                "active": self.active,
-                "action_server_id": action_server.id,
-                "trigger": dict_val["trigger"],
                     "trg_date_id": dict_val["model_field"].id,
-                "filter_domain": dict_val["filter_domain"],
-                "trg_date_range": dict_val["time"],
-                "trg_date_range_type": self.time_type,
-                "template_id": self.template_id,
                 }
+            )
+        trigger_field = dict_val["trigger_fields"]
+        if trigger_field:
+            automated_actions_vals.update(
+                {
+                    "trigger_field_ids": dict_val["trigger_fields"].ids,
+                }
+            )
         automated_actions_id.write(automated_actions_vals)
         vals.update({"automated_actions_id": automated_actions_id.id})
         return result
@@ -175,19 +178,10 @@ class PmsAutomatedMails(models.Model):
         action_server.unlink()
         return super(PmsAutomatedMails, self).unlink()
 
-    def _prepare_automated_actions_id(
+    @api.model
-        self, action, time, moment, properties, is_create
+    def _get_auto_action_fields_in_creation_action(self, moment, time):
-    ):
-        trigger = False
         model_field = False
-        model_id = False
+        model_id = self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
-        filter_domain = []
-        today = fields.Date.today()
-        # action: create reservation
-        if action == "creation":
-            model_id = (
-                self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
-            )
         if moment == "in_act":
             trigger = "on_create"
             time = 0
@@ -196,11 +190,18 @@ class PmsAutomatedMails(models.Model):
             model_field = self.env["ir.model.fields"].search(
                 [("model", "=", "pms.reservation"), ("name", "=", "create_date")]
             )
-        # action: write and cancel reservation
+        result = {
-        if action == "write" or action == "cancel":
+            "model_id": model_id,
-            model_id = (
+            "trigger": trigger,
-                self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
+            "model_field": model_field,
-            )
+            "time": time,
+        }
+        return result
+
+    @api.model
+    def _get_auto_action_fields_in_write_or_cancel_action(self, moment, time):
+        model_field = False
+        model_id = self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
         if moment == "in_act":
             trigger = "on_write"
             time = 0
@@ -209,54 +210,64 @@ class PmsAutomatedMails(models.Model):
             model_field = self.env["ir.model.fields"].search(
                 [("model", "=", "pms.reservation"), ("name", "=", "write_date")]
             )
-            if action == "cancel":
+        result = {
-                filter_domain = [
+            "model_id": model_id,
-                    ("state", "=", "cancelled"),
+            "trigger": trigger,
-                ]
+            "model_field": model_field,
-        # action: checkin
+            "time": time,
-        if action == "checkin":
+        }
-            model_id = (
+        return result
-                self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
+
-            )
+    @api.model
+    def _get_auto_action_fields_in_checkin_action(self, moment, time):
+        model_id = self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
         trigger = "on_time"
         model_field = self.env["ir.model.fields"].search(
             [("model", "=", "pms.reservation"), ("name", "=", "checkin")]
         )
-            if moment == "in_act":
+        if moment == "before":
-                time = 0
-                filter_domain = [
-                    ("checkin", "=", str(today)),
-                ]
-            elif moment == "before":
             time = time * (-1)
-        # action: checkout
+        if moment == "in_act":
-        if action == "checkout":
+            trigger = "on_write"
-            model_id = (
+            time = 0
-                self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
+        result = {
-            )
+            "model_id": model_id,
+            "trigger": trigger,
+            "model_field": model_field,
+            "time": time,
+        }
+        return result
+
+    @api.model
+    def _get_auto_action_fields_in_checkout_action(self, moment, time):
+        model_id = self.env["ir.model"].search([("model", "=", "pms.reservation")]).id
         trigger = "on_time"
         model_field = self.env["ir.model.fields"].search(
             [("model", "=", "pms.reservation"), ("name", "=", "checkout")]
         )
-            if moment == "in_act":
+        if moment == "before":
-                time = 0
-                filter_domain = [
-                    ("checkout", "=", str(today)),
-                ]
-            elif moment == "before":
             time = time * (-1)
-        # action: payments
+        if moment == "in_act":
-        if action == "payment":
+            trigger = "on_write"
+            time = 0
+        result = {
+            "model_id": model_id,
+            "trigger": trigger,
+            "model_field": model_field,
+            "time": time,
+        }
+        return result
+
+    @api.model
+    def _get_auto_action_fields_in_payment_action(self, moment, time):
+        model_field = False
         model_id = (
             self.env["ir.model"]
             .search([("model", "=", "account.payment"), ("transient", "=", False)])
             .id
         )
         if moment == "in_act":
-                trigger = "on_creation"
+            trigger = "on_create"
-                model_field = self.env["ir.model.fields"].search(
-                    [("model", "=", "account.payment"), ("name", "=", "create_date")]
-                )
             time = 0
         else:
             trigger = "on_time"
@@ -265,27 +276,84 @@ class PmsAutomatedMails(models.Model):
             )
             if moment == "before":
                 time = time * (-1)
-        # TODO: create automated action when the act is 'invoice'
+        result = {
-        # action: invoices
+            "model_id": model_id,
-        # if action == "invoice":
+            "trigger": trigger,
-        #     model_id = self.env["ir.model"].search(
+            "model_field": model_field,
-        #       [("model", "=", "account.move")]
+            "time": time,
-        #     ).id
+        }
-        #     filter_domain = [
+        return result
-        #         ("folio_ids", "!=", False),
+
-        #     ]
+    @api.model
-        #     if moment == "in_act":
+    def _get_auto_action_fields_in_invoice_action(self, moment, time):
-        #         trigger = "on_create"
+        trigger = False
-        #         time = 0
+        model_id = self.env["ir.model"].search([("model", "=", "account.move")]).id
+        if moment == "in_act":
+            trigger = "on_create"
+            time = 0
+        result = {
+            "model_id": model_id,
+            "time": time,
+            "trigger": trigger,
+            "model_field": False,
+        }
+        return result
+
+    def _prepare_automated_actions_id(
+        self, action, time, moment, properties, is_create
+    ):
+        filter_domain = []
+        filter_pre_domain = []
+        trigger_fields = False
+        dict_val = {}
+        if action == "creation":
+            dict_val = self._get_auto_action_fields_in_creation_action(moment, time)
+        elif action == "write" or action == "cancel":
+            dict_val = self._get_auto_action_fields_in_write_or_cancel_action(
+                moment, time
+            )
+            if action == "cancel":
+                filter_domain = [
+                    ("state", "=", "cancelled"),
+                ]
+        elif action == "checkin":
+            dict_val = self._get_auto_action_fields_in_checkin_action(moment, time)
+            if moment == "in_act":
+                trigger_fields = self.env["ir.model.fields"].search(
+                    [("model", "=", "pms.reservation"), ("name", "=", "state")]
+                )
+                filter_pre_domain = [("state", "=", "confirm")]
+                filter_domain = [
+                    ("state", "=", "onboard"),
+                ]
+        elif action == "checkout":
+            dict_val = self._get_auto_action_fields_in_checkout_action(moment, time)
+            if moment == "in_act":
+                trigger_fields = self.env["ir.model.fields"].search(
+                    [("model", "=", "pms.reservation"), ("name", "=", "state")]
+                )
+                filter_pre_domain = [("state", "=", "onboard")]
+                filter_domain = [
+                    ("state", "=", "out"),
+                ]
+        elif action == "payment":
+            dict_val = self._get_auto_action_fields_in_payment_action(moment, time)
+        elif action == "invoice":
+            dict_val = self._get_auto_action_fields_in_invoice_action(moment, time)
+            filter_domain = [
+                ("folio_ids", "!=", False),
+            ]
         pms_property_ids = self._get_pms_property_ids(properties, is_create)
         if pms_property_ids:
             filter_domain.append(("pms_property_id", "in", pms_property_ids))
         result = {
-            "trigger": trigger,
+            "trigger": dict_val["trigger"],
-            "model_field": model_field,
+            "model_field": dict_val["model_field"],
+            "trigger_fields": trigger_fields,
+            "filter_pre_domain": filter_pre_domain,
             "filter_domain": filter_domain,
-            "time": time,
+            "time": dict_val["time"],
-            "model_id": model_id,
+            "model_id": dict_val["model_id"],
         }
         return result