odoo-survey-addons/survey_custom_matrix/models/survey_user_input_line.py

# -*- coding: utf-8 -*-
from odoo import models, fields, api
from odoo.addons.survey.models.survey import dict_keys_startswith


class SurveyUserInputLine(models.Model):
    _inherit = "survey.user_input_line"

    answer_type = fields.Selection(
        selection_add=[("dropdown", "Dropdown")], string="Answer Type"
    )
    matrix_subtype = fields.Selection(related="question_id.matrix_subtype")
    value_id = fields.Many2one("survey.label.value", string="Value Dropdown")

    @api.model
    def save_line_matrix(self, user_input_id, question, post, answer_tag):
        vals = {
            "user_input_id": user_input_id,
            "question_id": question.id,
            "survey_id": question.survey_id.id,
            "skipped": False,
        }
        suil = self.search(
            [
                ("user_input_id", "=", user_input_id),
                ("survey_id", "=", question.survey_id.id),
                ("question_id", "=", question.id),
            ]
        )
        suil.sudo().unlink()

        no_answers = True
        ca_dict = dict_keys_startswith(post, answer_tag + "_")

        comment_answer = ca_dict.pop(("%s_%s" % (answer_tag, "comment")), "").strip()
        if comment_answer:
            vals.update(
                {
                    "answer_type": "text",
                    "value_text": comment_answer,
                }
            )
            self.create(vals)
            no_answers = False

        if question.matrix_subtype == "simple":
            for row in question.labels_ids_2:
                a_tag = "%s_%s" % (answer_tag, row.id)
                if a_tag in ca_dict:
                    no_answers = False
                    vals.update(
                        {
                            "answer_type": "suggestion",
                            "value_suggested": ca_dict[a_tag],
                            "value_suggested_row": row.id,
                        }
                    )
                    self.create(vals)

        elif question.matrix_subtype == "multiple":
            for col in question.labels_ids:
                for row in question.labels_ids_2:
                    a_tag = "%s_%s_%s" % (answer_tag, row.id, col.id)
                    if a_tag in ca_dict:
                        no_answers = False
                        vals.update(
                            {
                                "answer_type": "suggestion",
                                "value_suggested": col.id,
                                "value_suggested_row": row.id,
                            }
                        )
                        self.create(vals)

        elif question.matrix_subtype == "custom":
            for col in question.labels_ids:
                for row in question.labels_ids_2:
                    a_tag = "%s_%s_%s" % (answer_tag, row.id, col.id)
                    if a_tag in ca_dict:
                        no_answers = False
                        if post.get(a_tag):
                            sline = a_tag.split("_")[-1]
                            label_obj = question.labels_ids.browse(int(sline))
                            if label_obj.type == "textbox":
                                vals.update(
                                    {
                                        "answer_type": "text",
                                        "value_suggested": col.id,
                                        "value_suggested_row": row.id,
                                        "value_text": post.get(a_tag),
                                    }
                                )
                            elif label_obj.type == "free_text":
                                vals.update(
                                    {
                                        "answer_type": "free_text",
                                        "value_suggested": col.id,
                                        "value_suggested_row": row.id,
                                        "value_free_text": post.get(a_tag),
                                    }
                                )
                            elif label_obj.type == "numerical_box":
                                vals.update(
                                    {
                                        "answer_type": "number",
                                        "value_suggested": col.id,
                                        "value_suggested_row": row.id,
                                        "value_number": post.get(a_tag),
                                    }
                                )
                            elif label_obj.type == "date":
                                vals.update(
                                    {
                                        "answer_type": "date",
                                        "value_suggested": col.id,
                                        "value_suggested_row": row.id,
                                        "value_date": post.get(a_tag),
                                    }
                                )
                            elif label_obj.type == "dropdown":
                                vals.update(
                                    {
                                        "answer_type": "dropdown",
                                        "value_suggested": col.id,
                                        "value_suggested_row": row.id,
                                        "value_id": int(post.get(a_tag)),
                                    }
                                )
                            else:
                                vals.update(
                                    {
                                        "answer_type": "suggestion",
                                        "value_suggested": col.id,
                                        "value_suggested_row": row.id,
                                    }
                                )
                            self.create(vals)

        if no_answers:
            vals.update(
                {
                    "answer_type": None,
                    "skipped": True,
                }
            )
            self.create(vals)
        return True