Closes UNB-2217 Create storage interface for Azure [api library].

Author: whoseoyster

unboxapi/__init__.py

@@ -25,10 +25,11 @@ class DeploymentType(Enum):
     ONPREM = 1
     AWS = 2
     GCP = 3
+    AZURE = 4
 
 
 # NOTE: Don't modify this unless you are deploying on-prem.
-DEPLOYMENT = DeploymentType.AWS
+DEPLOYMENT = DeploymentType.AZURE
 
 
 class UnboxClient(object):
@@ -55,6 +56,8 @@ def __init__(self, api_key: str):
             self.upload = self.api.upload_blob_s3
         elif DEPLOYMENT == DeploymentType.GCP:
             self.upload = self.api.upload_blob_gcs
+        elif DEPLOYMENT == DeploymentType.AZURE:
+            self.upload = self.api.upload_blob_azure
         else:
             self.upload = self.api.transfer_blob
 

unboxapi/api.py

@@ -10,8 +10,8 @@
 from .exceptions import ExceptionMap, UnboxException
 from .version import __version__
 
-UNBOX_ENDPOINT = "https://api.unbox.ai/api"
-# UNBOX_ENDPOINT = "http://localhost:8080/api"
+# UNBOX_ENDPOINT = "https://api.unbox.ai/api"
+UNBOX_ENDPOINT = "http://localhost:8080/api"
 UNBOX_STORAGE_PATH = os.path.expanduser("~/unbox/unbox-onpremise/userStorage")
 
 
@@ -196,6 +196,36 @@ def upload_blob_gcs(
         else:
             self._raise_on_respose(res)
 
+    def upload_blob_azure(
+        self, endpoint: str, file_path: str, object_name: str = None, body=None
+    ):
+        """Generic method to upload data to Azure Blob Storage and create the appropriate resource
+        in the backend.
+        """
+        params = {"storageInterface": "azure", "objectName": object_name}
+        presigned_json = self.get_request(endpoint, params=params)
+        with open(file_path, "rb") as f:
+            with tqdm(
+                total=os.stat(file_path).st_size,
+                unit="B",
+                unit_scale=True,
+                unit_divisor=1024,
+            ) as t:
+                wrapped_file = CallbackIOWrapper(t.update, f, "read")
+                res = requests.put(
+                    presigned_json["url"],
+                    data=wrapped_file,
+                    headers={
+                        "Content-Type": "application/x-gzip",
+                        "x-ms-blob-type": "BlockBlob",
+                    },
+                )
+        if res.ok:
+            body["storagePath"] = presigned_json["storagePath"]
+            return self.post_request(f"{endpoint}/{presigned_json['id']}", body=body)
+        else:
+            self._raise_on_respose(res)
+
     def transfer_blob(self, endpoint: str, file_path: str, object_name: str, body=None):
         """Generic method to transfer data to the unbox folder and create the appropriate
         resource in the backend when using a local deployment.