Merge pull request #136 from jayzhenghan/dev

增加delete_bucket_referer 接口

Author: dt3310321

qcloud_cos/cos_client.py

@@ -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

ut/test.py

@@ -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',
@@ -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():