Merge pull request #120 from hellohellenmao/master

增加获取文件流、范围下载功能以及修改桶能拷贝功能代码

Author: jayzhenghan

qcloud_cos/tce_demo.py

@@ -21,7 +21,7 @@ def __init__(self, region, appid, secret_id, secret_key, end_point,
         self.cos_config = CosConfig(
             Secret_id=self.secret_id,
             Secret_key=self.secret_key,
-            Endpoint=self.end_point % region,
+            Endpoint=self.end_point,
             Scheme=self.scheme)
         self.client = CosS3Client(self.cos_config)
 
@@ -246,25 +246,15 @@ def delete_objects_with_quiet(self, bucket_name, obj_list, quiet):
             Bucket=bucket_name + '-' + self.appid, Delete=delete)
         return resp
 
-    def copy_object_in_same_bucket(self):
+    def copy_object_in_same_bucket(self, bucket_name, obj_name):
         # 桶内copy对象
-        bucket_name = "mainbkt"
-        object_key = "obj_copy"
-        copy_source = {
-            "Appid": self.appid,
-            "Bucket": "mainbkt",
-            "Key": "obj",
-            "Region": self.region
-        }
-        resp = self.client.copy_object(
-            Bucket=bucket_name + '-' + self.appid,
-            Key=object_key,
-            CopySource=copy_source,
-            CopyStatus="Copy")
+        bucket_with_appid = bucket_name + '-' + self.appid
+        copy_source = {'Bucket': bucket_with_appid, 'Key': obj_name, 'Endpoint': self.end_point}
+        resp = self.client.copy_object(Bucket=bucket_with_appid, Key='test_dst', CopySource=copy_source)
         return resp
 
     def copy_object_in_different_bucket(self):
-        # 桶内copy对象
+        # 不同桶内copy对象
         bucket_name = "wook-gao"
         object_key = "obj_copy"
         copy_source = {
@@ -378,6 +368,25 @@ def upload_file(self, bucket_name, obj_name, file_path):
             LocalFilePath=file_path)
         return resp
 
+    def get_file_flow(self, bucket_name, obj_name):
+        # 获取文件流
+        response = self.client.get_object(
+            Bucket=bucket_name + '-' + self.appid,
+            Key=obj_name,
+        )
+        fp = response['Body'].get_raw_stream()
+        print(fp.read(10))
+
+    def get_file_range(self, bucket_name, obj_name):
+        # 文件下载 指定下载范围
+        response = main.client.get_object(
+            Bucket=bucket_name + '-' + self.appid,
+            Key=obj_name,
+            Range='bytes=1-10'
+        )
+        fp = response['Body'].get_raw_stream()
+        print(fp.read())
+
 
 if __name__ == "__main__":
 
@@ -603,6 +612,15 @@ def upload_file(self, bucket_name, obj_name, file_path):
             obj_name="upload1g.txt",
             file_path="f:\\test1g.txt")
 
+        # 获取文件流
+        main.get_file_flow(bucket_name="mainbkt", obj_name="part100m.txt")
+
+        # 文件下载 指定下载范围
+        main.get_file_range(bucket_name="mainbkt", obj_name="part100m.txt")
+
+        # 桶内拷贝
+        main.copy_object_in_same_bucket(bucket_name="mainbkt", obj_name="str1m.txt")
+
         print("Succeeded")
 
     except Exception as e: