如何使用 argparse 实现以下功能？

Question

我想让我的 python 脚本如下所示：

• example.py -a 数字
  
• example.py -a -f 输入文件

所以第一个命令应该只取一个数字并执行一些操作。
但是第二个命令应该从文件（.txt）中获取数字并执行相同的操作

例如：
-a number 选项读取数字并打印到屏幕
-a -f inputfile 也应该读取数字并打印到屏幕，但来自 .txt 文件（可以有多个数字）

有可能吗？
如果没有，请提出一个最好的方法。 谢谢

Answer 1

从用户的角度来看，您的想法没有意义。

含义：当你说-f时，用户已经明白该工具将打开相应的文件并处理它。因此，您不应该要求他在此基础上提供-a。

但你当然可以做到。我会在这里做什么：为-a 开关提供一个默认 值，例如here。

然后您的脚本可以检查是否使用了该 default 值。然后你可以根据它做出决定，比如：

• 如果-a 有它的默认值并且给定了-f：从文件中读取并做-a 应该做的事情
• 如果-a 没有默认值并且-f 被给出： ??? （你必须想出这意味着什么）
• ...

那些......是我认为你的方法不优雅的一个症状：它确实使整个论点处理变得非常复杂。我会选择-a NUMBER 或 -f file 和 not 允许混合使用。你看，任何界面都应该容易做正确的事，而困难做错事。允许标志/开关的组合 只会导致可能路径的爆炸式增长。这意味着您必须处理所有这些，并且用户必须记住那些实际上是有效的。

并给出您的评论：当然，您可以将其统一为让-a 采用要么 一个数字或一个表示文件名的字符串。

然后您只需指示 argparse -a 存储 string 值。然后您的脚本可以决定该字符串是数字还是有效的文件名。然后做必要的事情。像这样的：

from __future__ import print_function
import os.path
import argparse

parser = argparse.ArgumentParser()
parser.add_argument("-a", "--add",
                    help="whatever",
                    type=str)
args = parser.parse_args()

if args.add:
    if args.add.isdigit():
        print("input contains only digits: {}".format(args.add))
    else:
        print("input contains other stuff: {}".format(args.add))
        if os.path.isfile(args.add):
            print("input is a valid file name!")