wehub-resource-sync
131 lines
3.9 KiB
Python
131 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
@File : aliyun_oss_api.py
|
|
@Date : 2024-02-02
|
|
@Author : Peng Shiyu
|
|
|
|
OSS Python SDK适用于Python 2.6、2.7、3.3、3.4、3.5、3.6、3.7、3.8及以上版本。
|
|
|
|
https://help.aliyun.com/zh/oss/developer-reference/map-custom-domain-names-4?spm=a2c4g.11186623.0.i6
|
|
"""
|
|
import oss2
|
|
from oss2.credentials import EnvironmentVariableCredentialsProvider, StaticCredentialsProvider
|
|
|
|
# https://next.api.aliyun.com/product/Oss
|
|
from domain_admin.utils import domain_util, dns_util
|
|
|
|
ENDPOINT_OPTIONS = [
|
|
{
|
|
'label': '华北1（青岛）',
|
|
'value': 'cn-qingdao',
|
|
'endpoint': 'https://oss-cn-qingdao.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '华北2（北京）',
|
|
'value': 'cn-beijing',
|
|
'endpoint': 'https://oss-cn-beijing.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '华北3（张家口）',
|
|
'value': 'cn-zhangjiakou',
|
|
'endpoint': 'https://oss-cn-zhangjiakou.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '华北6（乌兰察布）',
|
|
'value': 'cn-wulanchabu',
|
|
'endpoint': 'https://oss-cn-wulanchabu.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '华东1（杭州）',
|
|
'value': 'cn-hangzhou',
|
|
'endpoint': 'https://oss-cn-hangzhou.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '华东2（上海）',
|
|
'value': 'cn-shanghai',
|
|
'endpoint': 'https://oss-cn-shanghai.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '华南1（深圳）',
|
|
'value': 'cn-shenzhen',
|
|
'endpoint': 'https://oss-cn-shenzhen.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '华南3（广州）',
|
|
'value': 'cn-guangzhou',
|
|
'endpoint': 'https://oss-cn-guangzhou.aliyuncs.com',
|
|
},
|
|
{
|
|
'label': '西南1（成都）',
|
|
'value': 'cn-chengdu',
|
|
'endpoint': 'https://oss-cn-chengdu.aliyuncs.com',
|
|
}
|
|
]
|
|
|
|
|
|
def get_endpoint_by_value(value):
|
|
for item in ENDPOINT_OPTIONS:
|
|
if item['value'] == value:
|
|
return item['endpoint']
|
|
|
|
|
|
def cname_to_oss_info(cname):
|
|
"""
|
|
'zaiting.oss-cn-beijing.aliyuncs.com.'
|
|
:param cname:
|
|
:return: zaiting oss-cn-beijing.aliyuncs.com
|
|
"""
|
|
lst = dns_util.query_domain_cname(cname)
|
|
|
|
if lst and len(lst) > 0:
|
|
domain = lst[0]
|
|
|
|
return {
|
|
'bucket_name': domain.split('.')[0],
|
|
'endpoint': 'https://' + domain_util.get_domain_parent(domain).strip('.')
|
|
}
|
|
else:
|
|
return None
|
|
|
|
|
|
def put_bucket_cname(
|
|
access_key_id,
|
|
access_key_secret,
|
|
bucket_name,
|
|
domain,
|
|
certificate,
|
|
private_key,
|
|
endpoint='https://oss-cn-beijing.aliyuncs.com',
|
|
):
|
|
"""
|
|
将证书部署到oss
|
|
:param access_key_id:
|
|
:param access_key_secret:
|
|
:param bucket_name:
|
|
:param domain:
|
|
:param certificate:
|
|
:param private_key:
|
|
:param endpoint:
|
|
:return:
|
|
"""
|
|
# 从环境变量中获取访问凭证。运行本代码示例之前，请确保已设置环境变量OSS_ACCESS_KEY_ID和OSS_ACCESS_KEY_SECRET。
|
|
auth = oss2.ProviderAuth(
|
|
StaticCredentialsProvider(access_key_id=access_key_id, access_key_secret=access_key_secret)
|
|
)
|
|
|
|
# yourEndpoint填写Bucket所在地域对应的Endpoint。以华东1（杭州）为例，Endpoint填写为https://oss-cn-hangzhou.aliyuncs.com。
|
|
# 填写Bucket名称，例如examplebucket。
|
|
bucket = oss2.Bucket(
|
|
auth=auth,
|
|
# https://oss-cn-beijing.aliyuncs.com
|
|
endpoint=endpoint,
|
|
bucket_name=bucket_name
|
|
)
|
|
|
|
cert = oss2.models.CertInfo(certificate=certificate, private_key=private_key)
|
|
# 通过force=True设置强制覆盖旧版证书。
|
|
# 通过delete_certificate选择是否删除证书。设置为delete_certificate=True表示删除证书，设置为delete_certificate=False表示不删除证书。
|
|
# cert = oss2.models.CertInfo(certificate=certificate, private_key=private_key, force=True, delete_certificate=False)
|
|
input = oss2.models.PutBucketCnameRequest(domain, cert)
|
|
bucket.put_bucket_cname(input)
|