odoo-survey-addons/survey_partner_input/models/res_partner.py

from odoo import models, fields, api, _
from odoo.osv.expression import normalize_domain, AND


class ResPartner(models.Model):
    _inherit = "res.partner"

    tot_sent_survey = fields.Integer("Sent survey count", compute="_count_survey_input")
    tot_done_survey = fields.Integer(
        "Completed survey count", compute="_count_survey_input"
    )
    tot_sent_done_survey = fields.Integer(
        "Completed sent survey count", compute="_count_survey_input"
    )
    sent_done_ratio = fields.Integer(
        string="Completed sent survey ratio", compute="_get_sent_done_ratio"
    )

    # COMPUTES

    def _count_survey_input(self):
        UserInput = self.env["survey.user_input"]
        partners_survey = UserInput
        in_onchange = self.env.context.get("in_onchange", False)
        origin = in_onchange and self._origin or False
        if in_onchange:
            domain = [
                ("partner_id", "=", self._origin.id),
                "|",
                ("invite_token", "not in", [False, None]),
                ("state", "=", "dones"),
            ]
            if self.email:
                domain = ["|", ("email", "=", self.email)] + domain
            partners_survey = UserInput.search(domain)
        else:
            partners_survey = UserInput.search(
                [
                    "|",
                    ("partner_id", "in", self.ids),
                    ("email", "in", self.filtered("email").mapped("email")),
                    "|",
                    ("invite_token", "not in", [False, None]),
                    ("state", "=", "done"),
                ]
            )
        for partner in self:
            done = partners_survey.filtered(
                lambda sui: (
                    sui.partner_id == (origin or partner)
                    or partner.email
                    and sui.email == partner.email
                )
                and sui.state == "done"
            )
            link = partners_survey.filtered(
                lambda sui: (
                    sui.partner_id == (origin or partner)
                    or partner.email
                    and sui.email == partner.email
                )
                and sui.invite_token
            )
            partner.tot_sent_survey = len(link)
            partner.tot_done_survey = len(done)
            partner.tot_sent_done_survey = len(link & done)

    @api.depends("tot_sent_done_survey", "tot_sent_survey")
    def _get_sent_done_ratio(self):
        for survey in self:
            if survey.tot_sent_survey == 0:
                survey.sent_done_ratio = 0
            else:
                survey.sent_done_ratio = int(
                    round(100 * survey.tot_sent_done_survey / survey.tot_sent_survey, 0)
                )

    # ACTIONS

    def action_survey_user_input(self):
        self.ensure_one()
        action = self.env.ref("survey.action_survey_user_input").read()[0]
        action["display_name"] += _(" from {}").format(self.display_name)
        # manage context
        ctx = dict(self.env.context)
        link_only = ctx.pop("link_only", False)
        action["context"] = ctx
        # manage domain
        domain = action.get("domain") or []
        if isinstance(domain, str):
            domain = eval(domain)
        if len(domain) > 1:
            domain = AND(
                [
                    ["|", ("partner_id", "=", self.id), ("email", "ilike", self.email)],
                    normalize_domain(domain),
                ]
            )
        else:
            domain = ["|", ("partner_id", "=", self.id), ("email", "ilike", self.email)]
        if link_only:
            if len(domain) > 1:
                domain = AND(
                    [
                        [("invite_token", "not in", [False, None])],
                        normalize_domain(domain),
                    ]
                )
            else:
                domain = [("invite_token", "not in", [False, None])]
        action["domain"] = domain
        # return updated action
        return action

    # ONCHANGES

    @api.onchange("email")
    def onchange_email(self):
        self.ensure_one()
        if isinstance(self._origin.id, int):
            self.with_context(in_onchange=True)._count_survey_input()