initial commit 2

1 files changed, 44 insertions, 0 deletions

diff --git a/addons/point_of_sale/models/res_partner.py b/addons/point_of_sale/models/res_partner.py
new file mode 100644
index 00000000..a0246c6f
--- /dev/null
+++ b/addons/point_of_sale/models/res_partner.py
@@ -0,0 +1,44 @@
+# -*- 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):
+        partners_data = self.env['pos.order'].read_group([('partner_id', 'in', self.ids)], ['partner_id'], ['partner_id'])
+        mapped_data = dict([(partner['partner_id'][0], partner['partner_id_count']) for partner in partners_data])
+        for partner in self:
+            partner.pos_order_count = mapped_data.get(partner.id, 0)
+
+    @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
+
+    def unlink(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)
+            )
+        return super(ResPartner, self).unlink()