Add implementation of reading writing on the Swift object store

attachment_swift/models/ir_attachment.py

@@ -3,8 +3,14 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html)
 
 
+import base64
 import logging
-from odoo import api, models
+import os
+import swiftclient
+from swiftclient.exceptions import ClientException
+from ..swift_uri import SwiftUri
+
+from odoo import api, exceptions, models, _
 
 _logger = logging.getLogger(__name__)
 
@@ -12,8 +18,99 @@ _logger = logging.getLogger(__name__)
 class IrAttachment(models.Model):
     _inherit = 'ir.attachment'
 
+    @api.multi
+    def _store_in_db_when_swift(self):
+        # For testing lets save everything in Swift object store
+        # TODO: same than in attachment_s3
+        return False
+
     @api.model
-    def _file_write(self, value, checksum):
-        _logger.debug('Writing a file :)')
-        filename = super(IrAttachment, self)._file_write(value, checksum)
+    def _get_swift_connection(self):
+        """ Returns a connection object for the Swift object store """
+        host = os.environ.get('SWIFT_HOST')
+        account = os.environ.get('SWIFT_ACCOUNT')
+        password = os.environ.get('SWIFT_PASSWORD')
+        if not (host and account and password):
+            raise exceptions.UserError(
+                _('Problem connecting to Swift store, are not the env variables set ?'))
+        print 'Connection to host: {}, account: {}, password: {}'.format(host, account, password)
+        try:
+            conn = swiftclient.client.Connection(authurl=host, user=account, key=password)
+        except ClientException:
+            _logger.exception('Error connecting to Swift object store')
+            raise exceptions.UserError('Error connection to Swift')
+        return conn
+
+    @api.model
+    def _file_read_swift(self, fname, bin_size=False):
+        swifturi = SwiftUri(fname)
+        conn = self._get_swift_connection()
+        print 'Swift reading on {} of {} '.format(swifturi.container(), swifturi.item())
+        try:
+            resp_headers, obj_content = conn.get_object(swifturi.container(), swifturi.item())
+            read = base64.b64encode(obj_content)
+        except ClientException:
+            _logger.exception('Error reading object from Swift object store');
+            #raise exceptions.UserError('Error reading to Swift')
+            return ''
+        return read
+
+    @api.model
+    def _file_read(self, fname, bin_size=False):
+        if fname.startswith('swift://'):
+            return self._file_read_swift(fname, bin_size=bin_size)
+        else:
+            _super = super(IrAttachment, self)
+            return _super._file_read(fname, bin_size=bin_size)
+
+    def _file_write_swift(self, value, checksum):
+        container = os.environ.get('SWIFT_WRITE_CONTAINER')
+        conn = self._get_swift_connection()
+        conn.put_container(container)
+        bin_data = value.decode('base64')
+        # No keys given by the store, use checksum !?
+        key = self._compute_checksum(bin_data)
+        filename = 'swift://{}/{}'.format(container, key)
+        print 'Saving {}'.format(filename)
+        try:
+            conn.put_object(container, key, bin_data)
+        except ClientException:
+            _logger.exception('Error connecting to Swift object store')
+            raise exceptions.UserError('Error writting to Swift')
         return filename
+
+    def _file_write(self, value, checksum):
+        storage = self._storage()
+        if storage == 'swift':
+            filename = self._file_write_swift(value, checksum)
+        else:
+            filename = super(IrAttachment, self)._file_write(value, checksum)
+        return filename
+
+    @api.model
+    def _file_delete(self, fname):
+        if fname.startswith('swift://'):
+            swifturi = SwiftUri(fname)
+            container = swifturi.container()
+            print 'Deleting... container: {} | filename: {}'.format(container, swifturi.item())
+            if container == os.environ.get('SWIFT_WRITE_CONTAINER'):
+                conn = self._get_swift_connection()
+                try:
+                    conn.delete_object(container, swifturi.item())
+                except ClientException as error:
+                    _logger.exception('Error connecting to Swift object store');
+                    raise exceptions.UserError('Error deleting in Swift')
+        else:
+            super(IrAttachment, self)._file_delete(fname)
+
+    @api.model
+    def _force_storage_swift(self, new_cr=False):
+        return
+
+    @api.model
+    def force_storage(self):
+        storage = self._storage()
+        if storage == 'swift':
+            self._force_storage_swift()
+        else:
+            return super(IrAttachment, self).force_storage()