vertical-cooperative/easy_my_coop/wizard/update_share_line.py

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

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


class PartnerUpdateInfo(models.TransientModel):
    _name = "share.line.update.info"

    @api.model
    def _get_share_line(self):
        active_id = self.env.context.get('active_id')
        return self.env['share.line'].browse(active_id)

    @api.model
    def _get_effective_date(self):
        share_line = self._get_share_line()

        return share_line.effective_date

    effective_date = fields.Date(string="effective date",
                                 required=True,
                                 default=_get_effective_date)
    cooperator = fields.Many2one(related='share_line.partner_id',
                                 string="Cooperator")
    share_line = fields.Many2one('share.line',
                                 string="Share line",
                                 default=_get_share_line)

    @api.multi
    def update(self):

        line = self.share_line
        cooperator = line.partner_id

        sub_reg = self.env['subscription.register'].search(
            [('partner_id', '=', cooperator.id),
             ('share_product_id', '=', line.share_product_id.id),
             ('quantity', '=', line.share_number),
             ('date', '=', line.effective_date)])
        if sub_reg:
            if len(sub_reg) > 1:
                raise UserError(_("Error the update return more than one"
                                  " subscription register lines."))
            else:
                line.effective_date = self.effective_date
                sub_reg.date = self.effective_date
        return True