Yannick Tivisse
f5dfe4727c
The goal is to be coherent with the user property. Actually, company_id and company_ids on the environment are no fields. Calling env.company_id returns a browse record, not an id.
57 lines
2.5 KiB
Python
57 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
from odoo import api, fields, models, _
|
|
|
|
|
|
class ResPartner(models.Model):
|
|
_inherit = 'res.partner'
|
|
|
|
type = fields.Selection(selection_add=[('subcontractor', 'Subcontractor')])
|
|
bom_id = fields.Many2one('mrp.bom', domain=[('type', '=', 'subcontract')])
|
|
|
|
def _get_name(self):
|
|
name = super(ResPartner, self)._get_name()
|
|
if self.type == 'subcontractor':
|
|
subcontractor = _('Subcontractor')
|
|
if name.endswith(', '):
|
|
name = "%s%s" % (name, subcontractor)
|
|
else:
|
|
name = '%s, %s' % (name, subcontractor)
|
|
return name
|
|
|
|
@api.onchange('type')
|
|
def _onchange_partner_type(self):
|
|
if self.type == 'subcontractor':
|
|
company = self.company_id or self.parent_id.company_id
|
|
subcontracting_location = company.subcontracting_location_id
|
|
self.update({
|
|
'property_stock_customer': subcontracting_location,
|
|
'property_stock_supplier': subcontracting_location
|
|
})
|
|
else:
|
|
self.update({
|
|
'property_stock_customer': self.env.ref('stock.stock_location_customers'),
|
|
'property_stock_supplier': self.env.ref('stock.stock_location_suppliers')
|
|
})
|
|
|
|
@api.model_create_multi
|
|
def create(self, vals_list):
|
|
for vals in vals_list:
|
|
if vals.get('type') == 'subcontractor':
|
|
if not vals.get('property_stock_supplier') and not vals.get('property_stock_customer'):
|
|
company = vals.get('company_id') and self.env['res.company'].browse(vals['company_id']) or \
|
|
self.browse(vals.get('parent_id')).company_id
|
|
subcontracting_location = company.subcontracting_location_id
|
|
vals['property_stock_supplier'] = subcontracting_location
|
|
vals['property_stock_customer'] = subcontracting_location
|
|
return super(ResPartner, self).create(vals_list)
|
|
|
|
def write(self, vals):
|
|
if vals.get('type') == 'subcontractor':
|
|
subcontracting_location = self.env.company.subcontracting_location_id
|
|
if not vals.get('property_stock_supplier') and not vals.get('property_stock_customer'):
|
|
vals['property_stock_supplier'] = subcontracting_location.id
|
|
vals['property_stock_customer'] = subcontracting_location.id
|
|
return super(ResPartner, self).write(vals)
|