Burtiraven2/session_redis_public
1
0
Fork 0
mirror of https://github.com/camptocamp/odoo-cloud-platform.git synced 2026-06-24 08:47:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #174 from grindtildeath/12.0_attachment_s3_bin_size_close

Browse Source 
[12.0] attachment_s3: Improvements
This commit is contained in:
Guewen Baconnier
2020-06-22 07:45:26 +02:00
committed by GitHub
co-authored by GitHub
parent ef07381879 123947100b
commit 0a592403e8
1 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
attachment_s3/models/ir_attachment.py
+20 -18
View File
@@ -122,10 +122,12 @@ class IrAttachment(models.Model):
bucket.meta.client.head_object(
Bucket=bucket.name, Key=key
)
res = io.BytesIO()
bucket.download_fileobj(key, res)
res.seek(0)
read = base64.b64encode(res.read())
if bin_size:
return bucket.Object(key).content_length
with io.BytesIO() as res:
bucket.download_fileobj(key, res)
res.seek(0)
read = base64.b64encode(res.read())
except ClientError:
read = ''
_logger.info(
@@ -141,20 +143,20 @@ class IrAttachment(models.Model):
if location == 's3':
bucket = self._get_s3_bucket()
obj = bucket.Object(key=key)
file = io.BytesIO()
file.write(bin_data)
file.seek(0)
filename = 's3://%s/%s' % (bucket.name, key)
try:
obj.upload_fileobj(file)
except ClientError as error:
# log verbose error from s3, return short message for user
_logger.exception(
'Error during storage of the file %s' % filename
)
raise exceptions.UserError(
_('The file could not be stored: %s') % str(error)
)
with io.BytesIO() as file:
file.write(bin_data)
file.seek(0)
filename = 's3://%s/%s' % (bucket.name, key)
try:
obj.upload_fileobj(file)
except ClientError as error:
# log verbose error from s3, return short message for user
_logger.exception(
'Error during storage of the file %s' % filename
)
raise exceptions.UserError(
_('The file could not be stored: %s') % str(error)
)
else:
_super = super(IrAttachment, self)
filename = _super._store_file_write(key, bin_data)