[12.0] Add base_fileurl_field

test_base_fileurl_field/tests/ir_attachment.py

@@ -0,0 +1,44 @@
+# Copyright 2019 Camptocamp SA
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl)
+
+import logging
+
+from odoo import _, api, exceptions, models
+
+_logger = logging.getLogger(__name__)
+
+FAKE_S3_BUCKET = {}
+
+
+class IrAttachment(models.Model):
+    _inherit = "ir.attachment"
+
+    def _get_stores(self):
+        l = ['s3']
+        l += super(IrAttachment, self)._get_stores()
+        return l
+
+    @api.model
+    def _store_file_read(self, fname, bin_size=False):
+        if fname.startswith('s3://'):
+            return FAKE_S3_BUCKET.get(fname)
+        else:
+            return super(IrAttachment, self)._store_file_read(fname, bin_size)
+
+    @api.model
+    def _store_file_write(self, key, bin_data):
+        location = self.env.context.get('storage_location') or self._storage()
+        if location == 's3':
+            FAKE_S3_BUCKET[key] = bin_data
+            filename = 's3://fake_bucket/%s' % key
+        else:
+            _super = super(IrAttachment, self)
+            filename = _super._store_file_write(key, bin_data)
+        return filename
+
+    @api.model
+    def _store_file_delete(self, fname):
+        if fname.startswith('s3://'):
+            FAKE_S3_BUCKET.pop(fname)
+        else:
+            super(IrAttachment, self)._store_file_delete(fname)