Question

Ubuntu 14.04 LTS中有一个文件argComp.py。我成功安装了argcomplete 1.0.0模块。

chmod +x argComp.py

然后

data = [ name1 : { address1 : [Korea, Seoul ] }, name2 : { address2 : [ USA, LA ] } ]

./argComp.py --name [tab]

./argComp.py --name
name1 name2

./argComp.py --name name1 --address [tab]
Korea Seoul

我希望在用户按 ENTER 之前通过实时读取我的参数。为此，argcomplete模块应在--name name1中读取文本 name ，然后按name1检查数据。

我可以获取数据并制作--address列表吗？

我想实现这个概念来实时浏览浏览器。

请提供帮助，并提前感谢您。

Answer 1

理解您的问题有点困难，但我认为这可能是您想要的：

Argcomplete仅在您使用argparse管理命令行参数时才有效。

对于您的示例argComp.py应该是这样的

#!/usr/bin/env python
# PYTHON_ARGCOMPLETE_OK
import argcomplete, argparse

parser = argparse.ArgumentParser(description='Test passing arguments')
parser.add_argument('--name1', metavar='NAME1', nargs=1, choices=['Korea', 'Seoul'], help="First address line")
parser.add_argument('--name2', metavar='NAME2', nargs=1, choices=['USA', 'LA'], help="Second address line")

argcomplete.autocomplete(parser)
args = parser.parse_args()

# Your code here
# Print the arguments
print(args.name1[0],"-",args.name2[0])

您可以输入

来运行argComp.py

python argComp.py --name Korea --name2 LA

打印使用详情

python argComp.py -h

如果您希望它是完全动态的，您可以将参数存储为json格式，然后将其作为环境变量传递

假设我们要解析以下数据（json格式）

'{"name1": {"address": ["korea", "Seoul"], "gender": ["Male","Female"]}, "name2": {"address": ["USA", "LA"]} }'

然后我们将它存储在一个环境变量中，让它命名为PY_ARGS 所以在终端：

$ PY_ARGS='{"name1": {"address": ["korea", "Seoul"], "gender": ["Male","Female"]}, "name2": {"address": ["USA", "LA"]} }'
$ export PY_ARGS

当然，您可以从任何您想要的地方设置此环境变量，即另一个程序。

现在我们的计划argComp.py将是

import argcomplete, argparse
import os
import json

args = os.environ['PY_ARGS'] # get the environment variable

# Parse env variable to dictionary
args_dict = json.loads(args)

# create the top-level parser
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(dest='subparser')

# Get keys as list
top_level_keys = list(args_dict.keys()) # [name1, name2]

for i in range(0,len(top_level_keys)):
    key = top_level_keys[i] # when i=0 name1
    sub_dict = args_dict[key] # when i=0 {"address": ["korea", "Seoul"], "gender": ["Male","Female"]}
    sub_parser = subparsers.add_parser(key)
    sub_level_keys = list(sub_dict.keys()) # when i=0 [address, gender]

    for k in range(0, len(sub_level_keys)):
        sub_key = sub_level_keys[k] # when k=0 address
        choices = sub_dict[sub_key] # when k=0 [korea, Seoul]
        sub_parser.add_argument('--'+sub_key, nargs=1, choices=choices)


argcomplete.autocomplete(parser)
args = parser.parse_args()

# Your code here
print(args)

现在我们可以运行argComp.py

$ ./argComp.py name1 --gender Male
Namespace(address=None, gender=['Male'], subparser='name1')

或使用地址

$ ./argComp.py name1 --gender Male --address korea
Namespace(address=['korea'], gender=['Male'], subparser='name1')

这将打印错误

$ ./argComp.py name2 --gender Male --address korea
usage: argComp.py name2 [-h] [--address {USA,LA}]
argcomplete2.py name2: error: argument --address: invalid choice: 'korea' (choose from 'USA', 'LA')

帮助很好印刷

$ ./argComp.py -h
usage: argComp.py [-h] {name1,name2} ...

positional arguments:
  {name1,name2}

optional arguments:
  -h, --help     show this help message and exit

和

$ ./argComp.py name1 -h
usage: argComp.py name1 [-h] [--gender {Male,Female}]
                             [--address {korea,Seoul}]

optional arguments:
  -h, --help            show this help message and exit
  --gender {Male,Female}
  --address {korea,Seoul}

这里的技巧是从你的json字符串创建字典并添加subparser(s)。

您可以添加更多子级别以匹配您需要的参数结构。

如何在ENTER之前获得自动完成参数？

1 个答案: