115 lines
4.9 KiB
Python
115 lines
4.9 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
# contributor license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright ownership.
|
|
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
# (the "License"); you may not use this file except in compliance with
|
|
# the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from typing import Type
|
|
from typing import Union
|
|
from types import ModuleType
|
|
from typing import TYPE_CHECKING
|
|
|
|
from libcloud.common.providers import get_driver as _get_provider_driver
|
|
from libcloud.common.providers import set_driver as _set_provider_driver
|
|
from libcloud.storage.types import OLD_CONSTANT_TO_NEW_MAPPING
|
|
from libcloud.storage.types import Provider
|
|
|
|
if TYPE_CHECKING:
|
|
from libcloud.storage.base import StorageDriver
|
|
|
|
DRIVERS = {
|
|
Provider.DUMMY:
|
|
('libcloud.storage.drivers.dummy', 'DummyStorageDriver'),
|
|
Provider.CLOUDFILES:
|
|
('libcloud.storage.drivers.cloudfiles', 'CloudFilesStorageDriver'),
|
|
Provider.OPENSTACK_SWIFT:
|
|
('libcloud.storage.drivers.cloudfiles', 'OpenStackSwiftStorageDriver'),
|
|
Provider.S3:
|
|
('libcloud.storage.drivers.s3', 'S3StorageDriver'),
|
|
Provider.S3_US_EAST2:
|
|
('libcloud.storage.drivers.s3', 'S3USEast2StorageDriver'),
|
|
Provider.S3_US_WEST:
|
|
('libcloud.storage.drivers.s3', 'S3USWestStorageDriver'),
|
|
Provider.S3_US_WEST_OREGON:
|
|
('libcloud.storage.drivers.s3', 'S3USWestOregonStorageDriver'),
|
|
Provider.S3_US_GOV_WEST:
|
|
('libcloud.storage.drivers.s3', 'S3USGovWestStorageDriver'),
|
|
Provider.S3_CN_NORTH:
|
|
('libcloud.storage.drivers.s3', 'S3CNNorthStorageDriver'),
|
|
Provider.S3_CN_NORTHWEST:
|
|
('libcloud.storage.drivers.s3', 'S3CNNorthWestStorageDriver'),
|
|
Provider.S3_EU_WEST:
|
|
('libcloud.storage.drivers.s3', 'S3EUWestStorageDriver'),
|
|
Provider.S3_EU_WEST2:
|
|
('libcloud.storage.drivers.s3', 'S3EUWest2StorageDriver'),
|
|
Provider.S3_EU_CENTRAL:
|
|
('libcloud.storage.drivers.s3', 'S3EUCentralStorageDriver'),
|
|
Provider.S3_EU_NORTH1:
|
|
('libcloud.storage.drivers.s3', 'S3EUNorth1StorageDriver'),
|
|
Provider.S3_AP_SOUTH:
|
|
('libcloud.storage.drivers.s3', 'S3APSouthStorageDriver'),
|
|
Provider.S3_AP_SOUTHEAST:
|
|
('libcloud.storage.drivers.s3', 'S3APSEStorageDriver'),
|
|
Provider.S3_AP_SOUTHEAST2:
|
|
('libcloud.storage.drivers.s3', 'S3APSE2StorageDriver'),
|
|
Provider.S3_AP_NORTHEAST:
|
|
('libcloud.storage.drivers.s3', 'S3APNE1StorageDriver'),
|
|
Provider.S3_AP_NORTHEAST1:
|
|
('libcloud.storage.drivers.s3', 'S3APNE1StorageDriver'),
|
|
Provider.S3_AP_NORTHEAST2:
|
|
('libcloud.storage.drivers.s3', 'S3APNE2StorageDriver'),
|
|
Provider.S3_SA_EAST:
|
|
('libcloud.storage.drivers.s3', 'S3SAEastStorageDriver'),
|
|
Provider.S3_CA_CENTRAL:
|
|
('libcloud.storage.drivers.s3', 'S3CACentralStorageDriver'),
|
|
Provider.S3_RGW:
|
|
('libcloud.storage.drivers.rgw', 'S3RGWStorageDriver'),
|
|
Provider.S3_RGW_OUTSCALE:
|
|
('libcloud.storage.drivers.rgw', 'S3RGWOutscaleStorageDriver'),
|
|
Provider.NINEFOLD:
|
|
('libcloud.storage.drivers.ninefold', 'NinefoldStorageDriver'),
|
|
Provider.GOOGLE_STORAGE:
|
|
('libcloud.storage.drivers.google_storage', 'GoogleStorageDriver'),
|
|
Provider.NIMBUS:
|
|
('libcloud.storage.drivers.nimbus', 'NimbusStorageDriver'),
|
|
Provider.LOCAL:
|
|
('libcloud.storage.drivers.local', 'LocalStorageDriver'),
|
|
Provider.AZURE_BLOBS:
|
|
('libcloud.storage.drivers.azure_blobs', 'AzureBlobsStorageDriver'),
|
|
Provider.KTUCLOUD:
|
|
('libcloud.storage.drivers.ktucloud', 'KTUCloudStorageDriver'),
|
|
Provider.AURORAOBJECTS:
|
|
('libcloud.storage.drivers.auroraobjects', 'AuroraObjectsStorageDriver'),
|
|
Provider.BACKBLAZE_B2:
|
|
('libcloud.storage.drivers.backblaze_b2', 'BackblazeB2StorageDriver'),
|
|
Provider.ALIYUN_OSS:
|
|
('libcloud.storage.drivers.oss', 'OSSStorageDriver'),
|
|
Provider.DIGITALOCEAN_SPACES:
|
|
('libcloud.storage.drivers.digitalocean_spaces',
|
|
'DigitalOceanSpacesStorageDriver'),
|
|
Provider.MINIO:
|
|
('libcloud.storage.drivers.minio', 'MinIOStorageDriver'),
|
|
}
|
|
|
|
|
|
def get_driver(provider):
|
|
# type: (Union[Provider, str]) -> Type[StorageDriver]
|
|
deprecated_constants = OLD_CONSTANT_TO_NEW_MAPPING
|
|
return _get_provider_driver(drivers=DRIVERS, provider=provider,
|
|
deprecated_constants=deprecated_constants)
|
|
|
|
|
|
def set_driver(provider, module, klass):
|
|
# type: (Union[Provider, str], ModuleType, type) -> Type[StorageDriver]
|
|
return _set_provider_driver(drivers=DRIVERS, provider=provider,
|
|
module=module, klass=klass)
|