Đây là một tập lệnh Python 3 chia một tệp thành nhiều tệp dựa trên tên tệp được cung cấp bởi các dấu phân cách. Tệp đầu vào mẫu:
This goes in foo.conf.
This goes in bar.conf.
Đây là kịch bản:
import os
import argparse
start_delimiter = '######## FILTER BEGIN'
end_delimiter = '######## FILTER END'
parser = argparse.ArgumentParser()
parser.add_argument("-i", "--input-file", required=True, help="input filename")
parser.add_argument("-o", "--output-dir", required=True, help="output directory")
args = parser.parse_args()
with open(args.input_file, 'r') as input_file:
input_data = input_file.read()
input_lines = input_data.splitlines()
while input_lines:
while input_lines and not input_lines[0].startswith(start_delimiter):
input_lines.pop(0)
if not input_lines:
break
output_filename = input_lines.pop(0).replace(start_delimiter, "").strip()
output_path = os.path.join(args.output_dir, output_filename)
print("extracting file: {0}".format(output_path))
with open(output_path, 'w') as output_file:
while input_lines and not input_lines[0].startswith(end_delimiter):
output_file.write("{0}\n".format(input_lines.pop(0)))
if not input_lines:
input_lines.pop(0)
Cuối cùng, đây là cách bạn chạy nó:
$ python3 script.py -i input-file.txt -o ./output-folder/