基于argparse的调用函数

Question

我是python的新手，目前正在使用它。我有一个脚本，对设备执行一些API调用。我想扩展功能并根据调用脚本时给出的参数调用不同的函数。

目前，我有以下内容：

parser = argparse.ArgumentParser()
parser.add_argument("--showtop20", help="list top 20 by app",
                    action="store_true")
parser.add_argument("--listapps", help="list all available apps",
                    action="store_true")
args = parser.parse_args()

我也有一个

def showtop20():
    .....

和

def listapps():
....

如何基于给定的参数调用函数（并且仅此函数）？我不想跑步

if args.showtop20:
   #code here

if args.listapps:
   #code here

因为我想在以后将主要可执行文件保持整洁的同时将不同功能移至模块。

Answer 1

由于您似乎想根据给定的参数运行一个函数，并且仅运行一个函数，因此建议您使用强制性的位置参数./prog command，而不是可选的参数（./prog --command1或./prog --command2）。

因此，应该执行以下操作：

FUNCTION_MAP = {'top20' : my_top20_func,
                'listapps' : my_listapps_func }

parser.add_argument('command', choices=FUNCTION_MAP.keys())

args = parser.parse_args()

func = FUNCTION_MAP[args.command]
func()