From 38f03cd008867ca1643afabb2168776c9c9f57ca Mon Sep 17 00:00:00 2001
From: libertyzhu <libertyzhu@tencent.com>
Date: Mon, 3 Jun 2024 14:31:22 +0800
Subject: [PATCH] multiversioning recovery demo

---
 demo/disaster_recovery_demo.py | 92 ++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 demo/disaster_recovery_demo.py

diff --git a/demo/disaster_recovery_demo.py b/demo/disaster_recovery_demo.py
new file mode 100644
index 00000000..d4fe512d
--- /dev/null
+++ b/demo/disaster_recovery_demo.py
@@ -0,0 +1,92 @@
+# -*- coding=utf-8
+#
+# ref to @shezhangjun
+# https://github.com/shezhangjun/TencentCOS/blob/master/Python_SDK/COS_Disaster_Recovery/DisasterRecovery.py
+#
+from qcloud_cos import CosConfig
+from qcloud_cos import CosS3Client
+import sys
+import os
+import logging
+
+# logging.basicConfig(level=logging.INFO, stream=sys.stdout)
+
+
+def _recover_main(src_region, src_secret_id, src_secret_key, src_bucket, prefix,
+                  dst_region, dst_secret_id, dst_secret_key, dst_bucket):
+    src_client = CosS3Client(CosConfig(Region=src_region, SecretId=src_secret_id, SecretKey=src_secret_key))
+    dst_client = CosS3Client(CosConfig(Region=dst_region, SecretId=dst_secret_id, SecretKey=dst_secret_key))
+    while True:
+        key_marker = ''
+        versionId_marker = ''
+        response = src_client.list_objects_versions(
+            Bucket=src_bucket,
+            Prefix=prefix,
+            KeyMarker=key_marker,
+            VersionIdMarker=versionId_marker,
+        )
+        delete_marker_keys = list()
+        recovered_keys = list()
+        # 从 DeleteMarker 取出被删除的对象
+        if 'DeleteMarker' in response:
+            for version in response['DeleteMarker']:
+                if version['IsLatest'] == 'true':
+                    delete_marker_keys.append(version['Key'])
+        
+        if len(delete_marker_keys) == 0:
+            print('no delete markers found, no data to recover')
+            return
+
+        # 从 Version 取最新的版本号
+        if 'Version' in response:
+            for version in response['Version']:
+                key = version['Key']
+                versionId = version['VersionId']
+                if key in delete_marker_keys and not key in recovered_keys:
+                    print('recover from key:{key}, versionId:{versionId}'.format(key=key, versionId=versionId))
+                    try:
+                        dst_client.copy(
+                            Bucket=dst_bucket,
+                            Key=key,
+                            CopySource={
+                                'Bucket': src_bucket,
+                                'Key': key,
+                                'Region': src_region,
+                                'VersionId': versionId,
+                            }
+                        )
+                        recovered_keys.append(key)
+                        print("success recover object: {src_bucket}/{key}({versionId}) => {dst_bucket}/{key}".format(
+                            src_bucket=src_bucket, key=key, versionId=versionId, dst_bucket=dst_bucket))
+                    except Exception as e:
+                        print(e)
+                        pass
+
+        if response['IsTruncated'] == 'false':
+            break
+
+        key_marker = response['NextKeyMarker']
+        versionId_marker = response['NextVersionIdMarker']
+
+
+if __name__ == '__main__':
+    # 使用场景:
+    # 根据源桶src_bucket的删除标记从历史版本里把文件恢复到dst_bucket
+    # src_bucket和dst_bucket可以一致, 即原地恢复
+
+    # 源桶信息
+    src_region = 'ap-guangzhou'  # 源地域
+    src_secret_id = ''  # 源桶SecretID
+    src_secret_key = ''  # 源桶SecretKey
+    src_bucket = 'bucket-1200000000'  # 源桶名
+
+    # 目标桶信息
+    dst_region = 'ap-guangzhou'  # 目标桶地域
+    dst_secret_id = ''  # 目标桶SecretID
+    dst_secret_key = ''  # 目标桶SecretKey
+    dst_bucket = 'bucket-1250000000'  # 目标桶名
+
+    prefix = ''  # 设置要恢复的对象前缀
+
+    _recover_main(src_region, src_secret_id, src_secret_key, src_bucket, prefix,
+                  dst_region, dst_secret_id, dst_secret_key, dst_bucket)
\ No newline at end of file