from odoo import models, fields, api, _


class SurveySurvey(models.Model):
    _inherit = 'survey.survey'

    tot_selected_survey = fields.Integer("Number of selected surveys", compute="_get_selected_input")

    @api.depends('user_input_ids', 'user_input_ids.selected')
    def _get_selected_input(self):
        selected_survey = self.env['survey.user_input'].search([
            ('survey_id', 'in', self.ids),
            ('selected', '=', True)
        ])
        for survey in self:
            survey.tot_selected_survey = len(selected_survey.filtered(lambda user_input: user_input.survey_id == survey))

    # ACTIONS

    @api.multi
    def action_survey_user_input(self):
        action = super(SurveySurvey, self).action_survey_user_input()
        if self.env.context.get('search_default_selected', False):
            action['display_name'] += _(" selected")
        return action