使用 Python 命令行参数作为函数名和函数值

Question

与这个问题有关Command line arguments in python.

使用 SYS 模块，如何在我的代码中使用命令行参数作为函数名和函数值 - 而不导入其他模块？

我想要一个仅使用 sys.另外，请不要使用可变长度参数的答案。这些令人困惑。假设在命令行中只指定了函数名和一个函数变量。

import sys

def reversal(aaa): return aaa[::-1]

a = sys.argv[1]
b = sys.argv[2]

print a(b)

在命令行

cpu_location$ python blah.py reversal 'abcdefg'

Traceback (most recent call last):
File "blah.py", line 8, in <module>
print a(b)
TypeError: 'str' object is not callable

我想知道如何让 sys.argv[1] 被认为是一个函数名，从而调用我定义的函数。

我看到的其他帖子是以下内容的混搭： - 处理 C/C++ 并添加一些其他模块 - 根本不使用 sys - 使用 argv 项作为函数的值和其他文件的名称，而不是函数的名称

Answer 1

比eval 解决方案更好的是：

a = globals()[sys.argv[1]]
a(b)

globals() 返回一个将全局变量名称映射到这些全局变量的字典。所以globals()['reversal'] 计算为reversal 函数。

它比eval 函数更安全。使用您的方法，您可以执行以下操作：

python blah.py 'lambda x: x+"hi"' foobar

这将打印foobarhi，这是出乎意料的，因为这不是函数名。

Answer 2

2 小时后，我找到了答案。我认为以非常简单的方式在这里发布它是值得的。

Python 中基本上没有“函数”数据类型，但有人确实提到了一个内置的函数 eval。 Execute python commands passed as strings in command line using python -c（我自己的例子不需要-c）

解决办法，就是改变

a = sys.argv[1]

到

a = eval(sys.argv[1])

这将使传入的单词，反转，被评估。它将评估为一个函数。那么 a(b) 调用将是对字符串函数的完美调用，就像它的定义方式一样。输出如下：

cpu_location$ python blah.py reversal unquoted
detouqnu

cpu_location$ python blah.py reversal 'withquotes'
setouqhtiw

Answer 3

使用谷歌模块：火

pip install fire

这是一个简单的例子：

import fire

class Calculator(object):
  """A simple calculator class."""

  def double(self, number):
    return 2 * number

if __name__ == '__main__':
  fire.Fire(Calculator)

然后，您可以从命令行运行：

python calculator.py double 10  # 20
python calculator.py double --number=15  # 30