Skip to content

Commit

Permalink
Merge pull request #136 from jayzhenghan/dev
Browse files Browse the repository at this point in the history
增加delete_bucket_referer 接口
  • Loading branch information
dt3310321 authored Sep 16, 2020
2 parents 69ccc8f + f035ecd commit 11724de
Show file tree
Hide file tree
Showing 2 changed files with 42 additions and 2 deletions.
32 changes: 32 additions & 0 deletions qcloud_cos/cos_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -2798,6 +2798,38 @@ def get_bucket_referer(self, Bucket, **kwargs):
format_dict(data['DomainList'], ['Domain'])
return data

def delete_bucket_referer(self, Bucket, **kwargs):
"""删除bucket防盗链规则
:param Bucket(string): 存储桶名称.
:param kwargs(dict): 设置请求headers.
:return(dict): None.
.. code-block:: python
config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key, Token=token) # 获取配置对象
client = CosS3Client(config)
# 获取bucket标签
response = client.delete_bucket_referer(
Bucket='bucket'
)
"""
xml_config = ''
headers = mapped(kwargs)
headers['Content-MD5'] = get_md5(xml_config)
headers['Content-Type'] = 'application/xml'
params = {'referer': ''}
url = self._conf.uri(bucket=Bucket)
rt = self.send_request(
method='PUT',
url=url,
bucket=Bucket,
data=xml_config,
auth=CosS3Auth(self._conf, params=params),
headers=headers,
params=params)
return None

# service interface begin
def list_buckets(self, **kwargs):
"""列出所有bucket
Expand Down
12 changes: 10 additions & 2 deletions ut/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -1078,8 +1078,8 @@ def _test_put_get_delete_bucket_origin():
)


def test_put_get_bucket_referer():
"""测试设置获取bucket防盗链规则"""
def test_put_get_delete_bucket_referer():
"""测试设置获取删除bucket防盗链规则"""
referer_config = {
'Status': 'Enabled',
'RefererType': 'White-List',
Expand All @@ -1099,6 +1099,14 @@ def test_put_get_bucket_referer():
response = client.get_bucket_referer(
Bucket=test_bucket,
)
response = client.delete_bucket_referer(
Bucket=test_bucket,
)
time.sleep(4)
response = client.get_bucket_referer(
Bucket=test_bucket,
)
assert len(response)==0


def test_put_get_traffic_limit():
Expand Down

0 comments on commit 11724de

Please sign in to comment.