Question

我想实现Argparse简介中的示例：

import argparse

parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('integers', metavar='N', type=int, nargs='+',
                   help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',
                   const=sum, default=max,
                   help='sum the integers (default: find the max)')

args = parser.parse_args()
print(args.accumulate(args.integers))

但就我而言，我希望有更多可能的功能名称可供选择。

def a(): ...
def b(): ...
def c(): ...

parser.add_argument('-f', '--func', 
                     choices=[a, b, c], 
                     required=True,
                     help="""Choose one of the specified function to be run.""")
parser.func()

像那样使用它并没有按预期工作，我得到了

$ python program.py -f=a
program.py: error: argument -f/--func: invalid choice: 'a' (choose from 
<function a at 0x7fa15f32f5f0>, 
<function b at 0x7fa15f32faa0>, 
<function c at 0x7ff1967099b0>)

我知道我可以使用基本的字符串参数和流控制来解决它，但是如果我可以直接使用解析器参数作为函数名称，那么它将更容易混乱并且更容易维护。

Answer 1

您可以使用基本字符串参数并通过使用dict从名称中获取实际函数来最大限度地减少混乱。

我使用的是Python 2.6.6，因此我无法使用argparse，但此代码应该为您提供一般性的想法：

#!/usr/bin/env python

def a(): return 'function a'
def b(): return 'function b'
def c(): return 'function c'

func_list = [a, b, c]
func_names = [f.func_name for f in func_list]
funcs_dict = dict(zip(func_names, func_list))

f = funcs_dict['b']
print f()

<强>输出

function b

因此，您可以将func_names传递给argparse并使用funcs_dict紧凑地检索所需的函数。

Answer 2

您需要确保choice是字符串（用户无法在命令行上输入Python函数对象）。您可以使用字典将这些字符串解析为函数。例如：

# Example functions:
def a(i):
    return i + 1
def b(i):
    return i + 2
def c(i):
    return i + 3

# a dictionary mapping strings of function names to function objects:
funcs = {'a': a, 'b': b, 'c': c}

# Add the -f/--func argument: valid choices are function _names_
parser.add_argument('-f', '--func', dest='func',
                     choices=['a', 'b', 'c'], 
                     required=True,
                     help="""Choose one of the specified function to be run.""")

args = parser.parse_args()

# Resolve the chosen function object using its name:    
chosen_func = funcs[args.func]

Answer 3

我的分析，我可能会出错，但这是我的理解。

ArgParse允许您通过强制转换机制从命令行参数创建简单对象。如果你传递字符串＆＃39; 5＆＃39;并指定您正在等待整数。

实际上你正试图从字符串＆＃39; a＆＃39;中获取一个函数。没有铸造方法来做到这一点。这是我提出的解决问题的建议：

import argparse

def foo():
    print("called foo")

def bar():
    print("called bar")

functions = [foo, bar] #list your functions here
functions = { function.__name__ : function for function in functions}

parser = argparse.ArgumentParser()

parser.add_argument('-f', '--func', 
                  choices=list(functions.keys()),
                  required=True,
                  help="""Choose one of the specified function to be run.""")
args = parser.parse_args()
functions[args.func]()

你现在只需要在开始时将你的功能注册到te list函数并在最后一行之后调用它们，感谢构建的函数索引并自动替换函数列表

如何使用argparse参数作为函数名

3 个答案: