create_bucket OFS

Author: l-iberty

qcloud_cos/cos_client.py

@@ -1277,7 +1277,7 @@ def select_object_content(self, Bucket, Key, Expression, ExpressionType, InputSe
         return data
 
     # s3 bucket interface begin
-    def create_bucket(self, Bucket, BucketAZConfig=None, **kwargs):
+    def create_bucket(self, Bucket, BucketAZConfig=None, BucketArchConfig=None, **kwargs):
         """创建一个bucket
 
         :param Bucket(string): 存储桶名称. 存储桶名称不支持大写字母，COS 后端会将用户传入的大写字母自动转换为小写字母用于创建存储桶.
@@ -1301,11 +1301,16 @@ def create_bucket(self, Bucket, BucketAZConfig=None, **kwargs):
         """
         headers = mapped(kwargs)
         xml_config = None
+        bucket_config = dict()
         if BucketAZConfig == 'MAZ':
-            bucket_config = {'BucketAZConfig': 'MAZ'}
+            bucket_config.update({'BucketAZConfig': 'MAZ'})
+        if BucketArchConfig == 'OFS':
+            bucket_config.update({'BucketArchConfig': 'OFS'})
+        if len(bucket_config) != 0:
             xml_config = format_xml(data=bucket_config, root='CreateBucketConfiguration')
             headers['Content-MD5'] = get_md5(xml_config)
             headers['Content-Type'] = 'application/xml'
+
         url = self._conf.uri(bucket=Bucket)
         logger.info("create bucket, url=:{url} ,headers=:{headers}".format(
             url=url,

ut/test.py

@@ -562,13 +562,14 @@ def test_create_head_delete_bucket():
         Bucket=bucket_name
     )
 
-def test_create_head_delete_maz_bucket():
-    """创建一个多AZ bucket,head它是否存在,最后删除一个空bucket"""
+def test_create_head_delete_maz_ofs_bucket():
+    """创建一个多AZ OFS bucket,head它是否存在,最后删除一个空bucket"""
     bucket_id = str(random.randint(0, 1000)) + str(random.randint(0, 1000))
-    bucket_name = 'buckettest-maz' + bucket_id + '-' + APPID
+    bucket_name = 'buckettest-maz-ofs' + bucket_id + '-' + APPID
     response = client.create_bucket(
         Bucket=bucket_name,
         BucketAZConfig='MAZ',
+        BucketArchConfig='OFS',
         ACL='public-read'
     )
     response = client.head_bucket(
@@ -578,6 +579,7 @@ def test_create_head_delete_maz_bucket():
         Bucket=bucket_name
     )
 
+
 def test_put_bucket_acl():
     """正常设置bucket ACL"""
     response = client.put_bucket_acl(