[IMP] contentgrid: Enforce storage in ContentGrid

Author: etobella

contentgrid/models/contentgrid_configuration.py

@@ -50,6 +50,11 @@ class ContentgridConfiguration(models.Model):
     allow_manual_send = fields.Boolean(
         default=False,
     )
+    use_contentgrid_for_storage = fields.Boolean(
+        help="Use ContentGrid for storage data"
+    )
+    contentgrid_storage_model = fields.Char()
+    contentgrid_storage_field = fields.Char()
 
     @api.constrains("configuration_data")
     def _check_configuration_data(self):
@@ -181,3 +186,20 @@ def _push_to_contentgrid(self, attachment, record):
                             timeout=self.connection_id._timeout,
                         )
                         response.raise_for_status()
+        storage_model = self.contentgrid_storage_model
+        storage_field = self.contentgrid_storage_field
+        if (
+            self.use_contentgrid_for_storage
+            and storage_model
+            and storage_field
+            and not attachment.contentgrid_connection_id
+        ):
+            url = f"{storage_model}s/{processed[storage_model][0]}/{storage_field}"
+            attachment.write(
+                {
+                    "contentgrid_connection_id": self.connection_id.id,
+                    "contentgrid_url": url,
+                    "raw": False,
+                    "mimetype": attachment.mimetype,
+                }
+            )

contentgrid/models/contentgrid_connection.py

@@ -15,8 +15,6 @@ class ContentgridConnection(models.Model):
     openid_url = fields.Char(groups="base.group_system", required=True)
     openid_client_id = fields.Char(groups="base.group_system", required=True)
     openid_client_secret = fields.Char(groups="base.group_system", required=True)
-    username = fields.Char(groups="base.group_system", required=True)
-    password = fields.Char(groups="base.group_system", required=True)
     base_url = fields.Char(
         help="Base URL of the Contentgrid instance without models",
         required=True,

contentgrid/models/ir_attachment.py

@@ -1,6 +1,8 @@
 # Copyright 2025 Dixmit
 # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
 
+import requests
+
 from odoo import api, fields, models
 
 from odoo.addons.mail.tools.discuss import Store
@@ -16,6 +18,8 @@ class IrAttachment(models.Model):
         string="Content Grid Data",
         ondelete="cascade",
     )
+    contentgrid_connection_id = fields.Many2one("contentgrid.connection")
+    contentgrid_url = fields.Char()
 
     def _push_to_contentgrid(self, manual_send=False):
         self.ensure_one()
@@ -62,3 +66,29 @@ def get_contentgrid_data(self):
         return [
             record._get_contentgrid_data() for record in self.contentgrid_ids.sudo()
         ]
+
+    @api.depends("contentgrid_connection_id", "contentgrid_url")
+    def _compute_raw(self):
+        for attachment in self.filtered(lambda r: r.contentgrid_connection_id):
+            access_token = attachment.contentgrid_connection_id._get_token()
+            headers = {
+                "Authorization": f"Bearer {access_token}",
+            }
+            data_request = requests.get(
+                f"{attachment.contentgrid_connection_id.base_url}/{attachment.contentgrid_url}",
+                headers=headers,
+                timeout=attachment.contentgrid_connection_id._timeout,
+            )
+            try:
+                data_request.raise_for_status()
+                attachment.raw = data_request.content
+            except requests.HTTPError:
+                attachment.raw = b""
+        return super(
+            IrAttachment, self.filtered(lambda r: not r.contentgrid_connection_id)
+        )._compute_raw()
+
+    def _set_attachment_data(self, asbytes):
+        if self.contentgrid_connection_id:
+            return
+        return super()._set_attachment_data(asbytes)

contentgrid/views/contentgrid_configuration.xml

@@ -19,6 +19,15 @@
                         <field name="model_id" />
                         <field name="connection_id" />
                         <field name="allow_manual_send" />
+                        <field name="use_contentgrid_for_storage" />
+                        <field
+                            name="contentgrid_storage_model"
+                            invisible="not use_contentgrid_for_storage"
+                        />
+                        <field
+                            name="contentgrid_storage_field"
+                            invisible="not use_contentgrid_for_storage"
+                        />
                         <field name="res_model" invisible="1" />
                         <field name="active" invisible="1" />
                     </group>

contentgrid/views/contentgrid_connection.xml

@@ -30,8 +30,6 @@
                         <field name="openid_url" />
                         <field name="openid_client_id" />
                         <field name="openid_client_secret" password="True" />
-                        <field name="username" />
-                        <field name="password" password="True" />
                         <field name="base_url" />
                         <field name="app_url" />
                     </group>