1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
"""Example from https://click.palletsprojects.com/en/8.1.x/advanced/#command-aliases"""
from __future__ import annotations
from typing_extensions import assert_type
import click
class AliasedGroup(click.Group):
def get_command(self, ctx: click.Context, cmd_name: str) -> click.Command | None:
rv = click.Group.get_command(self, ctx, cmd_name)
if rv is not None:
return rv
matches = [x for x in self.list_commands(ctx) if x.startswith(cmd_name)]
if not matches:
return None
elif len(matches) == 1:
return click.Group.get_command(self, ctx, matches[0])
ctx.fail(f"Too many matches: {', '.join(sorted(matches))}")
def resolve_command(
self, ctx: click.Context, args: list[str]
) -> tuple[str | None, click.Command, list[str]]:
# always return the full command name
_, cmd, args = super().resolve_command(ctx, args)
assert cmd is not None
return cmd.name, cmd, args
@click.command(cls=AliasedGroup)
def cli() -> None:
pass
assert_type(cli, AliasedGroup)
@cli.command()
def push() -> None:
pass
@cli.command()
def pop() -> None:
pass
|
