使用许多可选的子参数设计 argparse

Question

我目前正在为我的库编写一个脚本，但我最终陷入了如何设计将有很多选项和子参数的 argparse 的问题。

目前我正在设计搜索功能，它有以下选项，有些是必需的，有些不是：

search_session_id - 必需
user_session_id - 必需
discover_fields - 可选
start_time - 可选
end_time - 可选
summary_fields - 可选
field_summary - 可选
local_search - 可选

我的问题如下：

如果所有可选项都需要一起工作，我如何制作 argparse 和 if 语句，但如果只定义其中一个也能工作？

如果我需要检查每一个组合，我最终会得到这样的结果：

#!/usr/bin/env python3

"""Script to generate searches on the ArcSight Logger"""

import arcsightrest
import argparse

parser = argparse.ArgumentParser(description='Script used to send search '
                                             'queries to ArcSight Logger API')
parser.add_argument('-t', '--target',
                    help='IP Address of the Loggger', required=True)
parser.add_argument('-u', '--username',
                    help='Username to access the logger', required=True)
parser.add_argument('-p', '--password',
                    help='Password to access the logger', required=True)
parser.add_argument('-ussl', '--unsecuressl', action='store_true',
                    help='Disable ssl warnings', )
parser.add_argument('-w', '--wait', action='store_true',
                    help='Wait for query to finish', )
parser.add_argument('-q', '--query',
                    help='Query to be used in the search')
parser.add_argument('-st', '--starttime',
                    help='From which time the query should look')
parser.add_argument('-et', '--endtime',
                    help='To which time the query should look')
parser.add_argument('-e', '--event',
                    help='Events based input search id')
parser.add_argument('-s', '--status',
                    help='Status of running search')
args = (parser.parse_args())

"""
Sets the target Logger Server
"""
arcsightrest.ArcsightLogger.TARGET = args.target

"""
Gets login token from the Logger API
"""
arc = arcsightrest.ArcsightLogger(args.username, args.password,
                                  args.unsecuressl)
"""
Checks if query is used, and starts a search
"""
if args.query:
    if args.starttime:
        search_id, response = arc.search(args.query, start_time=args.starttime,
                                         end_time=args.endtime)
    search_id, response = arc.search(args.query)

    if args.starttime and args.discover_fields:
        search_id, response = arc.search(args.query, start_time=args.starttime,
                                         end_time=args.endtime,
                                         discover_fields=args.discover_fields)
    print('The search id is {}'.format(search_id))
    if response:
        print('The search has successfully started')

如您所见，我可以无休止地继续，以创建具有可选参数的每一个组合的 if 语句。必须有更简单的方法来设计这个吗？如果我只是将它解析为 kwargs，它们将不会以正确的格式发送，或者我会要求使用脚本的人写 end_time=SOMETIME 之类的东西，而不仅仅是 --endtime TIME。现在这似乎是一个很小的代价，但如果我需要将每个函数及其所有参数添加到脚本中，那么这将变得更长更乏味。

Answer 1

您可以将传递给 arc.search 的所有可选关键字参数收集到 dict，然后在调用函数时将其解压缩：

import argparse

parser = argparse.ArgumentParser(description='Script used to send search '
                                             'queries to ArcSight Logger API')
parser.add_argument('-t', '--target',
                    help='IP Address of the Loggger', required=True)
parser.add_argument('-u', '--username',
                    help='Username to access the logger', required=True)
parser.add_argument('-p', '--password',
                    help='Password to access the logger', required=True)
parser.add_argument('-q', '--query',
                    help='Query to be used in the search')
parser.add_argument('-st', '--starttime',
                    help='From which time the query should look')
parser.add_argument('-et', '--endtime',
                    help='To which time the query should look')
args = (parser.parse_args())

# Mock search
def search(query, start_time=None, end_time=None, discover_fields=None):
    return 'Id', ', '.join(str(x) for x in [start_time, end_time, discover_fields])

"""
Checks if query is used, and starts a search
"""
if args.query:
    # {name used in argparse: search parameter name}
    query_args = {
        'starttime': 'start_time',
        'endtime': 'end_time',
        'discover_fields': 'discover_fields'
    }
    d = vars(args)
    real_args = {v: d[k] for k, v in query_args.items() if k in d}
    search_id, response = search(args.query, **real_args)

    print('The search id is {}'.format(search_id))
    print('Response is {}'.format(response))

输出：

>python test.py -t foo -u user -p pass -q
query -st start -et end
The search id is Id
Response is start, end, None

由于解析器使用的一些参数名称与传递给 search 的参数名称不同，因此需要重新映射名称。 vars is used to create a dict from Namespace parse_args() 返回的对象。然后字典理解迭代映射的参数名称，选择给定用户的参数并创建一个新字典，其中的键名是 arc.search 理解的。最后 **real_args 在函数调用中解压名为参数的字典。

使用许多可选的子参数设计 argparse

Designing argparse with many optional sub-arguments

python

argparse

python-3.x