Initial commit: OCA Technical packages (595 packages)

odoo-bringout-oca-geospatial-website_geoengine/website_geoengine/models/res_partner.py

@@ -0,0 +1,120 @@
+# Copyright 2011-2024 Camptocamp SA
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)
+
+from odoo import _, api, fields, models
+from odoo.exceptions import ValidationError
+
+
+class ResPartner(models.Model):
+    _inherit = "res.partner"
+
+    opening_hours = fields.Char(string="Opening hours")
+
+    AUTHORIZED_FIELDS = ["name", "city", "zip", "street", "street2", "tag"]
+
+    @api.model
+    def get_search_tags(self, search, lang):
+        sql = """
+        WITH
+            names as (
+                SELECT
+                    DISTINCT 'name' as column,
+                    name as value
+                FROM
+                    res_partner
+                WHERE
+                    type='store'),
+            cities as (
+                SELECT
+                    DISTINCT 'city' as column,
+                    city as value
+                FROM
+                    res_partner
+                WHERE
+                    type='store'),
+            zips as (
+                SELECT
+                    DISTINCT 'zip' as column,
+                    zip as value
+                FROM
+                    res_partner
+                WHERE
+                    type='store'),
+            streets as (
+                SELECT
+                    DISTINCT 'street' as column,
+                    concat(street, street2) as value
+                FROM
+                    res_partner
+                WHERE
+                    type='store'),
+            tags as (
+                SELECT
+                    DISTINCT 'tag' as column,
+                    res_partner_category.name->>%s as value
+                FROM
+                    res_partner_category,
+                    res_partner_res_partner_category_rel,
+                    res_partner
+                WHERE
+                    res_partner_res_partner_category_rel.partner_id = res_partner.id
+                    AND
+                    res_partner_res_partner_category_rel.category_id = res_partner_category.id
+                    AND res_partner.type='store'
+            ),
+            all_tags as (
+                SELECT * FROM names
+                UNION SELECT * FROM cities
+                UNION SELECT * FROM zips
+                UNION SELECT * FROM streets
+                UNION SELECT * FROM tags )
+
+
+        SELECT * FROM all_tags WHERE value ILIKE %s;
+        """
+        self._cr.execute(sql, (lang, f"%{search}%"))
+        results = self._cr.fetchall()
+        return results
+
+    @api.model
+    def fetch_partner_geoengine(self, tags, lang, maxResults):
+        domain = [("type", "=", "store")]
+        for tag in tags:
+            field, value = tag.values()
+            if field not in self.AUTHORIZED_FIELDS:
+                raise ValidationError(_("Unauthorized field"))
+            domain.append((field.replace("tag", "category_id.name"), "ilike", value))
+
+        partners = self.sudo().search(domain)
+        features = []
+
+        if len(partners) > int(maxResults):
+            return {
+                "error": "Too many results",
+                "message": f"Too many results: {len(partners)}",
+            }
+
+        for partner in partners:
+            features.append(
+                {
+                    "type": "Feature",
+                    "geometry": {
+                        "type": "Point",
+                        "coordinates": [
+                            partner.partner_longitude,
+                            partner.partner_latitude,
+                        ],
+                    },
+                    "properties": {
+                        "id": partner.id or None,
+                        "name": partner.name or "",
+                        "zip": partner.zip or "",
+                        "city": partner.city or "",
+                        "street": partner.street or "",
+                        "street2": partner.street2 or "",
+                        "tags": partner.category_id.mapped("name") or "",
+                        "opening_hours": partner.opening_hours or "",
+                    },
+                }
+            )
+        return features