from odoo import fields, models


class DanceWhatsappTemplate(models.Model):
    _inherit = "dance.whatsapp.template"

    trigger = fields.Selection(
        selection_add=[
            ("event_notification", "Event Notification"),
            ("rehearsal_reminder", "Rehearsal Reminder"),
            ("costume_update", "Costume Update"),
        ],
        ondelete={
            "event_notification": "cascade",
            "rehearsal_reminder": "cascade",
            "costume_update": "cascade",
        },
    )


class DanceWhatsappLog(models.Model):
    _inherit = "dance.whatsapp.log"

    trigger = fields.Selection(
        selection_add=[
            ("event_notification", "Event Notification"),
            ("rehearsal_reminder", "Rehearsal Reminder"),
            ("costume_update", "Costume Update"),
        ],
        ondelete={
            "event_notification": "cascade",
            "rehearsal_reminder": "cascade",
            "costume_update": "cascade",
        },
    )
    event_id = fields.Many2one("dance.event", string="Event", ondelete="set null")
    performance_id = fields.Many2one(
        "dance.event.performance",
        string="Performance",
        ondelete="set null",
    )
    rehearsal_id = fields.Many2one("dance.rehearsal", string="Rehearsal", ondelete="set null")