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

from odoo import _, api, fields, models

class LoyaltyRule(models.Model):
    _inherit = 'loyalty.rule'

    @api.model_create_multi
    def create(self, vals_list):
        records = super().create(vals_list)
        for record in records:
            if record.program_id and record.program_id.state in ['pending', 'approved']:
                record.program_id.write({'state': 'draft'})
                record.program_id.message_post(body=_("A loyalty rule was added, resetting the program to Draft status for re-approval."))
        return records

    def write(self, vals):
        programs_to_reset = self.env['loyalty.program']
        for record in self:
            if record.program_id and record.program_id.state in ['pending', 'approved']:
                programs_to_reset |= record.program_id
        
        res = super().write(vals)
        
        if 'program_id' in vals:
            new_program = self.env['loyalty.program'].browse(vals['program_id'])
            if new_program and new_program.state in ['pending', 'approved']:
                programs_to_reset |= new_program
                
        for program in programs_to_reset:
            program.write({'state': 'draft'})
            program.message_post(body=_("A loyalty rule was modified, resetting the program to Draft status for re-approval."))
            
        return res

    def unlink(self):
        programs_to_reset = self.env['loyalty.program']
        for record in self:
            if record.program_id and record.program_id.state in ['pending', 'approved']:
                programs_to_reset |= record.program_id
                
        res = super().unlink()
        
        for program in programs_to_reset:
            program.write({'state': 'draft'})
            program.message_post(body=_("A loyalty rule was deleted, resetting the program to Draft status for re-approval."))
            
        return res