从 Linux 控制台运行函数（来自 .py 文件）

Question

可能标题说的不是很清楚，我详细说一下。

我有一个 python 脚本，可以打开 ppm 文件，应用选定的过滤器（旋转...）并创建新图片。直到这里一切正常。

但我想通过 Linux 控制台执行相同的操作，例如：

ppmfilter.py ROTD /path/imageIn.ppm /path/imageOut.ppm

这里 ROTD 是应用旋转的函数的名称。

我不知道该怎么做，我正在寻找一个可以让我做到这一点的图书馆。

期待您的帮助。

PS：我使用的是python 2.7

Answer 1

有一个相对简单的方法：

您可以使用“globals()”确定全局名称（函数、变量等）。这为您提供了所有全局符号的字典。您只需要检查类型（使用 type() 和模块类型），如果它是一个函数，您可以使用 sys.argv 调用它：

import types
import sys

def ROTD(infile, outfile):
    # do something

if __name__ == '__main__':
    symbol = globals().get(sys.argv[1])
    if hasattr(symbol, '__call__'):
        symbol(*sys.argv[2:])

这会将程序参数（不包括文件名和命令名）传递给的功能。

编辑：请不要忘记错误处理。为了清楚起见，我省略了它。

Answer 2

使用 main() 函数：

def main()
   # call your function here

if __name__ == "__main__":
   main()

Answer 3

一个很好的方法是在模块内定义一个大字典{alias: function}。例如：

actions = {
    'ROTD': ROTD,
    'REFL': reflect_image,
    'INVT': invIm,
}

你明白了。然后获取第一个命令行参数并将其解释为该字典的键，并将 actions[k] 应用于其余参数。

Answer 4

您可以在 ppmfilter.py main 部分中执行以下操作：

if __name__ == "__main__":
  import sys
  ROTD(sys.argv[1], sys.argv[2]) # change according to the signature of the function

并调用它： python ppmfilter.py file1 file2

您还可以运行 python -c 在包含 *.py 文件的目录中：

python -c "import ppmfilter; ppmfilter.ROTD('/path/to/file1', '/path/to/file2')"