Odoo18-Base/addons/event_sale/models/sale_order.py
2025-01-06 10:57:38 +07:00

95 lines
4.5 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from werkzeug.urls import url_encode, url_join
from odoo import fields, models, _
from odoo.exceptions import ValidationError
from odoo.osv import expression
class SaleOrder(models.Model):
_inherit = "sale.order"
attendee_count = fields.Integer('Attendee Count', compute='_compute_attendee_count')
def write(self, vals):
""" Synchronize partner from SO to registrations. This is done notably
in website_sale controller shop/address that updates customer, but not
only. """
result = super(SaleOrder, self).write(vals)
if any(line.service_tracking == 'event' for line in self.order_line) and vals.get('partner_id'):
registrations_toupdate = self.env['event.registration'].sudo().search([('sale_order_id', 'in', self.ids)])
registrations_toupdate.write({'partner_id': vals['partner_id']})
return result
def action_confirm(self):
unconfirmed_registrations = self.order_line.registration_ids.filtered(
lambda reg: reg.state in ["draft", "cancel"]
)
res = super(SaleOrder, self).action_confirm()
unconfirmed_registrations._update_mail_schedulers()
for so in self:
if not any(line.service_tracking == 'event' for line in so.order_line):
continue
so_lines_missing_events = so.order_line.filtered(lambda line: line.service_tracking == 'event' and not line.event_id)
if so_lines_missing_events:
so_lines_descriptions = "".join(f"\n- {so_line_description.name}" for so_line_description in so_lines_missing_events)
raise ValidationError(_("Please make sure all your event related lines are configured before confirming this order:%s", so_lines_descriptions))
# Initialize registrations
so.order_line._init_registrations()
if len(self) == 1:
return self.env['ir.actions.act_window'].with_context(
default_sale_order_id=so.id
)._for_xml_id('event_sale.action_sale_order_event_registration')
return res
def action_view_attendee_list(self):
action = self.env["ir.actions.actions"]._for_xml_id("event.event_registration_action_tree")
action['domain'] = [('sale_order_id', 'in', self.ids)]
return action
def _compute_attendee_count(self):
sale_orders_data = self.env['event.registration']._read_group(
[('sale_order_id', 'in', self.ids),
('state', '!=', 'cancel')],
['sale_order_id'], ['__count'],
)
attendee_count_data = {
sale_order.id: count for sale_order, count in sale_orders_data
}
for sale_order in self:
sale_order.attendee_count = attendee_count_data.get(sale_order.id, 0)
def _get_product_catalog_domain(self):
"""Override of `_get_product_catalog_domain` to extend the domain.
:returns: A list of tuples that represents a domain.
:rtype: list
"""
domain = super()._get_product_catalog_domain()
return expression.AND([domain, [('service_tracking', '!=', 'event')]])
def _notify_get_recipients_groups(self, message, model_description, msg_vals=None):
groups = super()._notify_get_recipients_groups(message, model_description, msg_vals)
if not self or self.state != 'sale' or not self.order_line.registration_ids:
return groups
customer_portal_group = next((group for group in groups if group[0] == 'portal_customer'), None)
if not customer_portal_group:
return groups
if customer_portal_group[2]['has_button_access']:
actions_opt = customer_portal_group[2].setdefault('actions', [])
has_single_event = len(self.order_line.event_id) == 1
registrations = self.order_line.registration_ids
for event, event_registrations in registrations.grouped('event_id').items():
actions_opt.append({
'url': url_join(event.get_base_url(), f'/event/{event.id}/my_tickets?' + url_encode({
'registration_ids': str(event_registrations.ids),
'tickets_hash': event._get_tickets_access_hash(event_registrations.ids),
})),
'title': _("Get Your Tickets") if has_single_event else _("%(event_name)s - Tickets", event_name=event.name)
})
return groups