Update followers on crm-lead

energy_communities/wizards/assign_crm_to_coordinator_company.py

@@ -2,25 +2,56 @@ from odoo import models, fields, api
 
 
 class AssignCRMToCoordinatorCompanyWizard(models.TransientModel):
-    _name = 'assign.crm.to.coordinator.company.wizard'
-    _description = 'Assign CRM to coordinator company wizard'
+    _name = "assign.crm.to.coordinator.company.wizard"
+    _description = "Assign CRM to coordinator company wizard"
 
-    crm_lead_id = fields.Many2one('crm.lead')
+    crm_lead_id = fields.Many2one("crm.lead")
     assigned_company_id = fields.Many2one(
-        'res.company',
-        string='Assigned company',
+        "res.company",
+        string="Assigned company",
         required=True,
-        domain=[('hierarchy_level', '=', 'coordinator')]
+        domain=[("hierarchy_level", "=", "coordinator")],
     )
 
     @api.model
     def default_get(self, fields_list):
         defaults = super().default_get(fields_list)
-        defaults['crm_lead_id'] = self.env.context['active_id']
+        defaults["crm_lead_id"] = self.env.context["active_id"]
         return defaults
 
     def assign_to_coordinator_company(self):
         self.ensure_one()
-        self.crm_lead_id.write(
-            {'company_id': self.assigned_company_id}
+        self.remove_follower()
+        self.crm_lead_id.write({"company_id": self.assigned_company_id})
+        self.add_follower()
+
+    def remove_follower(self):
+        instance_admin = self.env.ref("energy_communities.role_ce_manager").id
+        company_id = self.crm_lead_id.company_id.id
+        followers = self.env["res.users"].search(
+            [
+                ("role_line_ids.role_id", "=", instance_admin),
+                ("company_ids.id", "=", company_id),
+            ]
+        )
+        if followers:
+            self.crm_lead_id.message_unsubscribe(partner_ids=followers.partner_id.ids)
+
+    def add_follower(self):
+        coordinator_admin = self.env.ref("energy_communities.role_coord_admin").id
+        company_id = self.crm_lead_id.company_id.id
+        followers = self.env["res.users"].search(
+            [
+                ("role_line_ids.role_id", "=", coordinator_admin),
+                ("company_ids.id", "=", company_id),
+            ]
         )
+        if followers:
+            self.crm_lead_id.message_subscribe(partner_ids=followers.partner_id.ids)
+            # notify followers
+            for follower in followers:
+                email_values = {"email_to": follower.partner_id.email}
+                template = self.env.ref(
+                    "energy_communities.email_templ_lead_assigned_to_coordinator_id"
+                ).with_context(email_values)
+                self.crm_lead_id.message_post_with_template(template.id)