59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
|
from pywxdump.common.config.oss_config.storage_config import StorageConfig, TYPE_KEY, DescriptionBuilder
|
||
|
|
from pywxdump.common.config.oss_config.storage_config_factory import StorageConfigFactory
|
||
|
|
|
||
|
|
S3 = "s3"
|
||
|
|
ENDPOINT_URL = "endpoint_url"
|
||
|
|
SECRET_KEY = "secret_key"
|
||
|
|
ACCESS_KEY = "access_key"
|
||
|
|
|
||
|
|
|
||
|
|
@StorageConfigFactory.register(S3)
|
||
|
|
class S3Config(StorageConfig):
|
||
|
|
|
||
|
|
def __init__(self, access_key, secret_key, endpoint_url):
|
||
|
|
self.access_key = access_key
|
||
|
|
self.secret_key = secret_key
|
||
|
|
self.endpoint_url = endpoint_url
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def type(cls) -> str:
|
||
|
|
return S3
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def describe(cls):
|
||
|
|
builder = DescriptionBuilder()
|
||
|
|
builder.add_description(ENDPOINT_URL, "https://cos.<your-regin>.myqcloud.com", ENDPOINT_URL)
|
||
|
|
builder.add_description(ACCESS_KEY, "腾讯云的SecretId", ACCESS_KEY)
|
||
|
|
builder.add_description(SECRET_KEY, "腾讯云的SecretKey", SECRET_KEY)
|
||
|
|
return builder.build()
|
||
|
|
|
||
|
|
def get_config(self):
|
||
|
|
return {
|
||
|
|
TYPE_KEY: S3,
|
||
|
|
ACCESS_KEY: self.access_key,
|
||
|
|
SECRET_KEY: self.secret_key,
|
||
|
|
ENDPOINT_URL: self.endpoint_url
|
||
|
|
}
|
||
|
|
|
||
|
|
def validate_config(self):
|
||
|
|
if not self.access_key or not self.secret_key or not self.endpoint_url:
|
||
|
|
raise ValueError("S3 configuration is not valid")
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def value_of(cls, config: dict):
|
||
|
|
access_key = config.get(ACCESS_KEY)
|
||
|
|
secret_key = config.get(SECRET_KEY)
|
||
|
|
endpoint_url = config.get(ENDPOINT_URL)
|
||
|
|
s3_config = cls(access_key, secret_key, endpoint_url)
|
||
|
|
s3_config.validate_config()
|
||
|
|
return s3_config
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def isSupported(cls, path: str) -> bool:
|
||
|
|
if not path:
|
||
|
|
return False
|
||
|
|
if path.startswith(f"{S3}://"):
|
||
|
|
return True
|
||
|
|
|
||
|
|
|