[IMP] survey_event_registration_generation: event registration generation

survey_event_registration_generation/models/survey_user_input.py

@@ -15,7 +15,14 @@ _logger = logging.getLogger(__name__)
 class SurveyUserInput(models.Model):
     _inherit = 'survey.user_input'
 
+    registration_id = fields.Many2one('event.registration', 'Event registration')
+
     def save_lines(self, question, answer, comment=None):
+        """
+            Inherit save_lines method, called after each answer
+            to save selected event or event_product in user_input.line
+            depending on the question type
+        """
         old_answers = self.env['survey.user_input.line'].search([
             ('user_input_id', '=', self.id),
             ('question_id', '=', question.id)
@@ -29,6 +36,9 @@ class SurveyUserInput(models.Model):
 
     
     def _save_event_product(self, question, old_answers, answer):
+        """
+            Save event product to user_input.line
+        """
         vals = self._get_line_answer_values(question, answer, question.question_type)
         vals['value_event_product'] = int(vals['value_event_product'])
         if old_answers:
@@ -38,10 +48,87 @@ class SurveyUserInput(models.Model):
             return self.env['survey.user_input.line'].create(vals)
         
     def _save_event(self, question, old_answers, answer):
+        """
+            Save event to user_input.line
+        """
         vals = self._get_line_answer_values(question, answer, question.question_type)
         vals['value_event'] = int(vals['value_event'])
         if old_answers:
             old_answers.write(vals)
             return old_answers
         else:
-            return self.env['survey.user_input.line'].create(vals)
+            return self.env['survey.user_input.line'].create(vals)
+        
+    def _prepare_registration(self):
+        """Extract registration values from the answers"""
+        
+        elegible_inputs = self.user_input_line_ids.filtered(
+            lambda x: x.question_id.event_registration_field and not x.skipped
+        )
+        basic_inputs = elegible_inputs.filtered(
+            lambda x: x.answer_type not in {"suggestion"}
+            and x.question_id.event_registration_field.name != "comment"
+        )
+
+        vals = {}
+        for line in basic_inputs:
+            if line.question_id.event_registration_field.ttype == 'many2one':
+                vals[line.question_id.event_registration_field.name] = line[f"value_{line.answer_type}"].id
+            else:
+                vals[line.question_id.event_registration_field.name] = line[f"value_{line.answer_type}"]
+     
+        for line in elegible_inputs - basic_inputs:
+            field_name = line.question_id.event_registration_field.name
+            if line.question_id.event_registration_field.ttype == "many2one":
+                vals[
+                    field_name
+                ] = line.suggested_answer_id.event_registration_field_resource_ref.id
+            elif line.question_id.event_registration_field.ttype == "many2many":
+                vals.setdefault(field_name, [])
+                vals[field_name] += [
+                    (4, line.suggested_answer_id.event_registration_field_resource_ref.id)
+                ]
+            # We'll use the comment field to add any other infos
+            elif field_name == "comment":
+                vals.setdefault("comment", "")
+                value = (
+                    line.suggested_answer_id.value
+                    if line.answer_type == "suggestion"
+                    else line[f"value_{line.answer_type}"]
+                )
+                vals["comment"] += f"\n{line.question_id.title}: {value}"
+            else:
+                if line.question_id.question_type == "multiple_choice":
+                    if not vals.get(field_name):
+                        vals[field_name] = line.suggested_answer_id.value
+                    else:
+                        vals[field_name] += line.suggested_answer_id.value
+                else:
+                    vals[field_name] = line.suggested_answer_id.value
+        return vals
+
+    def _create_registration_post_process(self, registration):
+        """After creating the lead send an internal message with the input link"""
+        registration.message_post_with_view(
+            "mail.message_origin_link",
+            values={"self": registration, "origin": self.survey_id},
+            subtype_id=self.env.ref("mail.mt_note").id,
+        )
+
+    def _mark_done(self):
+        """Generate registration when the survey is submitted"""
+        for user_input in self.filtered(
+            lambda r: r.survey_id.generate_registration and not self.registration_id
+        ):
+            vals = user_input._prepare_registration()
+            
+            registration = self.env["event.registration"].create(vals)
+            self._create_registration_post_process(registration)
+            self.update({"registration_id": registration.id})
+        res = super()._mark_done()
+
+        # after all, set partner id of registration as the partner of user input
+        for user_input in self:
+            if user_input.registration_id and user_input.partner_id:
+                user_input.registration_id.partner_id = user_input.partner_id
+        return res