mirror of
https://github.com/camptocamp/odoo-cloud-platform.git
synced 2026-06-23 18:04:34 +00:00
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2017 Camptocamp SA
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html)
|
|
|
|
|
|
import base64
|
|
import logging
|
|
import os
|
|
import swiftclient
|
|
from swiftclient.exceptions import ClientException
|
|
from ..swift_uri import SwiftUri
|
|
|
|
from odoo import api, exceptions, models, _
|
|
|
|
_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 _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()
|