使用argparse Python3实现任务自动化

import argparse

def main(task, input_file, output_file):
    if task == 'copy':
        with open(input_file, 'r') as infile, open(output_file, 'w') as outfile:
            outfile.write(infile.read())
        print(f"Copied {input_file} to {output_file}")
    elif task == 'count':
        with open(input_file, 'r') as infile:
            content = infile.read()
            word_count = len(content.split())
        print(f"Word count in {input_file}: {word_count}")
    else:
        print("Unknown task")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Task automation tool")
    parser.add_argument('task', choices=['copy', 'count'], help="Task to perform")
    parser.add_argument('--input-file', required=True, help="Input file path")
    parser.add_argument('--output-file', help="Output file path (required for copy task)")

    args = parser.parse_args()

    main(args.task, args.input_file, args.output_file)