From 2daf2a33c0f51d8045f923d3d7b79a25d3fa9100 Mon Sep 17 00:00:00 2001
From: David <david.vidal@tecnativa.com>
Date: Wed, 2 Aug 2023 16:49:14 +0200
Subject: [PATCH] [IMP] survey_sale_generation: multiple choice multiplier

TT40983
---
 survey_sale_generation/demo/survey_sale_demo.xml         | 9 +++++----
 survey_sale_generation/models/survey_question.py         | 7 +++++++
 survey_sale_generation/models/survey_user_input.py       | 9 +++++++--
 .../tests/test_survey_sale_generation.py                 | 2 +-
 survey_sale_generation/views/survey_question_views.xml   | 9 +++++++++
 5 files changed, 29 insertions(+), 7 deletions(-)

diff --git a/survey_sale_generation/demo/survey_sale_demo.xml b/survey_sale_generation/demo/survey_sale_demo.xml
index f2f9efa3..3b4e6322 100644
--- a/survey_sale_generation/demo/survey_sale_demo.xml
+++ b/survey_sale_generation/demo/survey_sale_demo.xml
@@ -51,7 +51,7 @@
     </record>
     <record model="survey.question" id="survey_support_q0">
         <field name="survey_id" ref="survey_hire_support" />
-        <field name="sequence">0</field>
+        <field name="sequence">1</field>
         <field name="title">E-mail address</field>
         <field name="question_type">text_box</field>
         <field name="show_in_sale_order_comment" eval="True" />
@@ -59,7 +59,7 @@
     </record>
     <record model="survey.question" id="survey_support_q1">
         <field name="survey_id" ref="survey_hire_support" />
-        <field name="sequence">1</field>
+        <field name="sequence">2</field>
         <field name="title">How many hours will you hire monthly?</field>
         <field name="question_type">numerical_box</field>
         <field name="product_id" ref="support_hours" />
@@ -67,10 +67,11 @@
     </record>
     <record model="survey.question" id="survey_support_q2">
         <field name="survey_id" ref="survey_hire_support" />
-        <field name="sequence">2</field>
+        <field name="sequence">3</field>
         <field name="title">Choose your subscription level</field>
         <field name="question_type">simple_choice</field>
         <field name="constr_mandatory" eval="True" />
+        <field name="product_uom_qty_question_id" ref="survey_support_q1" />
     </record>
     <record model="survey.question.answer" id="survey_support_q2_sug1">
         <field name="question_id" ref="survey_support_q2" />
@@ -92,7 +93,7 @@
     </record>
     <record model="survey.question" id="survey_support_q3">
         <field name="survey_id" ref="survey_hire_support" />
-        <field name="sequence">2</field>
+        <field name="sequence">5</field>
         <field name="title">Choose your extras</field>
         <field name="question_type">multiple_choice</field>
         <field name="constr_mandatory" eval="False" />
diff --git a/survey_sale_generation/models/survey_question.py b/survey_sale_generation/models/survey_question.py
index f0bf978b..72928995 100644
--- a/survey_sale_generation/models/survey_question.py
+++ b/survey_sale_generation/models/survey_question.py
@@ -7,6 +7,13 @@ class SurveyQuestion(models.Model):
     _inherit = "survey.question"
 
     product_id = fields.Many2one(comodel_name="product.product")
+    product_uom_qty_question_id = fields.Many2one(
+        comodel_name="survey.question",
+        help="For multiple answer questions we might want to consider how much of that "
+        "product we want to order. For example, one question could be: 'How many "
+        "members in your team?' and other could be 'Choose the membership type'. We'll "
+        "want to multiply members by memberships in our sale line.",
+    )
     show_in_sale_order_comment = fields.Boolean()
 
 
diff --git a/survey_sale_generation/models/survey_user_input.py b/survey_sale_generation/models/survey_user_input.py
index 428ec041..737e3d5a 100644
--- a/survey_sale_generation/models/survey_user_input.py
+++ b/survey_sale_generation/models/survey_user_input.py
@@ -11,7 +11,7 @@ class SurveyUserInput(models.Model):
 
     def _prepare_quotation(self):
         return {
-            "partner_id": self.create_uid.partner_id.id,
+            "partner_id": self.partner_id.id or self.create_uid.partner_id.id,
             "origin": self.survey_id.title,
             "survey_user_input_id": self.id,
             "company_id": self.create_uid.company_id.id,
@@ -24,7 +24,12 @@ def _prepare_quotation_line(self, input_line):
             qty = input_line.value_numerical_box
         else:
             product_id = input_line.suggested_answer_id.product_id
-            qty = 1
+            # We can set a related question that will be the qty multiplier
+            qty_question = self.user_input_line_ids.filtered(
+                lambda x: x.question_id
+                == input_line.question_id.product_uom_qty_question_id
+            ).value_numerical_box
+            qty = qty_question is not False and qty_question or 1
         return {
             "product_id": product_id.id,
             "product_uom_qty": qty,
diff --git a/survey_sale_generation/tests/test_survey_sale_generation.py b/survey_sale_generation/tests/test_survey_sale_generation.py
index aa6020df..c2e2760a 100644
--- a/survey_sale_generation/tests/test_survey_sale_generation.py
+++ b/survey_sale_generation/tests/test_survey_sale_generation.py
@@ -42,7 +42,7 @@ def test_sale_generation(self):
         """
         expected_lines = {
             self.support_hours: 3,
-            self.gold_service: 1,
+            self.gold_service: 3,
             self.advanced_backup: 1,
             self.mail_management: 1,
         }
diff --git a/survey_sale_generation/views/survey_question_views.xml b/survey_sale_generation/views/survey_question_views.xml
index 7bbbac9d..3c412fcb 100644
--- a/survey_sale_generation/views/survey_question_views.xml
+++ b/survey_sale_generation/views/survey_question_views.xml
@@ -11,6 +11,15 @@
                 >
                     <field name="product_id" />
                 </group>
+                <group
+                    name="Linked product"
+                    attrs="{'invisible': [('question_type', 'not in', ['simple_choice', 'multiple_choice'])]}"
+                >
+                    <field
+                        name="product_uom_qty_question_id"
+                        domain="[('survey_id', '=', survey_id), ('question_type', '=', 'numerical_box')]"
+                    />
+                </group>
             </xpath>
             <xpath expr="//field[@name='value']" position="after">
                 <field