1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
| """This file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license.
| """
| from django.conf import settings # noqa: I001
|
| from .azure_blob.models import ( # noqa: F401
| AzureBlobImportStorage,
| AzureBlobImportStorageLink,
| AzureBlobExportStorage,
| AzureBlobExportStorageLink,
| )
| from .s3.models import ( # noqa: F401
| S3ImportStorage,
| S3ImportStorageLink,
| S3ExportStorage,
| S3ExportStorageLink,
| )
| from .gcs.models import ( # noqa: F401
| GCSImportStorage,
| GCSImportStorageLink,
| GCSExportStorage,
| GCSExportStorageLink,
| )
| from .redis.models import ( # noqa: F401
| RedisImportStorage,
| RedisImportStorageLink,
| RedisExportStorage,
| RedisExportStorageLink,
| )
|
| from label_studio.core.utils.common import load_func
|
|
| def get_storage_classes(storage_type='import'):
| """Helper function to return all registered ***ImportStorage classes.
| It's been made through the APIViews rather than using models directly to make it consistent with what we expose.
| Note: this func doesn't include LocalFiles storages!
| storage_type: import, export
| """
| storage_list = load_func(settings.GET_STORAGE_LIST)
| storage_classes = []
| for storage_decl in storage_list():
| storage_api_class = storage_decl[f'{storage_type}_list_api']
| storage_classes.append(storage_api_class.serializer_class.Meta.model)
| return storage_classes
|
|
