Odoo18-Base/addons/point_of_sale/models/res_partner.py

68 lines
2.7 KiB
Python
Raw Permalink Normal View History

2025-03-10 11:12:23 +07:00
# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import api, fields, models, _
from odoo.exceptions import UserError
class ResPartner(models.Model):
_inherit = 'res.partner'
pos_order_count = fields.Integer(
compute='_compute_pos_order',
help="The number of point of sales orders related to this customer",
groups="point_of_sale.group_pos_user",
)
pos_order_ids = fields.One2many('pos.order', 'partner_id', readonly=True)
def _compute_pos_order(self):
# retrieve all children partners and prefetch 'parent_id' on them
all_partners = self.with_context(active_test=False).search([('id', 'child_of', self.ids)])
all_partners.read(['parent_id'])
pos_order_data = self.env['pos.order']._read_group(
domain=[('partner_id', 'in', all_partners.ids)],
fields=['partner_id'], groupby=['partner_id']
)
self.pos_order_count = 0
for group in pos_order_data:
partner = self.browse(group['partner_id'][0])
while partner:
if partner in self:
partner.pos_order_count += group['partner_id_count']
partner = partner.parent_id
def action_view_pos_order(self):
'''
This function returns an action that displays the pos orders from partner.
'''
action = self.env['ir.actions.act_window']._for_xml_id('point_of_sale.action_pos_pos_form')
if self.is_company:
action['domain'] = [('partner_id.commercial_partner_id', '=', self.id)]
else:
action['domain'] = [('partner_id', '=', self.id)]
return action
@api.model
def create_from_ui(self, partner):
""" create or modify a partner from the point of sale ui.
partner contains the partner's fields. """
# image is a dataurl, get the data after the comma
if partner.get('image_1920'):
partner['image_1920'] = partner['image_1920'].split(',')[1]
partner_id = partner.pop('id', False)
if partner_id: # Modifying existing partner
self.browse(partner_id).write(partner)
else:
partner_id = self.create(partner).id
return partner_id
@api.ondelete(at_uninstall=False)
def _unlink_except_active_pos_session(self):
running_sessions = self.env['pos.session'].sudo().search([('state', '!=', 'closed')])
if running_sessions:
raise UserError(
_("You cannot delete contacts while there are active PoS sessions. Close the session(s) %s first.")
% ", ".join(session.name for session in running_sessions)
)