summaryrefslogtreecommitdiff
path: root/examples/bashcompletion/bashcompletion.py
blob: 8121105999a8b3095e9b5d2e28862766d98e5595 (plain)
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
import click
import os

@click.group()
def cli():
    pass

@cli.command()
@click.option('-c', '--count', type=click.INT, default=1)
@click.argument("first", type=click.STRING, autocompletion=["John", "Bob", "Fred"])
@click.argument("last", type=click.STRING, autocompletion=["Smith", "Simon", "Doe"])
def cmd1(count, first, last):
    for c in range(count):
        click.echo('Name: %s %s' % (first, last))

def get_env_vars(ctx, incomplete, cwords, cword):
    return os.environ.keys()

@cli.command()
@click.argument("envvar", type=click.STRING, autocompletion=get_env_vars)
def cmd2(envvar):
    click.echo('Environment variable: %s' % envvar)
    click.echo('Value: %s' % os.environ[envvar])

@cli.command()
def cmd3():
    pass