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

[13.0][IMP] attachment_azure: Write and read storage name along filename for extended usability

Browse Source
This commit is contained in:
Patrick Tombez
2021-11-15 12:57:17 +01:00
committed by Yannick Vaucher
co-authored by Yannick Vaucher
parent bb636eba9d
commit 438cb2adc9
2 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
attachment_azure/models/ir_attachment.py
+14 -5
View File
@@ -128,8 +128,9 @@ class IrAttachment(models.Model):
return str.lower(storage_name)[:63]
@api.model
def _get_azure_container(self):
container_name = self._get_container_name()
def _get_azure_container(self, container_name=None):
if not container_name:
container_name = self._get_container_name()
blob_service_client = self._get_blob_service_client()
container_client = blob_service_client.get_container_client(container_name)
try:
@@ -145,8 +146,12 @@ class IrAttachment(models.Model):
@api.model
def _store_file_read(self, fname, bin_size=False):
if fname.startswith("azure://"):
container_client = self._get_azure_container()
key = fname.replace("azure://", "", 1).lower()
if '/' in key:
container_name, key = key.split('/')
else:
container_name = None
container_client = self._get_azure_container(container_name)
try:
blob_client = container_client.get_blob_client(key)
read = blob_client.download_blob().readall()
@@ -162,11 +167,11 @@ class IrAttachment(models.Model):
location = self.env.context.get("storage_location") or self._storage()
if location == "azure":
container_client = self._get_azure_container()
filename = "azure://%s/%s" % (container_client.container_name, key)
with io.BytesIO() as file:
blob_client = container_client.get_blob_client(key.lower())
file.write(bin_data)
file.seek(0)
filename = "azure://%s" % (key)
try:
blob_client.upload_blob(file, blob_type="BlockBlob")
except ResourceExistsError:
@@ -185,8 +190,12 @@ class IrAttachment(models.Model):
@api.model
def _store_file_delete(self, fname):
if fname.startswith("azure://"):
container_client = self._get_azure_container()
key = fname.replace("azure://", "", 1).lower()
if '/' in key:
container_name, key = key.split('/')
else:
container_name = None
container_client = self._get_azure_container(container_name)
# delete the file only if it is on the current configured container
# otherwise, we might delete files used on a different environment
try: