# Copyright 2014-2018 Therp BV . # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html). # pylint: disable=no-member import logging from lxml import etree from flectra import api, fields, models _logger = logging.getLogger(__name__) # pylint: disable=invalid-name class ResPartner(models.Model): _inherit = 'res.partner' @api.multi def browse(self, arg=None, prefetch=None): for tab in self._get_tabs(): fieldname = tab.get_fieldname() if fieldname not in self._fields: # Check this for performance reasons. self.add_field(tab) return super(ResPartner, self).browse(arg=arg, prefetch=prefetch) @api.model def fields_view_get( self, view_id=None, view_type='form', toolbar=False, submenu=False): """Override to add relation tabs to form.""" result = super(ResPartner, self).fields_view_get( view_id=view_id, view_type=view_type, toolbar=toolbar, submenu=submenu) if view_type != 'form' or self.env.context.get('check_view_ids'): return result view = etree.fromstring(result['arch']) extra_fields = self._add_tab_pages(view) view_model = self.env['ir.ui.view'] result['arch'], original_fields = view_model.postprocess_and_fields( self._name, view, result['view_id']) for fieldname in extra_fields: result['fields'][fieldname] = original_fields[fieldname] return result def _add_tab_pages(self, view): """Adds the relevant tabs to the partner's formview.""" # pylint: disable=no-member def add_invisible_extra_field(view, extra_fields, fieldname): """Add invisible field to view.""" view.append( etree.Element('field', name=fieldname, invisible='True')) extra_fields.append(fieldname) last_page_nodes = view.xpath('//page[last()]') if not last_page_nodes: # Nothing to do if form contains no pages/tabs. return [] extra_fields = [] if not view.xpath('//field[@name="id"]'): add_invisible_extra_field(view, extra_fields, 'id') last_page = last_page_nodes[0] for tab in self._get_tabs(): # get all tabs self.add_field(tab) add_invisible_extra_field( view, extra_fields, tab.get_visible_fieldname()) extra_fields.append(tab.get_fieldname()) tab_page = tab.create_page() last_page.addnext(tab_page) last_page = tab_page # Keep ordering of tabs return extra_fields @api.depends('is_company', 'category_id') def _compute_tabs_visibility(self): """Compute for all tabs wether they should be visible.""" for tab in self._get_tabs(): # get all tabs for this in self: this[tab.get_visible_fieldname()] = \ tab.compute_visibility(this) def _get_tabs(self): tab_model = self.env['res.partner.tab'] return tab_model.get_tabs() def add_field(self, tab): """Add tab field to model. Will replace existing field if already present. """ # Visible field determines wether first field will be visible. # This is because domains on many2many no longer work in 9.0 # and above. visible_field = fields.Boolean(compute='_compute_tabs_visibility') self._add_field(tab.get_visible_fieldname(), visible_field) if visible_field not in self._field_computed: self._field_computed[visible_field] = [visible_field] tab_field = fields.One2many( comodel_name='res.partner.relation.all', inverse_name='this_partner_id', domain=[('tab_id', '=', tab.tab_record.id)], string=tab.tab_record.name) self._add_field(tab.get_fieldname(), tab_field)