19.0 vanilla

odoo-bringout-oca-ocb-hr/hr/models/res_partner.py

@@ -1,54 +1,111 @@
-# -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
 from odoo import api, fields, models, _
-from odoo.exceptions import AccessError
+from odoo.exceptions import RedirectWarning, UserError
+from odoo.addons.mail.tools.discuss import Store
 
 
-class Partner(models.Model):
-    _inherit = ['res.partner']
+class ResPartner(models.Model):
+    _inherit = 'res.partner'
 
     employee_ids = fields.One2many(
-        'hr.employee', 'address_home_id', string='Employees', groups="hr.group_hr_user",
+        'hr.employee', 'work_contact_id', string='Employees', groups="hr.group_hr_user",
         help="Related employees based on their private address")
     employees_count = fields.Integer(compute='_compute_employees_count', groups="hr.group_hr_user")
-
-    def name_get(self):
-        """ Override to allow an employee to see its private address in his profile.
-            This avoids to relax access rules on `res.parter` and to add an `ir.rule`.
-            (advantage in both security and performance).
-            Use a try/except instead of systematically checking to minimize the impact on performance.
-            """
-        try:
-            return super(Partner, self).name_get()
-        except AccessError as e:
-            if len(self) == 1 and self in self.env.user.employee_ids.mapped('address_home_id'):
-                return super(Partner, self.sudo()).name_get()
-            raise e
+    employee = fields.Boolean(help="Whether this contact is an Employee.", compute='_compute_employee', store=True, readonly=False, copy=False)
 
     def _compute_employees_count(self):
         for partner in self:
-            partner.employees_count = len(partner.employee_ids)
+            partner.employees_count = len(partner.sudo().employee_ids.filtered(lambda e: e.company_id in self.env.companies))
 
     def action_open_employees(self):
         self.ensure_one()
+        if self.employees_count > 1:
+            return {
+                'name': _('Related Employees'),
+                'type': 'ir.actions.act_window',
+                'res_model': 'hr.employee',
+                'view_mode': 'kanban',
+                'domain': [('id', 'in', self.employee_ids.ids),
+                           ('company_id', 'in', self.env.companies.ids)],
+            }
         return {
-            'name': _('Related Employees'),
+            'name': _('Employee'),
             'type': 'ir.actions.act_window',
             'res_model': 'hr.employee',
-            'view_mode': 'kanban,tree,form',
-            'domain': [('id', 'in', self.employee_ids.ids)],
+            'res_id': self.employee_ids.filtered(lambda e: e.company_id in self.env.companies).id,
+            'view_mode': 'form',
         }
 
-class ResPartnerBank(models.Model):
-    _inherit = ['res.partner.bank']
+    def _get_all_addr(self):
+        self.ensure_one()
+        employee_id = self.env['hr.employee'].search(
+            [('id', 'in', self.employee_ids.ids)],
+            limit=1,
+        )
+        if not employee_id:
+            return super()._get_all_addr()
 
-    @api.depends_context('uid')
-    def _compute_display_name(self):
-        account_employee = self.browse()
-        if not self.user_has_groups('hr.group_hr_user'):
-            account_employee = self.sudo().filtered("partner_id.employee_ids")
-            for account in account_employee:
-                account.sudo(self.env.su).display_name = \
-                    account.acc_number[:2] + "*" * len(account.acc_number[2:-4]) + account.acc_number[-4:]
-        super(ResPartnerBank, self - account_employee)._compute_display_name()
+        pstl_addr = {
+            'contact_type': 'employee',
+            'street': employee_id.private_street,
+            'zip': employee_id.private_zip,
+            'city': employee_id.private_city,
+            'country': employee_id.private_country_id.code,
+        }
+        return [pstl_addr] + super()._get_all_addr()
+
+    @api.depends('employee_ids')
+    def _compute_employee(self):
+        employee_data = self.env['hr.employee']._read_group(
+            domain=[('work_contact_id', 'in', self.ids)],
+            groupby=['work_contact_id'],
+        )
+        employees = {employee for [employee] in employee_data}
+        for partner in self:
+            partner.employee = partner in employees
+
+    @api.ondelete(at_uninstall=False)
+    def _unlink_contact_rel_employee(self):
+        partners = self.filtered(lambda partner: partner.sudo().employee_ids)
+        if len(self) == 1 and len(partners) == 1 and self.id == partners[0].id:
+            raise UserError(_('You cannot delete contact that are linked to an employee, please archive them instead.'))
+        if partners:
+            error_msg = _(
+                'You cannot delete contact(s) linked to employee(s).\n'
+                'Please archive them instead.\n\n'
+                'Affected contact(s): %(names)s', names=", ".join([u.name for u in partners]),
+            )
+            action_error = partners._action_show()
+            raise RedirectWarning(error_msg, action_error, _('Go to contact'))
+
+    def _action_show(self):
+        """If self is a singleton, directly access the form view. If it is a recordset, open a list view"""
+        view_id = self.env.ref('base.view_partner_form').id
+        action = {
+            'type': 'ir.actions.act_window',
+            'res_model': 'res.partner',
+            'context': {'create': False},
+        }
+        if len(self) > 1:
+            action.update({
+                'name': _('Contacts'),
+                'view_mode': 'list,form',
+                'views': [[None, 'list'], [view_id, 'form']],
+                'domain': [('id', 'in', self.ids)],
+            })
+        else:
+            action.update({
+                'view_mode': 'form',
+                'views': [[view_id, 'form']],
+                'res_id': self.id,
+            })
+        return action
+
+    def _get_store_avatar_card_fields(self, target):
+        avatar_card_fields = super()._get_store_avatar_card_fields(target)
+        if target.is_internal(self.env):
+            # sudo: res.partner - internal users can access employee information of partner
+            employee_fields = self.sudo().employee_ids._get_store_avatar_card_fields(target)
+            avatar_card_fields.append(Store.Many("employee_ids", employee_fields, mode="ADD", sudo=True))
+        return avatar_card_fields