File tree 2 files changed +11
-3
lines changed
2 files changed +11
-3
lines changed Original file line number Diff line number Diff line change @@ -1155,6 +1155,13 @@ class OrgStorageReplicaRefs(BaseModel):
1155
1155
storageReplicas : List [StorageRef ]
1156
1156
1157
1157
1158
+ # ============================================================================
1159
+ class OrgAllStorages (BaseModel ):
1160
+ """Response model for listing all available storages"""
1161
+
1162
+ allStorages : List [StorageRef ]
1163
+
1164
+
1158
1165
# ============================================================================
1159
1166
class S3StorageIn (BaseModel ):
1160
1167
"""Custom S3 Storage input model"""
Original file line number Diff line number Diff line change 49
49
OrgStorageRefs ,
50
50
OrgStorageRef ,
51
51
OrgStorageReplicaRefs ,
52
+ OrgAllStorages ,
52
53
DeletedResponse ,
53
54
UpdatedResponse ,
54
55
UpdatedResponseId ,
@@ -387,14 +388,14 @@ async def _run_post_storage_replica_update_tasks(
387
388
org , replica_storage , remove = True
388
389
)
389
390
390
- def get_available_storages (self , org : Organization ) -> List [StorageRef ]:
391
+ def get_available_storages (self , org : Organization ) -> Dict [ str , List [StorageRef ] ]:
391
392
"""return a list of available default + custom storages"""
392
393
refs : List [StorageRef ] = []
393
394
for name in self .default_storages :
394
395
refs .append (StorageRef (name = name , custom = False ))
395
396
for name in org .customStorages :
396
397
refs .append (StorageRef (name = name , custom = True ))
397
- return refs
398
+ return { "allStorages" : refs }
398
399
399
400
@asynccontextmanager
400
401
async def get_s3_client (
@@ -893,7 +894,7 @@ def get_storage_refs(
893
894
@router .get (
894
895
"/all-storages" ,
895
896
tags = ["organizations" , "storage" ],
896
- response_model = List [ StorageRef ] ,
897
+ response_model = OrgAllStorages ,
897
898
)
898
899
def get_available_storages (org : Organization = Depends (org_owner_dep )):
899
900
return storage_ops .get_available_storages (org )
You can’t perform that action at this time.
0 commit comments