Merge pull request #309 from camptocamp/15.0-fp-attachment_azure-containername

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

attachment_azure/README.rst

@@ -27,6 +27,9 @@ The container name can be overridden with environment variable ``AZURE_STORAGE_N
 The strings ``{db}`` and ``{env}`` can be used inside that variable and the values
 will be replaced respectively by the database name and environment name.
 
+The container name will also be stored in the database for each attachment,
+and will be used to access the right container in the storage.
+
 This addon must be added in the server wide addons with (``--load`` option):
 
 ``--load=web,attachment_azure``

attachment_azure/models/ir_attachment.py

@@ -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: