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


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

    patient_ids = fields.Many2many(comodel_name='sports.patient',
                                   relation='sports_team_patient_rel',
                                   column1='team_id',
                                   column2='patient_id',
                                   string='Players')
    type = fields.Selection(selection_add=[('team', 'Sports Team'),
                                           ('treatment_pro', 'Treatment Professional'),])
    staff_partner_ids = fields.Many2many(comodel_name='res.partner',
                                         relation='sports_team_staff_partner_rel',
                                         column1='team_id',
                                         column2='staff_partner_id',
                                         string='Staff')

    def write(self, vals):
        # Override to validate changes to the 'type' field
        treatment_pros = self.filtered(lambda r: r.type == 'treatment_pro')
        if treatment_pros and 'type' in vals and vals['type'] != 'treatment_pro':
            injuries = self.env['sports.patient.injury'].search_count([('treatment_professional_ids', 'in', treatment_pros.ids)])
            if injuries:
                raise UserError(_('Partner type cannot be changed since this treatment '
                                  'professional appears on a patient injury record.'))
        teams = self.filtered(lambda r: r.type == 'team')
        if teams and 'type' in vals and vals['type'] != 'team':
            if teams.mapped('staff_partner_ids') or teams.mapped('patient_ids'):
                raise UserError(_('Sports team is related to patients and/or staff. '
                                  'Its type cannot be changed until these relations are '
                                  'removed.'))
        return super().write(vals)

    def unlink(self):
        treatment_pros = self.filtered(lambda r: r.type == 'treatment_pro')
        if treatment_pros:
            injuries = self.env['sports.patient.injury'].search_count([('treatment_professional_ids', 'in', treatment_pros.ids)])
            if injuries:
                raise UserError(_('Treatment professional cannot be deleted since they '
                                  'appears on a patient injury record.'))
        teams = self.filtered(lambda r: r.type == 'team')
        if teams:
            if teams.mapped('staff_partner_ids') or teams.mapped('patient_ids'):
                raise UserError(_('Sports team is related to patients and/or staff. '
                                  'Its type cannot be deleted until these relations are '
                                  'removed. You may try archiving it instead.'))
        return super().unlink()