Python如何生成含嵌套必填可选参数的命令全组合以开发Bash补全

核心逻辑是把嵌套的命令语法规则分层拆解，用递归方式处理每一层的规则类型，再拼接所有合法组合：

• 先将命令规则拆成三种基础类型节点：
  • 序列节点：所有子项按固定顺序出现，所有子项的可能取值做笛卡尔积后拼接
  • 多选一节点：子项只能选一个，所有子项的可能取值取并集
  • 可选节点：子项可以选或不选，在子项的可能取值基础上额外加空值
• 从最内层嵌套规则开始逐层向外生成所有合法取值，最终拼接得到完整命令的全组合

import itertools

# 处理多选一规则：输入各分支的可能值，返回所有分支的取值合集
def alternative(*branches):
    result = []
    for branch in branches:
        result.extend(branch)
    return result

# 处理可选规则：输入子项的可能值，返回加空值的结果（空值代表不选该参数）
def optional(sub_items):
    return [""] + sub_items

# 处理序列规则：输入各部分的可能值，返回笛卡尔积拼接后的结果
def sequence(*parts):
    products = itertools.product(*parts)
    return [" ".join(filter(None, item)).strip() for item in products]

# ------------------- 示例规则拆解生成 -------------------
# 示例规则：add disk -pool <name> { -diskid <diskid> | -diskid auto [-fx | -tdr] } [-fx] [-status { enable | disable } ]

# 1. 最内层嵌套：[-fx | -tdr] 可选多选一
inner_fx_tdr = optional(alternative(["-fx"], ["-tdr"]))
# 2. -diskid auto 加上述可选参数的序列
disk_auto_part = sequence(["-diskid auto"], inner_fx_tdr)
# 3. {} 包裹的必填多选一部分
required_disk_part = alternative(["-diskid <diskid>"], disk_auto_part)
# 4. 后面的可选 [-fx]
opt_fx = optional(["-fx"])
# 5. 可选 [-status { enable | disable } ]
opt_status = optional(sequence(["-status"], alternative(["enable"], ["disable"])))

# 最外层完整命令序列
full_commands = sequence(
    ["add disk -pool <name>"],
    required_disk_part,
    opt_fx,
    opt_status
)

# 结果验证
print(f"合法组合总数：{len(full_commands)}")
# 打印前5条组合示例
for cmd in full_commands[:5]:
    print(cmd)

运行后输出的组合总数为24，和预期完全一致。计算逻辑验证：

• 必填disk部分取值数：1（直接传diskid） + 1(auto) * 3（空/-fx/-tdr）= 4种
• 可选-fx取值数：2种（空/选中）
• 可选status取值数：3种（空/enable/disable）
• 总组合数：423=24

如果需要自动从帮助文本的语法字符串解析生成规则结构，只需要额外实现一个简单的语法解析器，处理括号匹配、分隔符识别即可，核心的组合生成逻辑不需要修改。

内容的提问来源于stack exchange，提问作者onieks