python 从 bash shell 调用自定义函数

Question

如何从 bash shell 调用 python 脚本中的自定义函数？

我尝试使用 sys.argv[1]，但无法正常工作。

例如，

import sys

if __name__=='__main__':
    try:
        func = sys.argv[1]
    except: func = None

def function1():
~~~~~~~~
return a

def function2():
~~~~~~~~
return b

在这里，我想通过键入 like 来调用函数 1 或函数 2

$script.py 函数1

$script.py 函数2

Answer 1

您正在获取函数的名称，但并未运行它。你应该首先检查函数名称是否是你的函数之一而不是执行它:

if __name__=='__main__':
    try:
        func = sys.argv[1]
    except: 
        func = None

functions = {    
              "function1": function1,
              "function2": function2
              }

if func in functions:
    functions[func]()

---

更简单的解决方案:

 if func == "function1":
     function1()
 elif func == "function2":
     function2()