From 66452f698348478fd8489e6462f0abd5d89a6cab Mon Sep 17 00:00:00 2001
From: xiao-77 <berylbao@taosdata.com>
Date: Thu, 23 Jan 2025 13:56:44 +0800
Subject: [PATCH] Add help info.

---
 tests/script/repair/repair_tsdb_data.py | 58 ++++++++++++++-----------
 1 file changed, 32 insertions(+), 26 deletions(-)

diff --git a/tests/script/repair/repair_tsdb_data.py b/tests/script/repair/repair_tsdb_data.py
index 27e57fb59a..d8aad23802 100644
--- a/tests/script/repair/repair_tsdb_data.py
+++ b/tests/script/repair/repair_tsdb_data.py
@@ -3,38 +3,44 @@ import sys
 import shutil
 import time
 import os
+import argparse
 
-if len(sys.argv) < 2:
-    print("Usage: python3 delete_fid.py <fid> [file_path]")
-    sys.exit(1)
+def main():
+    parser = argparse.ArgumentParser(description='Repair TSDB data by removing specified fid.')
+    parser.add_argument('fid', type=int, help='The fid to be removed')
+    parser.add_argument('file_path', nargs='?', default='current.json', help='The path to the JSON file (default: current.json)')
+    args = parser.parse_args()
 
-target_fid = int(sys.argv[1])
-file_path = sys.argv[2] if len(sys.argv) > 2 else "current.json"
+    target_fid = args.fid
+    file_path = args.file_path
 
-# Read file content
-with open(file_path, 'r') as file:
-    data = json.load(file)
+    # Read file content
+    with open(file_path, 'r') as file:
+        data = json.load(file)
 
-# Check if the fid exists
-fid_exists = any(item.get('fid') == target_fid for item in data['fset'])
-if not fid_exists:
-    print(f"Error: fid {target_fid} does not exist in the file.")
-    sys.exit(1)
+    # Check if the fid exists
+    fid_exists = any(item.get('fid') == target_fid for item in data['fset'])
+    if not fid_exists:
+        print(f"Error: fid {target_fid} does not exist in the file.")
+        sys.exit(1)
 
-# Generate backup file name
-timestamp = time.strftime("%Y%m%d%H%M%S")
-parent_directory = os.path.dirname(os.path.dirname(file_path))
-backup_file_path = os.path.join(parent_directory, f"current.json.{timestamp}")
+    # Generate backup file name
+    timestamp = time.strftime("%Y%m%d%H%M%S")
+    parent_directory = os.path.dirname(os.path.dirname(file_path))
+    backup_file_path = os.path.join(parent_directory, f"current.json.{timestamp}")
 
-# Backup file
-shutil.copy(file_path, backup_file_path)
-print(f"Backup created: {backup_file_path}")
+    # Backup file
+    shutil.copy(file_path, backup_file_path)
+    print(f"Backup created: {backup_file_path}")
 
-# Remove objects with the specified fid from the fset list
-data['fset'] = [item for item in data['fset'] if item.get('fid') != target_fid]
+    # Remove objects with the specified fid from the fset list
+    data['fset'] = [item for item in data['fset'] if item.get('fid') != target_fid]
 
-# Write the updated content back to the file, preserving the original format
-with open(file_path, 'w') as file:
-    json.dump(data, file, separators=(',', ':'), ensure_ascii=False)
+    # Write the updated content back to the file, preserving the original format
+    with open(file_path, 'w') as file:
+        json.dump(data, file, separators=(',', ':'), ensure_ascii=False)
 
-print(f"Removed content with fid {target_fid}.")
\ No newline at end of file
+    print(f"Removed content with fid {target_fid}.")
+
+if __name__ == '__main__':
+    main()
\ No newline at end of file