[12.0] Add base_fileurl_field

test_base_fileurl_field/tests/__init__.py

@@ -0,0 +1,2 @@
+from . import ir_attachment
+from . import test_fileurl_fields

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)

test_base_fileurl_field/tests/test_fileurl_fields.py

@@ -0,0 +1,39 @@
+# Copyright 2019 Camptocamp SA
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl)
+import base64
+
+from odoo.tests import TransactionCase
+from odoo.modules.module import get_module_resource
+from odoo.exceptions import ValidationError
+
+
+class TestFileUrlFields(TransactionCase):
+
+    def test_fileurl_fields(self):
+        file_path = get_module_resource('test_base_fileurl_field', 'data',
+                                        'sample.txt')
+        image_path = get_module_resource('test_base_fileurl_field', 'data',
+                                         'pattern.png')
+        partner = self.env.ref('base.main_partner')
+        with open(file_path, 'rb') as f:
+            with open(image_path, 'rb') as i:
+                partner.write({
+                    'url_file': base64.b64encode(f.read()),
+                    'url_file_fname': 'sample.txt',
+                    'url_image': base64.b64encode(i.read()),
+                    'url_image_fname': 'pattern.png',
+                })
+
+        with open(file_path, 'rb') as f:
+            self.assertEqual(base64.decodebytes(partner.url_file), f.read())
+
+        with open(image_path, 'rb') as image:
+            self.assertEqual(base64.decodebytes(partner.url_image), i.read())
+
+        partner2 = self.env.ref('base.partner_admin')
+        with open(file_path, 'rb') as f:
+            with self.assertRaises(ValidationError):
+                partner2.write({
+                    'url_file': base64.b64encode(f.read()),
+                    'url_file_fname': 'sample.txt',
+                })