Add support of open stack bucket provider (#2371)

* Add support of open stack bucket provider * Add changelog
neuro-inc · Oct 25, 2021 · fa755c6 · fa755c6
1 parent 9ab9ee7
commit fa755c6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/CHANGELOG.D/2371.feature b/CHANGELOG.D/2371.feature
@@ -0,0 +1 @@
+Added support of open stack as bucket provider.
diff --git a/neuro-sdk/src/neuro_sdk/buckets.py b/neuro-sdk/src/neuro_sdk/buckets.py
@@ -994,6 +994,7 @@ class Provider(str, enum.Enum):
         MINIO = "minio"
         AZURE = "azure"
         GCP = "gcp"
+        OPEN_STACK = "open_stack"
 
 
 @dataclass(frozen=True)
@@ -1198,7 +1199,11 @@ async def _get_new_credentials() -> BucketCredentials:
             return await self.request_tmp_credentials(bucket.id, cluster_name)
 
         provider: BucketProvider
-        if bucket.provider in (Bucket.Provider.AWS, Bucket.Provider.MINIO):
+        if bucket.provider in (
+            Bucket.Provider.AWS,
+            Bucket.Provider.MINIO,
+            Bucket.Provider.OPEN_STACK,
+        ):
             async with S3Provider.create(bucket, _get_new_credentials) as provider:
                 yield provider
         elif bucket.provider == Bucket.Provider.AZURE: