click_custom_group_with_command_prefix_aliasing.py

1import click
2
3class AliasedGroup(click.Group):
4
5    def get_command(self, ctx, name):
6        rv = click.Group.get_command(self, ctx, name)
7        if rv is not None:
8            return rv
9        matches = [x for x in self.list_commands(ctx)
10                   if x.startswith(name)]
11        if not matches:
12            return None
13        elif len(matches) == 1:
14            return click.Group.get_command(self, ctx, matches[0])
15        ctx.fail(f"Too many matches: {', '.join(sorted(matches))}")
16
17    def resolve_command(self, ctx, args):
18        # always return the full command name
19        _, cmd, args = super().resolve_command(ctx, args)
20        return cmd.name, cmd, args
21
22@click.group(cls=AliasedGroup)
23def cli():
24    pass
25
26@cli.command()
27def push():
28    click.echo("Pushing...")
29
30@cli.command()
31def pull():
32    click.echo("Pulling...")
33
34if __name__ == '__main__':
35    cli()