基于前一个参数的条件选择

Question

我想创建一个脚本，它接受两个应该被使用的参数:

• directory_path,
• files——目录下的文件列表。

我写过这样的东西:

#!/usr/bin/python3
import argparse
import os
import argcomplete
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("directory_path",
help="a path to some directory",
nargs=1)
# conditional choices of an argument
conditional_choices = [os.listdir(parser.parse_known_args()[0].directory_path[0])]
parser.add_argument("files",
metavar="FILES",
nargs='+',
choices=conditional_choices)
argcomplete.autocomplete(parser)
args = parser.parse_args()
print("directory_path {}".format(args.directory_path))
print("files {}".format(args.files))

所以文件参数取决于directory_path论点。

使用:Python3.8

问题对于上面的代码片段，bash-completion(从register-python-argcomplete3构建))查找文件争论不起作用。如果我在有效的命令(包含路径和文件)之后按enter键，那么我会得到一个错误

error: argument FILES: invalid choice: ...

Answer 1

第一步是值得的argcomplete我根据它创建了一个解决方案

#!/usr/bin/python3
# PYTHON_ARGCOMPLETE_OK
import argparse
import os
import argcomplete

def files_names(prefix, parsed_args, **kwargs):
absolute_pat = os.path.abspath(parsed_args.directory_path[0])
files = [file for file in os.listdir(absolute_pat) if os.path.isfile(os.path.join(absolute_pat, file))]
return files

if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("directory_path",
help="a path to some directory",
nargs=1)
parser.add_argument("files",
metavar="FILES",
nargs='+').completer = files_names
argcomplete.autocomplete(parser)
args = parser.parse_args()
print("directory_path {}".format(args.directory_path))
print("files {}".format(args.files))

*从argcomplete测试目录

要调试完成，可以设置_ARC_DEBUG变量，以启用详细调试输出