31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from odoo import fields, models
|
||
|
|
|
||
|
|
|
||
|
|
class BusPresence(models.Model):
|
||
|
|
_inherit = ['bus.presence']
|
||
|
|
|
||
|
|
guest_id = fields.Many2one('mail.guest', 'Guest', ondelete='cascade')
|
||
|
|
|
||
|
|
def init(self):
|
||
|
|
self.env.cr.execute("CREATE UNIQUE INDEX IF NOT EXISTS bus_presence_guest_unique ON %s (guest_id) WHERE guest_id IS NOT NULL" % self._table)
|
||
|
|
|
||
|
|
_sql_constraints = [
|
||
|
|
("partner_or_guest_exists", "CHECK((user_id IS NOT NULL AND guest_id IS NULL) OR (user_id IS NULL AND guest_id IS NOT NULL))", "A bus presence must have a user or a guest."),
|
||
|
|
]
|
||
|
|
|
||
|
|
def _get_bus_target(self):
|
||
|
|
return self.guest_id or super()._get_bus_target()
|
||
|
|
|
||
|
|
def _get_identity_field_name(self):
|
||
|
|
return "guest_id" if self.guest_id else super()._get_identity_field_name()
|
||
|
|
|
||
|
|
def _get_identity_data(self):
|
||
|
|
self.ensure_one()
|
||
|
|
return {"guest_id": self.guest_id.id} if self.guest_id else super()._get_identity_data()
|
||
|
|
|
||
|
|
def _invalidate_im_status(self, fnames=None, flush=True):
|
||
|
|
super()._invalidate_im_status()
|
||
|
|
self.guest_id.invalidate_recordset(["im_status"])
|