|
|
# -*- coding: utf-8 -*-# © 2016 Antonio Espinosa <antonio.espinosa@tecnativa.com># License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from openerp import models, fields, api
class AccountMove(models.Model): _inherit = 'account.move'
move_type = fields.Selection( string="Move type", selection=[ ('other', 'Other'), ('liquidity', 'Liquidity'), ('receivable', 'Receivable'), ('receivable_refund', 'Receivable refund'), ('payable', 'Payable'), ('payable_refund', 'Payable refund'), ], compute='_compute_move_type', store=True, readonly=True)
@api.multi @api.depends('line_ids.account_id.internal_type', 'line_ids.balance') def _compute_move_type(self): sequence = ( ("liquidity", lambda balance: "liquidity"), ("payable", lambda balance: ('payable' if balance < 0 else 'payable_refund')), ("receivable", lambda balance: ('receivable' if balance > 0 else 'receivable_refund')), (False, lambda balance: "other"), ) chunked_self = ( self[i:i + models.PREFETCH_MAX] for i in range(0, len(self), models.PREFETCH_MAX) ) for chunk in chunked_self: move_ids = set(chunk.ids) for internal_type, criteria in sequence: if not move_ids: break # Find balances of the expected type for this move domain = [ ("move_id", "in", list(move_ids)), ] if internal_type: domain += [ ("account_id.internal_type", "=", internal_type), ] balances = self.env["account.move.line"].read_group( domain=domain, fields=["move_id", "balance"], groupby=["move_id"], ) for balance in balances: move = self.browse(balance["move_id"][0]) # Discard the move for next searches move_ids.discard(move.id) # Update its type move.move_type = criteria(balance["balance"])
|
