add put/get bucket intelligenttiering

Author: jayzhenghan

qcloud_cos/cos_client.py

@@ -2797,6 +2797,76 @@ def get_bucket_referer(self, Bucket, **kwargs):
             format_dict(data['DomainList'], ['Domain'])
         return data
 
+    def put_bucket_intelligenttiering(self, Bucket, IntelligentTieringConfiguration={}, **kwargs):
+        """设置存储桶智能分层配置
+
+        :param Bucket(string): 存储桶名称.
+        :param IntelligentTieringConfiguration(dict): 只能分层配置
+        :param kwargs(dict): 设置请求headers.
+        :return: None.
+
+        .. code-block:: python
+
+            config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key, Token=token)  # 获取配置对象
+            client = CosS3Client(config)
+            
+            intelligent_tiering_conf = {
+                'Status': 'Enable',
+                'Transition': {
+                    'Days': '30|60|90',
+                    'RequestFrequent': '1'
+                }
+            }
+            client.put_bucket_intelligenttiering(Bucket="bucket", IntelligentTieringConfiguration=intelligent_tiering_conf)
+        """
+
+        xml_config = format_xml(data=IntelligentTieringConfiguration, root='IntelligentTieringConfiguration')
+        headers = mapped(kwargs)
+        headers['Content-Type'] = 'application/xml'
+        params = {'intelligenttiering': ''}
+        url = self._conf.uri(bucket=Bucket)
+        logger.info("put bucket intelligenttiering, url=:{url} ,headers=:{headers}".format(
+            url=url,
+            headers=headers))
+        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
+
+    def get_bucket_intelligenttiering(self, Bucket, **kwargs):
+        """获取存储桶智能分层配置
+        :param Bucket(string): 存储桶名称.
+        :param IntelligentTieringConfiguration(dict): 只能分层配置
+        :param kwargs(dict): 设置请求headers.
+        :return(dict): 智能分层配置.
+
+        .. code-block:: python
+            config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key, Token=token)  # 获取配置对象
+            client = CosS3Client(config)
+            client.get_bucket_intelligenttiering(Bucket='bucket')
+        """
+
+        headers = mapped(kwargs)
+        params = {'intelligenttiering': ''}
+        url = self._conf.uri(bucket=Bucket)
+        logger.info("get bucket intelligenttiering, url=:{url} ,headers=:{headers}".format(
+            url=url,
+            headers=headers))
+        rt = self.send_request(
+            method='GET',
+            url=url,
+            bucket=Bucket,
+            auth=CosS3Auth(self._conf, params=params),
+            headers=headers,
+            params=params)
+        data = xml_to_dict(rt.content)
+        return data
+
     # service interface begin
     def list_buckets(self, **kwargs):
         """列出所有bucket

ut/test.py

@@ -1241,6 +1241,24 @@ def test_download_file():
     if os.path.exists(file_name):
         os.remove(file_name)
 
+def test_put_get_bucket_intelligenttiering():
+    """测试设置获取智能分层"""
+    intelligent_tiering_conf = {
+                'Status': 'Enable',
+                'Transition': {
+                    'Days': '30',
+                    'RequestFrequent': '1'
+                }
+            }
+    response = client.put_bucket_intelligenttiering(
+        Bucket=test_bucket,
+        IntelligentTieringConfiguration=intelligent_tiering_conf
+    )
+    time.sleep(2)
+    response = client.get_bucket_intelligenttiering(
+        Bucket=test_bucket,
+    )
+
 if __name__ == "__main__":
     setUp()
     """
@@ -1268,6 +1286,7 @@ def test_download_file():
     test_select_object()
     _test_get_object_sensitive_content_recognition()
     test_download_file()
+    test_put_get_bucket_intelligenttiering()
     """
 
     tearDown()