vertical-cooperative/easy_my_coop/wizard/update_partner_info.py

# -*- coding: utf-8 -*-

from openerp import api, fields, models, _
from openerp.exceptions import UserError


class PartnerUpdateInfo(models.TransientModel):
    _name = "partner.update.info"
    _description = "Update Partner Info"

    @api.model
    def _get_partner(self):
        active_id = self.env.context.get('active_id')
        return self.env['res.partner'].browse(active_id)

    @api.model
    def _get_register_number(self):
        partner = self._get_partner()
        if partner.is_company:
            return partner.company_register_number
        else:
            return partner.national_register_number

    register_number = fields.Char(string="Register Number",
                                  required=True,
                                  default=_get_register_number)
    cooperator = fields.Many2one('res.partner',
                                 string="Cooperator",
                                 default=_get_partner)

    def check_belgian_ident_id(self, register_number):
        if self.env['subscription.request'].check_belgian_identification_id(
                                                register_number):
            return True
        else:
            raise UserError(_("The national register number is not valid."))

    @api.multi
    def update(self):

        cooperator = self.cooperator
        coop_vals = {}

        if cooperator.is_company:
            coop_vals['company_register_number'] = self.register_number
        else:
            if self.check_belgian_ident_id(self.register_number):
                coop_vals['national_register_number'] = self.register_number

        if coop_vals:
            cooperator.write(coop_vals)

        return True