python - 如何在python中添加标志

Question

目前我有一个包含许多文件和独特文本的文件夹基本上有1.txt2.txt3.txt

1.txt的内容是1111112.txt的内容是2222223.txt的内容是333333

我有一个如下的脚本来打印这个txt文件的内容，如下

from os import system
import sys

text=sys.argv[1]

def loop():
    system("cat /root/Music/loop/"+text+".txt")


if __name__ == "__main__":
    loop()

所以现在当我输入 python Loop.py 1 时它将打印并显示111111

当我输入 python Loop.py 2 时它将打印并显示222222

我的问题是如何添加一个标志来打印全部？例如 python Loop.py --all

111111

222222

333333

Answer 1

您可以使用argparse以结构化方式解析命令行参数:

from os import system
import sys
import argparse
import os

base_dir = '/root/Music/loop'

parser = argparse.ArgumentParser()
parser.add_argument('--all', action='store_true')
parser.add_argument('files', nargs='*')
args = parser.parse_args()
if args.all:
    files = os.listdir(base_dir)
else:
    files = args.files

def print_file(path):
    system('cat ' + path)

if __name__ == "__main__":
    for file in files:
        print_file(os.path.join(base_dir, file))