data-protection/privacy_consent/models/privacy_activity.py

# Copyright 2018 Tecnativa - Jairo Llopis
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo import _, api, fields, models
from odoo.exceptions import ValidationError
from odoo.tools.safe_eval import safe_eval


class PrivacyActivity(models.Model):
    _inherit = "privacy.activity"

    server_action_id = fields.Many2one(
        "ir.actions.server",
        "Server action",
        domain=[("model_id.model", "=", "privacy.consent")],
        help="Run this action when a new consent request is created or its "
        "acceptance status is updated.",
    )
    consent_ids = fields.One2many("privacy.consent", "activity_id", "Consents",)
    consent_count = fields.Integer("Consents count", compute="_compute_consent_count",)
    consent_required = fields.Selection(
        [("auto", "Automatically"), ("manual", "Manually")],
        "Ask subjects for consent",
        help="Enable if you need to track any kind of consent "
        "from the affected subjects",
    )
    consent_template_id = fields.Many2one(
        "mail.template",
        "Email template",
        default=lambda self: self._default_consent_template_id(),
        domain=[("model", "=", "privacy.consent")],
        help="Email to be sent to subjects to ask for consent. "
        "A good template should include details about the current "
        "consent request status, how to change it, and where to "
        "get more information.",
    )
    default_consent = fields.Boolean(
        "Accepted by default",
        help="Should we assume the subject has accepted if we receive no " "response?",
    )

    # Hidden helpers help user design new templates
    consent_template_default_body_html = fields.Text(
        compute="_compute_consent_template_defaults",
    )
    consent_template_default_subject = fields.Char(
        compute="_compute_consent_template_defaults",
    )

    @api.model
    def _default_consent_template_id(self):
        return self.env.ref("privacy_consent.template_consent", False)

    @api.depends("consent_ids")
    def _compute_consent_count(self):
        groups = self.env["privacy.consent"].read_group(
            [("activity_id", "in", self.ids)], ["activity_id"], ["activity_id"],
        )
        for group in groups:
            self.browse(group["activity_id"][0], self._prefetch).consent_count = group[
                "activity_id_count"
            ]

    def _compute_consent_template_defaults(self):
        """Used in context values, to help users design new templates."""
        template = self._default_consent_template_id()
        if template:
            self.update(
                {
                    "consent_template_default_body_html": template.body_html,
                    "consent_template_default_subject": template.subject,
                }
            )

    @api.constrains("consent_required", "consent_template_id")
    def _check_auto_consent_has_template(self):
        """Require a mail template to automate consent requests."""
        for one in self:
            if one.consent_required == "auto" and not one.consent_template_id:
                raise ValidationError(
                    _("Specify a mail template to ask automated consent.")
                )

    @api.constrains("consent_required", "subject_find")
    def _check_consent_required_subject_find(self):
        for one in self:
            if one.consent_required and not one.subject_find:
                raise ValidationError(
                    _(
                        "Require consent is available only for subjects "
                        "in current database."
                    )
                )

    @api.model
    def _cron_new_consents(self):
        """Ask all missing automatic consent requests."""
        automatic = self.search([("consent_required", "=", "auto")])
        automatic.action_new_consents()

    @api.onchange("consent_required")
    def _onchange_consent_required_subject_find(self):
        """Find subjects automatically if we require their consent."""
        if self.consent_required:
            self.subject_find = True

    def action_new_consents(self):
        """Generate new consent requests."""
        consents_vals = []
        # Skip activitys where consent is not required
        for one in self.with_context(active_test=False).filtered("consent_required"):
            domain = [
                ("id", "not in", one.mapped("consent_ids.partner_id").ids),
                ("email", "!=", False),
            ] + safe_eval(one.subject_domain)
            # Store values for creating missing consent requests
            for missing in self.env["res.partner"].search(domain):
                consents_vals.append(
                    {
                        "partner_id": missing.id,
                        "accepted": one.default_consent,
                        "activity_id": one.id,
                    }
                )
        # Create and send consent request emails for automatic activitys
        consents = self.env["privacy.consent"].create(consents_vals)
        consents.action_auto_ask()
        # Redirect user to new consent requests generated
        return {
            "domain": [("id", "in", consents.ids)],
            "name": _("Generated consents"),
            "res_model": consents._name,
            "type": "ir.actions.act_window",
            "view_mode": "tree,form",
        }