File size: 1,209 Bytes
73df34b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | import os
import json
import glob
import argparse
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--input-dir", type=str, required=True)
parser.add_argument("--pattern", type=str, required=True)
parser.add_argument("--output-path", type=str, required=True)
args = parser.parse_args()
input_pattern = os.path.join(args.input_dir, args.pattern)
input_paths = sorted(glob.glob(input_pattern))
if not input_paths:
raise FileNotFoundError(f"No files matched pattern: {input_pattern}")
merged = []
for path in input_paths:
with open(path, "r", encoding="utf-8") as f:
data = json.load(f)
if not isinstance(data, list):
raise ValueError(f"{path} is not a JSON list, got {type(data)}")
merged.extend(data)
output_dir = os.path.dirname(args.output_path)
if output_dir:
os.makedirs(output_dir, exist_ok=True)
with open(args.output_path, "w", encoding="utf-8") as f:
json.dump(merged, f, ensure_ascii=False, indent=2)
print(f"Merged {len(input_paths)} files into {args.output_path}")
print(f"Total items: {len(merged)}")
if __name__ == "__main__":
main()
|