python argparse，如果省略选项，则搜索特定文件

Question

我有一个输入文件的参数，并且很容易用 argparse 处理它

parser.add_argument( '-al', nargs = 1, type = argparse.FileType( 'r' ),
                     dest = 'alphabet' )

此参数是可选的，但如果省略，我仍然需要通过搜索当前目录来获取此输入文件。如果有多个 .al 扩展名的文件或没有文件，我将引发错误，否则打开我找到的文件。

#if no alphabet look for .al file in current directory
if( args.alphabet == None ):
    AlFiles = [ f for f in os.listdir( '.' ) 
                if os.path.isfile( f ) and f.endswith( '.al' ) ]

    #there should be one file with .al extension
    if( len( AlFiles ) != 1 ):
            sys.exit( 'error: no alphabet file provided and '
                    'there are more than one or no .al file in current directory, '
                    'which leads to ambiguity' )

    args.alphabet = open( AlFiles[0], 'r' )

是否可以使用 argparse 执行此操作，可能使用默认或操作参数。 如果我在解析参数之前进行搜索并且存在异常情况，我仍然无法引发它，因为所需的文件可能由命令行参数提供。如果解析器不满足所需的参数，我想执行操作，但无法找到如何使用 argparse 执行操作。

Answer 1

您可以通过设置参数的默认属性来解决此问题。

parser = argparse.ArgumentParser()
parser.add_argument('-al',
                    type = argparse.FileType('r'),
                    default = [ f for f in os.listdir( '.' )
                                if os.path.isfile( f ) and f.endswith( '.al' )],
                    dest = 'alphabet' )

然后进行检查。这样一来，您只有一个函数来检查是否存在多个或没有 *.al 文件，无论参数是否被省略。

例如，这可以通过以下方式完成：

args =  parser.parse_args()
if isinstance(args.alphabet,types.ListType):
    if len(args.alphabet) != 1:
        parser.error("There must be exactly one alphabet in the directory")
    else:
        args.alphabet = open(args.alphabet[0])

这样，如果指定了一个字母文件或当前工作目录中只有一个字母文件，则 args.alphabet 将保存一个打开的文件，但如果 cwd 中有更多或没有，则会引发错误。

注意： 因为如果-al 参数被省略，我们会得到一个列表，argparse.FileType('r') 不会打开任何文件。 您还必须省略nargs=1，因为这将创建一个列表，其中包含一个打开的文件，即-al 参数中指定的用户。省略此属性将为我们提供用户指定的原始打开文件。

编辑：你必须import types。