[MIG] pos_remove_pos_category: migration 18.0

Author: thibaultrey

pos_remove_pos_category/__manifest__.py

@@ -11,10 +11,14 @@
     ],
     "website": "https://github.com/OCA/pos",
     "data": [
-        "views/assets.xml",
         "views/pos_view.xml",
         "views/pos_category_view.xml",
     ],
+    "assets": {
+        "point_of_sale._assets_pos": [
+            "/pos_remove_pos_category/static/src/overrides/components/**"
+        ],
+    },
     "installable": True,
     "license": "AGPL-3",
 }

pos_remove_pos_category/models/__init__.py

@@ -1,2 +1,6 @@
 from . import module
 from . import product
+from . import pos_session
+from . import product_category
+from . import res_config_settings
+from . import pos_config

pos_remove_pos_category/models/module.py

@@ -10,28 +10,6 @@ class Module(models.Model):
     def module_uninstall(self):
         for module in self:
             if module.name == "pos_remove_pos_category":
-                # As we have loose previous POS categs restore them
-                # in a sane empty state
-
-                self._cr.execute("UPDATE product_template SET pos_categ_id=NULL")
-                # And restore original constraint
-                self._cr.execute(
-                    """
-                    ALTER TABLE product_template
-                    DROP CONSTRAINT IF EXISTS
-                    product_template_pos_categ_id_fkey
-                """
-                )
-
-                self._cr.execute(
-                    """
-                    ALTER TABLE product_template ADD CONSTRAINT
-                    "product_template_pos_categ_id_fkey"
-                    FOREIGN KEY (pos_categ_id)
-                    REFERENCES pos_category(id) ON DELETE SET NULL;
-                """
-                )
-
                 # Restore POS category menu action
                 # in SQL because pool/env is not available here
                 self._cr.execute(

pos_remove_pos_category/models/pos_config.py

@@ -0,0 +1,10 @@
+# Copyright (C) 2015-TODAY Akretion (<http://www.akretion.com>).
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+from odoo import fields, models
+
+
+class PosConfig(models.Model):
+    _inherit = "pos.config"
+
+    iface_available_categ_ids = fields.Many2many("product.category")

pos_remove_pos_category/models/pos_session.py

@@ -0,0 +1,27 @@
+# Copyright (C) 2015-TODAY Akretion (<http://www.akretion.com>).
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+from odoo import models
+
+
+class PosSession(models.Model):
+    _inherit = "pos.session"
+
+    def load_data(self, models_to_load, only_data=False):
+        res = super().load_data(models_to_load, only_data=only_data)
+        res["pos.category"] = res.pop("product.category")
+        return res
+
+    def _load_pos_data_models(self, config_id):
+        res = super()._load_pos_data_models(config_id)
+        res.append("product.category")
+        return res
+
+    def _load_pos_data_relations(self, model, response):
+        super()._load_pos_data_relations(model, response)
+        if model == "product.category":
+            response[model]["relations"]["parent_id"]["relation"] = "pos.category"
+            response[model]["relations"]["child_ids"]["relation"] = "pos.category"
+        if model == "product.product":
+            response[model]["relations"]["pos_categ_ids"]["relation"] = "pos.category"
+        return

pos_remove_pos_category/models/product.py

@@ -7,32 +7,40 @@
 class ProductTemplate(models.Model):
     _inherit = "product.template"
 
-    pos_categ_id = fields.Many2one(
+    pos_categ_ids = fields.Many2many(
         "product.category",
         store=False,
-        related="categ_id",
-        search="_search_pos_categ_id",
+        compute="_compute_pos_categ_ids",
+        search="_search_pos_categ_ids",
     )
 
-    def _search_pos_categ_id(self, operator, value):
+    def _compute_pos_categ_ids(self):
+        for product in self:
+            if product.categ_id and product.categ_id.available_in_pos:
+                product.pos_categ_ids = [(6, 0, [product.categ_id.id])]
+            else:
+                product.pos_categ_ids = False
+
+    def _search_pos_categ_ids(self, operator, value):
         return [("categ_id", operator, value)]
 
-    @api.model
-    def create(self, vals):
-        if "categ_id" in vals:
-            vals["pos_categ_id"] = vals["categ_id"]
-        return super().create(vals)
+    @api.model_create_multi
+    def create(self, vals_list):
+        for vals in vals_list:
+            if "categ_id" in vals:
+                vals["pos_categ_ids"] = [(6, 0, [vals["categ_id"]])]
+        return super().create(vals_list)
 
     def write(self, vals):
-        if "pos_categ_id" in vals and not vals["pos_categ_id"]:
-            del vals["pos_categ_id"]
+        if "pos_categ_ids" in vals and not vals["pos_categ_ids"]:
+            del vals["pos_categ_ids"]
         return super().write(vals)
 
 
 class ProductCategory(models.Model):
     _inherit = "product.category"
 
-    image_128 = fields.Image("Image 128", max_width=128, max_height=128)
+    image_128 = fields.Image(max_width=128, max_height=128)
 
     available_in_pos = fields.Boolean(
         string="Available in the Point of Sale",

pos_remove_pos_category/models/product_category.py

@@ -0,0 +1,49 @@
+# Copyright (C) 2015-TODAY Akretion (<http://www.akretion.com>).
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+import random
+
+from odoo import api, fields, models
+
+
+class ProductCategory(models.Model):
+    _inherit = ["product.category", "pos.load.mixin"]
+    _name = "product.category"
+
+    def get_default_color(self):
+        return random.randint(0, 10)
+
+    child_ids = fields.One2many(string="POS child categories", related="child_id")
+    has_image = fields.Boolean(compute="_compute_has_image")
+    color = fields.Integer(required=False, default=get_default_color)
+    sequence = fields.Integer(
+        help="Gives the sequence order when displaying a list of product categories."
+    )
+
+    @api.model
+    def _load_pos_data_domain(self, data):
+        config_id = self.env["pos.config"].browse(data["pos.config"]["data"][0]["id"])
+        domain = (
+            [("id", "in", config_id._get_available_categories().ids)]
+            if config_id.limit_categories and config_id.iface_available_categ_ids
+            else []
+        )
+        return domain
+
+    @api.model
+    def _load_pos_data_fields(self, config_id):
+        return [
+            "id",
+            "name",
+            "parent_id",
+            "child_ids",
+            "write_date",
+            "has_image",
+            "color",
+            "sequence",
+        ]
+
+    @api.depends("has_image")
+    def _compute_has_image(self):
+        for category in self:
+            category.has_image = bool(category.image_128)

pos_remove_pos_category/models/res_config_settings.py

@@ -0,0 +1,10 @@
+# Copyright (C) 2015-TODAY Akretion (<http://www.akretion.com>).
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+from odoo import fields, models
+
+
+class ResConfigSettings(models.TransientModel):
+    _inherit = "res.config.settings"
+
+    pos_iface_available_categ_ids = fields.Many2many("product.category")

pos_remove_pos_category/static/src/js/CategoryButton.js

@@ -0,0 +1,19 @@
+import {ProductScreen} from "@point_of_sale/app/screens/product_screen/product_screen";
+import {patch} from "@web/core/utils/patch";
+import {pick} from "@web/core/utils/objects";
+
+patch(ProductScreen.prototype, {
+    getChildCategoriesInfo(category) {
+        return {
+            ...pick(category, "id", "name", "color"),
+            imgSrc:
+                this.pos.config.show_category_images && category.has_image
+                    ? `/web/image?model=product.category&field=image_128&id=${category.id}`
+                    : undefined,
+            isSelected: this.getAncestorsAndCurrent().includes(category),
+            isChildren: this.getChildCategories(this.pos.selectedCategory).includes(
+                category
+            ),
+        };
+    },
+});

pos_remove_pos_category/static/src/js/Chrome.js

@@ -10,6 +10,9 @@
                 <group>
                     <field name="available_in_pos" />
                 </group>
+                <group>
+                    <field name="color" widget="color_picker" />
+                </group>
             </field>
             <xpath expr="//field[@name='name']/../.." position="before">
                 <field name="image_128" widget="image" class="oe_avatar" />

pos_remove_pos_category/static/src/js/models.js

@@ -1,12 +1,13 @@
 <?xml version="1.0" encoding="utf-8" ?>
 <odoo>
     <record id="product_template_form_view" model="ir.ui.view">
-        <field name="name">hide.pos_categ_id.product.template.form</field>
+        <field name="name">hide.pos_categ_ids.product.template.form</field>
         <field name="model">product.template</field>
         <field name="inherit_id" ref="point_of_sale.product_template_form_view" />
         <field name="arch" type="xml">
-            <field name="pos_categ_id" position="attributes">
-                <attribute name="invisible">1</attribute>
+            <field name="pos_categ_ids" position="attributes">
+                <attribute name="invisible">0</attribute>
+                <attribute name="options">{}</attribute>
             </field>
         </field>
     </record>