Initial commit: Pos packages

odoo-bringout-oca-ocb-pos_restaurant/pos_restaurant/models/pos_config.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models, _
+
+
+class PosConfig(models.Model):
+    _inherit = 'pos.config'
+
+    iface_splitbill = fields.Boolean(string='Bill Splitting', help='Enables Bill Splitting in the Point of Sale.')
+    iface_printbill = fields.Boolean(string='Bill Printing', help='Allows to print the Bill before payment.')
+    iface_orderline_notes = fields.Boolean(string='Internal Notes', help='Allow custom internal notes on Orderlines.')
+    floor_ids = fields.One2many('restaurant.floor', 'pos_config_id', string='Restaurant Floors', help='The restaurant floors served by this point of sale.')
+    printer_ids = fields.Many2many('restaurant.printer', 'pos_config_printer_rel', 'config_id', 'printer_id', string='Order Printers')
+    is_table_management = fields.Boolean('Floors & Tables')
+    is_order_printer = fields.Boolean('Order Printer')
+    set_tip_after_payment = fields.Boolean('Set Tip After Payment', help="Adjust the amount authorized by payment terminals to add a tip after the customers left or at the end of the day.")
+    module_pos_restaurant = fields.Boolean(default=True)
+
+    def _force_http(self):
+        enforce_https = self.env['ir.config_parameter'].sudo().get_param('point_of_sale.enforce_https')
+        if not enforce_https and self.printer_ids.filtered(lambda pt: pt.printer_type == 'epson_epos'):
+            return True
+        return super(PosConfig, self)._force_http()
+
+    def get_tables_order_count(self):
+        """         """
+        self.ensure_one()
+        floors = self.env['restaurant.floor'].search([('pos_config_id', 'in', self.ids)])
+        tables = self.env['restaurant.table'].search([('floor_id', 'in', floors.ids)])
+        domain = [('state', '=', 'draft'), ('table_id', 'in', tables.ids)]
+
+        order_stats = self.env['pos.order'].read_group(domain, ['table_id'], 'table_id')
+        orders_map = dict((s['table_id'][0], s['table_id_count']) for s in order_stats)
+
+        result = []
+        for table in tables:
+            result.append({'id': table.id, 'orders': orders_map.get(table.id, 0)})
+        return result
+
+    def _get_forbidden_change_fields(self):
+        forbidden_keys = super(PosConfig, self)._get_forbidden_change_fields()
+        forbidden_keys.append('is_table_management')
+        forbidden_keys.append('floor_ids')
+        return forbidden_keys
+
+    def write(self, vals):
+        if ('is_table_management' in vals and vals['is_table_management'] == False):
+            vals['floor_ids'] = [(5, 0, 0)]
+        if ('is_order_printer' in vals and vals['is_order_printer'] == False):
+            vals['printer_ids'] = [(5, 0, 0)]
+        return super(PosConfig, self).write(vals)
+
+    @api.model
+    def add_cash_payment_method(self):
+        companies = self.env['res.company'].search([])
+        for company in companies.filtered('chart_template_id'):
+            pos_configs = self.search([('company_id', '=', company.id), ('module_pos_restaurant', '=', True)])
+            journal_counter = 2
+            for pos_config in pos_configs:
+                if pos_config.payment_method_ids.filtered('is_cash_count'):
+                    continue
+                cash_journal = self.env['account.journal'].search([('company_id', '=', company.id), ('type', '=', 'cash'), ('pos_payment_method_ids', '=', False)], limit=1)
+                if not cash_journal:
+                    cash_journal = self.env['account.journal'].create({
+                        'name': _('Cash %s', journal_counter),
+                        'code': 'RCSH%s' % journal_counter,
+                        'type': 'cash',
+                        'company_id': company.id
+                    })
+                    journal_counter += 1
+                payment_methods = pos_config.payment_method_ids
+                payment_methods |= self.env['pos.payment.method'].create({
+                    'name': _('Cash Bar'),
+                    'journal_id': cash_journal.id,
+                    'company_id': company.id,
+                })
+                pos_config.write({'payment_method_ids': [(6, 0, payment_methods.ids)]})