yaltik_odoo_custom/yaltik_crm_action/models/res_partner.py

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

#    Copyright 2017 Fabien Bourgeois <fabien@yaltik.com>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.


""" Partner adaptations """

from odoo import models, fields, api

class ResPartner(models.Model):
    """ Partner adaptations """
    _inherit = 'res.partner'

    action_count = fields.Char(compute='_compute_action_count')

    @api.multi
    def _compute_action_count(self):
        for partner in self:
            domain = [('partner_id', '=', partner.id)]
            action_count = self.env['crm.action'].search_count(domain)
            if action_count:
                domain.append(('state', '=', 'draft'))
                draft_count = self.env['crm.action'].search_count(domain)
            else:
                draft_count = 0
            partner.action_count = u'{} / {}'.format(draft_count, action_count)