19.0 vanilla

odoo-bringout-oca-ocb-pos_hr/pos_hr/models/pos_session.py

@@ -1,43 +1,102 @@
 # -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
-
-from odoo import models
+from odoo import fields, models, api, _
+from odoo.tools import plaintext2html
 
 
 class PosSession(models.Model):
     _inherit = 'pos.session'
+    employee_id = fields.Many2one(
+        "hr.employee",
+        string="Cashier",
+        help="The employee who currently uses the cash register",
+        tracking=True,
+    )
 
-    def _pos_data_process(self, loaded_data):
-        super()._pos_data_process(loaded_data)
-        if self.config_id.module_pos_hr:
-            loaded_data['employee_by_id'] = {employee['id']: employee for employee in loaded_data['hr.employee']}
+    @api.model
+    def _load_pos_data_models(self, config):
+        data = super()._load_pos_data_models(config)
+        if config.module_pos_hr:
+            data += ['hr.employee']
+        return data
 
-    def _pos_ui_models_to_load(self):
-        result = super()._pos_ui_models_to_load()
-        if self.config_id.module_pos_hr:
-            new_model = 'hr.employee'
-            if new_model not in result:
-                result.append(new_model)
-        return result
+    def _set_opening_control_data(self, cashbox_value: int, notes: str):
+        super()._set_opening_control_data(cashbox_value, notes)
+        if author_id := self._get_message_author():
+            self.message_post(body=plaintext2html(_('Opened register')), author_id=author_id.id)
 
-    def _loader_params_hr_employee(self):
-        if len(self.config_id.employee_ids) > 0:
-            domain = ['&', ('company_id', '=', self.config_id.company_id.id), '|', ('user_id', '=', self.user_id.id), ('id', 'in', self.config_id.employee_ids.ids)]
+    def post_close_register_message(self):
+        if author_id := self._get_message_author():
+            self.message_post(body=plaintext2html(_('Closed Register')), author_id=author_id.id)
         else:
-            domain = [('company_id', '=', self.config_id.company_id.id)]
-        return {'search_params': {'domain': domain, 'fields': ['name', 'id', 'user_id'], 'load': False}}
+            return super().post_close_register_message()
 
-    def _get_pos_ui_hr_employee(self, params):
-        employees = self.env['hr.employee'].search_read(**params['search_params'])
-        employee_ids = [employee['id'] for employee in employees]
-        user_ids = [employee['user_id'] for employee in employees if employee['user_id']]
-        manager_ids = self.env['res.users'].browse(user_ids).filtered(lambda user: self.config_id.group_pos_manager_id in user.groups_id).mapped('id')
+    def _get_message_author(self):
+        if not self.employee_id:
+            return None
+        
+        if related_partners := self.employee_id._get_related_partners():
+            return related_partners[0]
+        
+        return self.user_id.partner_id
 
-        employees_barcode_pin = self.env['hr.employee'].browse(employee_ids).get_barcodes_and_pin_hashed()
-        bp_per_employee_id = {bp_e['id']: bp_e for bp_e in employees_barcode_pin}
-        for employee in employees:
-            employee['role'] = 'manager' if employee['user_id'] and employee['user_id'] in manager_ids else 'cashier'
-            employee['barcode'] = bp_per_employee_id[employee['id']]['barcode']
-            employee['pin'] = bp_per_employee_id[employee['id']]['pin']
+    def _aggregate_payments_amounts_by_employee(self, all_payments):
+        payments_by_employee = []
 
-        return employees
+        for employee, payments_group in all_payments.grouped('employee_id').items():
+            payments_by_employee.append({
+                'id': employee.id if employee else 'others',
+                'name': employee.name if employee else _('Others'),
+                'amount': sum(payments_group.mapped('amount')),
+            })
+
+        # Sort such that "Others" is always the last item
+        return sorted(
+            payments_by_employee,
+            key=lambda p: (p['id'] == 'others', p['name'])
+        )
+
+    def _aggregate_moves_by_employee(self):
+        moves_per_employee = {}
+        for employee, moves in self.sudo().statement_line_ids.grouped('employee_id').items():
+            moves_per_employee[employee.id] = {
+                'id': employee.id,
+                'name': employee.name,
+                'amount': sum(moves.mapped('amount')),
+            }
+
+        return sorted(moves_per_employee.values(), key=lambda p: -p['amount'])
+
+    def get_closing_control_data(self):
+        data = super().get_closing_control_data()
+
+        orders = self._get_closed_orders()
+        payments = orders.payment_ids.filtered(lambda p: p.payment_method_id.type != "pay_later")
+        cash_payment_method_ids = self.payment_method_ids.filtered(lambda pm: pm.type == 'cash')
+        default_cash_payment_method_id = cash_payment_method_ids[0] if cash_payment_method_ids else None
+        default_cash_payments = payments.filtered(lambda p: p.payment_method_id == default_cash_payment_method_id) if default_cash_payment_method_id else self.env['pos.payment']
+        non_cash_payment_method_ids = self.payment_method_ids - default_cash_payment_method_id if default_cash_payment_method_id else self.payment_method_ids
+        non_cash_payments_grouped_by_method_id = {pm.id: orders.payment_ids.filtered(lambda p: p.payment_method_id == pm) for pm in non_cash_payment_method_ids}
+
+        data['default_cash_details']['amount_per_employee'] = self._aggregate_payments_amounts_by_employee(default_cash_payments)
+        for payment_method in data['non_cash_payment_methods']:
+            payment_method['amount_per_employee'] = self._aggregate_payments_amounts_by_employee(non_cash_payments_grouped_by_method_id[payment_method['id']])
+
+        data['default_cash_details']['moves_per_employee'] = self._aggregate_moves_by_employee()
+
+        return data
+
+    def _prepare_account_bank_statement_line_vals(self, session, sign, amount, reason, partner_id, extras):
+        vals = super()._prepare_account_bank_statement_line_vals(session, sign, amount, reason, partner_id, extras)
+        if extras.get('employee_id'):
+            vals['employee_id'] = extras['employee_id']
+        return vals
+
+    def get_cash_in_out_list(self):
+        cash_in_out_list = super().get_cash_in_out_list()
+        if self.config_id.module_pos_hr:
+            for cash_in_out in cash_in_out_list:
+                cash_move = self.env['account.bank.statement.line'].browse(cash_in_out['id'])
+                if cash_move.employee_id:
+                    cash_in_out['cashier_name'] = cash_move.partner_id.name
+        return cash_in_out_list